Rename ClientState to Client

src/server/client.zig

@@ -1,7 +1,8 @@
 const Message = @import("message_parser.zig").Message;
 const std = @import("std");
 
-pub const ClientState = struct {
+const Client = @This();
+
 connect: ?Message.Connect,
 
 write_lock: std.Io.Mutex,
@@ -13,7 +14,7 @@ pub const ClientState = struct {
     connect: ?Message.Connect,
     in: *std.Io.Reader,
     out: *std.Io.Writer,
-    ) ClientState {
+) Client {
     return .{
         .connect = connect,
         .write_lock = .init,
@@ -23,7 +24,7 @@ pub const ClientState = struct {
 }
 
 /// Return true if the value was put in the clients buffer to process, else false.
-    pub fn send(self: *ClientState, io: std.Io, msg: Message) !void {
+pub fn send(self: *Client, io: std.Io, msg: Message) !void {
     try self.write_lock.lock(io);
     defer self.write_lock.unlock(io);
 
@@ -46,14 +47,13 @@ pub const ClientState = struct {
     }
 }
 
-    pub fn next(self: *ClientState, allocator: std.mem.Allocator) !Message {
+pub fn next(self: *Client, allocator: std.mem.Allocator) !Message {
     // std.debug.print("in client awaiting next message\n", .{});
     // errdefer std.debug.print("actually it was canceled\n", .{});
     // defer std.debug.print("client returning next message!\n", .{});
     return Message.next(allocator, self.from_client);
     // return self.send_queue.getOne(io);
 }
-};
 
 fn writeOk(out: *std.Io.Writer) !void {
     _ = try out.write("+OK\r\n");

src/server/main.zig

@@ -2,7 +2,7 @@ const std = @import("std");
 const Message = @import("./message_parser.zig").Message;
 pub const ServerInfo = Message.ServerInfo;
 
-const ClientState = @import("./client.zig").ClientState;
+const Client = @import("./client.zig");
 const Server = @This();
 
 const Subscription = struct {
@@ -12,7 +12,7 @@ const Subscription = struct {
 };
 
 info: ServerInfo,
-clients: std.AutoHashMapUnmanaged(usize, *ClientState) = .empty,
+clients: std.AutoHashMapUnmanaged(usize, *Client) = .empty,
 
 subs_lock: std.Io.Mutex = .init,
 subscriptions: std.ArrayList(Subscription) = .empty,
@@ -43,7 +43,6 @@ pub fn main(gpa: std.mem.Allocator, server_config: ServerInfo) !void {
     defer threaded.deinit();
     const io = threaded.io();
 
-
     var tcp_server = try std.Io.net.IpAddress.listen(try std.Io.net.IpAddress.parse(
         server.info.host,
         server.info.port,
@@ -88,7 +87,7 @@ fn processMsgs(server: *Server, io: std.Io, alloc: std.mem.Allocator) void {
     }
 }
 
-fn addClient(server: *Server, allocator: std.mem.Allocator, id: usize, client: *ClientState) !void {
+fn addClient(server: *Server, allocator: std.mem.Allocator, id: usize, client: *Client) !void {
     // server.clients.lockPointers();
     try server.clients.put(allocator, id, client);
     // server.clients.unlockPointers();
@@ -132,25 +131,25 @@ fn handleConnection(
     var reader = stream.reader(io, r_buffer);
     const in = &reader.interface;
 
-    var client_state: ClientState = .init(null, in, out);
-    try client_state.send(io, .{ .info = server.info });
+    var client: Client = .init(null, in, out);
+    try client.send(io, .{ .info = server.info });
 
     var connect_arena: std.heap.ArenaAllocator = .init(allocator);
     defer connect_arena.deinit();
-    client_state.connect = (Message.next(connect_arena.allocator(), in) catch return).connect;
+    client.connect = (Message.next(connect_arena.allocator(), in) catch return).connect;
 
-    try server.addClient(server_allocator, id, &client_state);
+    try server.addClient(server_allocator, id, &client);
     defer server.removeClient(io, server_allocator, id);
 
     // Messages are owned by the server after they are received from the client
-    while (client_state.next(server_allocator)) |msg| {
+    while (client.next(server_allocator)) |msg| {
         switch (msg) {
             .ping => {
                 // Respond to ping with pong.
-                try client_state.send(io, .pong);
+                try client.send(io, .pong);
             },
             .@"pub" => |pb| {
-                _ = io.async(publishMessage, .{ server, io, server_allocator, &client_state, pb });
+                _ = io.async(publishMessage, .{ server, io, server_allocator, &client, pb });
             },
             .sub => |sub| {
                 try server.subscribe(io, server_allocator, id, sub);
@@ -181,7 +180,7 @@ fn subjectMatches(expected: []const u8, actual: []const u8) bool {
     return std.mem.eql(u8, expected, actual);
 }
 
-fn publishMessage(server: *Server, io: std.Io, gpa: std.mem.Allocator, source_client: *ClientState, msg: Message.Pub) !void {
+fn publishMessage(server: *Server, io: std.Io, gpa: std.mem.Allocator, source_client: *Client, msg: Message.Pub) !void {
     errdefer {
         if (source_client.connect) |c| {
             if (c.verbose) {