Compiler Explorer

Source code

// This is the original file written by andrewrk in his chat demo
// on youtube
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 {
    const allocator = std.heap.direct_allocator;
    const req_listen_addr = try net.Address.parseIp4("127.0.0.1", 9000);

std.event.Loop.instance.?.beginOneEvent();

var server = net.StreamServer.init(.{});
    defer server.deinit();

try server.listen(req_listen_addr);

std.debug.warn("listening at {}\n", server.listen_address);

var room = Room{
        .clients = std.AutoHashMap(*Client, void).init(allocator),
        .lock = std.Mutex.init(),
    };

while (true) {
        const conn = try server.accept();
        const client = try allocator.create(Client);
        client.* = Client{
            .file = conn.file,
            .address = conn.address,
            .handle_frame = async client.handle(&room),
        };
        var held_lock = room.lock.acquire();
        defer held_lock.release();
        try room.clients.putNoClobber(client, {});
    }
}

const Client = struct {
    file: fs.File,
    address: net.Address,
    handle_frame: @Frame(handle),

fn handle(self: *Client, room: *Room) !void {
        var stream = self.file.inStream().stream;
        std.debug.warn("{}\n", self.address);
        try self.file.write("server: welcome to teh chat server \n");
        //try self.file.write(self.address.toString());
        //try stream.print("{}\n", self.address);

while (true) {
            var buf: [100]u8 = undefined;
            const amt = try stream.read(&buf);
            const msg = buf[0..amt];
            try room.broadcast(msg, self);
        }
    }
};

const Room = struct {
    clients: std.AutoHashMap(*Client, void),
    lock: std.Mutex,

fn broadcast(room: *Room, msg: []const u8, sender: *Client) !void {
        var held_lock = room.lock.acquire();
        defer held_lock.release();

var it = room.clients.iterator();
        while (it.next()) |entry| {
            const client = entry.key;
            if (client == sender) continue;

const fd = client.file.handle;
            _ = try os.send(fd, msg, os.MSG_NOSIGNAL);
        }
    }
};