mirror of
https://git.robbyzambito.me/zits
synced 2026-02-04 03:34:48 +00:00
sending errors on pub sub!
This commit is contained in:
@@ -6,7 +6,7 @@ pub const ClientState = struct {
|
|||||||
connect: Message.AllocatedConnect,
|
connect: Message.AllocatedConnect,
|
||||||
|
|
||||||
/// Messages that this client should receive.
|
/// Messages that this client should receive.
|
||||||
recv_queue: std.Io.Queue(Message),
|
recv_queue: *std.Io.Queue(Message),
|
||||||
recv_queue_buffer: [1024]Message = undefined,
|
recv_queue_buffer: [1024]Message = undefined,
|
||||||
|
|
||||||
from_client: *std.Io.Reader,
|
from_client: *std.Io.Reader,
|
||||||
@@ -16,6 +16,7 @@ pub const ClientState = struct {
|
|||||||
|
|
||||||
pub fn init(
|
pub fn init(
|
||||||
io: std.Io,
|
io: std.Io,
|
||||||
|
allocator: std.mem.Allocator,
|
||||||
id: usize,
|
id: usize,
|
||||||
connect: Message.AllocatedConnect,
|
connect: Message.AllocatedConnect,
|
||||||
in: *std.Io.Reader,
|
in: *std.Io.Reader,
|
||||||
@@ -24,13 +25,13 @@ pub const ClientState = struct {
|
|||||||
var res: ClientState = .{
|
var res: ClientState = .{
|
||||||
.id = id,
|
.id = id,
|
||||||
.connect = connect,
|
.connect = connect,
|
||||||
.recv_queue = undefined,
|
.recv_queue = try allocator.create(std.Io.Queue(Message)),
|
||||||
.from_client = in,
|
.from_client = in,
|
||||||
.to_client = out,
|
.to_client = out,
|
||||||
.task = undefined,
|
.task = undefined,
|
||||||
};
|
};
|
||||||
res.recv_queue = .init(&res.recv_queue_buffer);
|
res.recv_queue.* = .init(&res.recv_queue_buffer);
|
||||||
res.task = try io.concurrent(processWrite, .{ &res, io, out });
|
res.task = try io.concurrent(processWrite, .{ &res, io });
|
||||||
|
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
@@ -38,23 +39,24 @@ pub const ClientState = struct {
|
|||||||
fn processWrite(
|
fn processWrite(
|
||||||
self: *ClientState,
|
self: *ClientState,
|
||||||
io: std.Io,
|
io: std.Io,
|
||||||
out: *std.Io.Writer,
|
|
||||||
) void {
|
) void {
|
||||||
|
std.debug.print("out pointer in write proc: {*}\n", .{self.to_client});
|
||||||
|
std.debug.print("recv queue in write proc: {*}\n", .{&self.recv_queue});
|
||||||
while (true) {
|
while (true) {
|
||||||
const message = self.recv_queue.getOne(io) catch continue;
|
const message = self.recv_queue.getOne(io) catch continue;
|
||||||
std.debug.print("got message in write loop to send to client: {any}\n", .{message});
|
std.debug.print("got message in write loop to send to client: {any}\n", .{message});
|
||||||
switch (message) {
|
switch (message) {
|
||||||
.@"+ok" => {
|
.@"+ok" => {
|
||||||
writeOk(out) catch break;
|
writeOk(self.to_client) catch break;
|
||||||
},
|
},
|
||||||
.pong => {
|
.pong => {
|
||||||
writePong(out) catch break;
|
writePong(self.to_client) catch break;
|
||||||
},
|
},
|
||||||
.info => |info| {
|
.info => |info| {
|
||||||
writeInfo(out, info) catch break;
|
writeInfo(self.to_client, info) catch break;
|
||||||
},
|
},
|
||||||
.msg => |m| {
|
.msg => |m| {
|
||||||
writeMsg(out, m) catch break;
|
writeMsg(self.to_client, m) catch break;
|
||||||
},
|
},
|
||||||
else => {
|
else => {
|
||||||
std.debug.panic("unimplemented write", .{});
|
std.debug.panic("unimplemented write", .{});
|
||||||
@@ -63,9 +65,10 @@ pub const ClientState = struct {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn deinit(self: *ClientState, io: std.Io) void {
|
pub fn deinit(self: *ClientState, io: std.Io, allocator: std.mem.Allocator) void {
|
||||||
self.task.cancel(io);
|
self.task.cancel(io);
|
||||||
self.connect.deinit();
|
self.connect.deinit();
|
||||||
|
allocator.destroy(self.recv_queue);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Return true if the value was put in the clients buffer to process, else false.
|
/// Return true if the value was put in the clients buffer to process, else false.
|
||||||
@@ -89,11 +92,10 @@ fn writeOk(out: *std.Io.Writer) !void {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub fn writePong(out: *std.Io.Writer) !void {
|
pub fn writePong(out: *std.Io.Writer) !void {
|
||||||
|
std.debug.print("out pointer: {*}\n", .{out});
|
||||||
std.debug.print("writing pong\n", .{});
|
std.debug.print("writing pong\n", .{});
|
||||||
for (0..10000) |_| {
|
_ = try out.write("PONG\r\n");
|
||||||
_ = try out.write("PONG\r\n");
|
try out.flush();
|
||||||
try out.flush();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn writeInfo(out: *std.Io.Writer, info: Message.ServerInfo) !void {
|
pub fn writeInfo(out: *std.Io.Writer, info: Message.ServerInfo) !void {
|
||||||
|
|||||||
@@ -63,6 +63,8 @@ fn handleConnection(
|
|||||||
var writer = stream.writer(io, &w_buffer);
|
var writer = stream.writer(io, &w_buffer);
|
||||||
const out = &writer.interface;
|
const out = &writer.interface;
|
||||||
|
|
||||||
|
std.debug.print("out pointer in client handler: {*}\n", .{out});
|
||||||
|
|
||||||
var r_buffer: [8192]u8 = undefined;
|
var r_buffer: [8192]u8 = undefined;
|
||||||
var reader = stream.reader(io, &r_buffer);
|
var reader = stream.reader(io, &r_buffer);
|
||||||
const in = &reader.interface;
|
const in = &reader.interface;
|
||||||
@@ -71,8 +73,8 @@ fn handleConnection(
|
|||||||
var connect_arena: std.heap.ArenaAllocator = .init(allocator);
|
var connect_arena: std.heap.ArenaAllocator = .init(allocator);
|
||||||
defer connect_arena.deinit();
|
defer connect_arena.deinit();
|
||||||
const connect = (Message.next(connect_arena.allocator(), in) catch return).connect;
|
const connect = (Message.next(connect_arena.allocator(), in) catch return).connect;
|
||||||
var client_state: ClientState = try .init(io, id, connect, in, out);
|
var client_state: ClientState = try .init(io, allocator, id, connect, in, out);
|
||||||
defer client_state.deinit(io);
|
defer client_state.deinit(io, allocator);
|
||||||
|
|
||||||
try server.addClient(allocator, id, &client_state);
|
try server.addClient(allocator, id, &client_state);
|
||||||
defer server.removeClient(allocator, id);
|
defer server.removeClient(allocator, id);
|
||||||
@@ -107,7 +109,9 @@ fn handleConnection(
|
|||||||
switch (msg) {
|
switch (msg) {
|
||||||
.ping => {
|
.ping => {
|
||||||
std.debug.print("got a ping! sending a pong.\n", .{});
|
std.debug.print("got a ping! sending a pong.\n", .{});
|
||||||
@import("./client.zig").writePong(out) catch return;
|
|
||||||
|
std.debug.print("recv queue in server loop: {*}\n", .{&client_state.recv_queue});
|
||||||
|
// @import("./client.zig").writePong(out) catch return;
|
||||||
for (0..5) |_| {
|
for (0..5) |_| {
|
||||||
if (try client_state.send(io, .pong)) {
|
if (try client_state.send(io, .pong)) {
|
||||||
std.debug.print("sent pong\n", .{});
|
std.debug.print("sent pong\n", .{});
|
||||||
|
|||||||
Reference in New Issue
Block a user