-
-
Save r2dev/ca1ec5f74c0cfc68c8dfbedf66eff4ff to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| const std = @import("std"); | |
| const net = std.net; | |
| const fs = std.fs; | |
| const os = std.os; | |
| pub const io_mode = .evented; | |
| pub fn main() anyerror!void { | |
| var general_purpose_allocator = std.heap.GeneralPurposeAllocator(.{}){}; | |
| const allocator = &general_purpose_allocator.allocator; | |
| var server = net.StreamServer.init(.{}); | |
| defer server.deinit(); | |
| // TODO handle concurrent accesses to this hash map | |
| var room = Room{ .clients = std.AutoHashMap(*Client, void).init(allocator) }; | |
| try server.listen(net.Address.parseIp("127.0.0.1", 0) catch unreachable); | |
| std.debug.warn("listening at {}\n", .{server.listen_address}); | |
| while (true) { | |
| const client = try allocator.create(Client); | |
| client.* = Client{ | |
| .conn = try server.accept(), | |
| .handle_frame = async client.handle(&room), | |
| }; | |
| try room.clients.putNoClobber(client, {}); | |
| } | |
| } | |
| const Client = struct { | |
| conn: net.StreamServer.Connection, | |
| handle_frame: @Frame(handle), | |
| fn handle(self: *Client, room: *Room) !void { | |
| try self.conn.file.writeAll("server: welcome to teh chat server\n"); | |
| while (true) { | |
| var buf: [100]u8 = undefined; | |
| const amt = try self.conn.file.read(&buf); | |
| const msg = buf[0..amt]; | |
| room.broadcast(msg, self); | |
| } | |
| } | |
| }; | |
| const Room = struct { | |
| clients: std.AutoHashMap(*Client, void), | |
| fn broadcast(room: *Room, msg: []const u8, sender: *Client) void { | |
| var it = room.clients.iterator(); | |
| while (it.next()) |entry| { | |
| const client = entry.key; | |
| if (client == sender) continue; | |
| client.conn.file.writeAll(msg) catch |e| std.debug.warn("unable to send: {}\n", .{e}); | |
| } | |
| } | |
| }; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Tested with zig 0.6.0+4e63ca. The latest master branch build | |
| from https://ziglang.org/download/ should work just fine. | |
| zig build-exe basic-tcp-chat.zig | |
| ./basic-tcp-chat | |
| This will print "listening at listening at 127.0.0.1:$PORT" | |
| To play with it you have to open 2 terminal windows and in each one | |
| (replacing $PORT with the one printed): | |
| nc 127.0.0.1 $PORT | |
| Now the terminals can talk to each other when you type your message and press enter. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment