1
0
Fork 0
mirror of https://github.com/zigzap/zap.git synced 2025-10-20 23:24:09 +00:00

some refactoring

This commit is contained in:
stringnick 2024-01-18 21:33:26 +01:00
parent f13160eda6
commit 5e76e17baf
5 changed files with 56 additions and 55 deletions

View file

@ -34,7 +34,7 @@ pub fn build(b: *std.build.Builder) !void {
.{ .name = "hello", .src = "examples/hello/hello.zig" }, .{ .name = "hello", .src = "examples/hello/hello.zig" },
.{ .name = "https", .src = "examples/https/https.zig" }, .{ .name = "https", .src = "examples/https/https.zig" },
.{ .name = "hello2", .src = "examples/hello2/hello2.zig" }, .{ .name = "hello2", .src = "examples/hello2/hello2.zig" },
.{ .name = "hello3", .src = "examples/hello3/hello3.zig" }, .{ .name = "simple_router", .src = "examples/simple_router/simple_router.zig" },
.{ .name = "routes", .src = "examples/routes/routes.zig" }, .{ .name = "routes", .src = "examples/routes/routes.zig" },
.{ .name = "serve", .src = "examples/serve/serve.zig" }, .{ .name = "serve", .src = "examples/serve/serve.zig" },
.{ .name = "hello_json", .src = "examples/hello_json/hello_json.zig" }, .{ .name = "hello_json", .src = "examples/hello_json/hello_json.zig" },

View file

@ -7,46 +7,28 @@ const RouterError = error{
EmptyPath, EmptyPath,
}; };
// inline closure for RequestFn with self argument const Self = @This();
pub inline fn RequestHandler(self: anytype, func: *const fn (@TypeOf(self), zap.Request) void) *const fn (zap.Request) void {
return (opaque {
var hidden_self: @TypeOf(self) = undefined;
var hidden_func: *const fn (@TypeOf(self), zap.Request) void = undefined;
pub fn init(h_self: @TypeOf(self), h_func: *const fn (@TypeOf(self), zap.Request) void) *const @TypeOf(run) {
hidden_self = h_self;
hidden_func = h_func;
return &run;
}
fn run(req: zap.Request) void { pub const Options = struct {
hidden_func(hidden_self, req);
}
}).init(self, func);
}
pub const RouterOptions = struct {
not_found: ?zap.HttpRequestFn = null, not_found: ?zap.HttpRequestFn = null,
}; };
pub const Router = struct { routes: std.StringHashMap(zap.HttpRequestFn),
const Self = @This(); not_found: ?zap.HttpRequestFn,
routes: std.StringHashMap(zap.HttpRequestFn), pub fn init(allocator: Allocator, options: Options) Self {
not_found: ?zap.HttpRequestFn,
pub fn init(allocator: Allocator, options: RouterOptions) Self {
return .{ return .{
.routes = std.StringHashMap(zap.HttpRequestFn).init(allocator), .routes = std.StringHashMap(zap.HttpRequestFn).init(allocator),
.not_found = options.not_found, .not_found = options.not_found,
}; };
} }
pub fn deinit(self: *Self) void { pub fn deinit(self: *Self) void {
self.routes.deinit(); self.routes.deinit();
} }
pub fn handle_func(self: *Self, path: []const u8, h: zap.HttpRequestFn) !void { pub fn handle_func(self: *Self, path: []const u8, h: zap.HttpRequestFn) !void {
if (path.len == 0) { if (path.len == 0) {
return RouterError.EmptyPath; return RouterError.EmptyPath;
} }
@ -58,10 +40,12 @@ pub const Router = struct {
} }
try self.routes.put(path, h); try self.routes.put(path, h);
} }
pub fn serve(self: *Self, r: zap.Request) void { pub fn serve(self: *Self, r: zap.Request) void {
var route = self.routes.get(r.path.?); const path = if (r.path) |p| p else "/";
var route = self.routes.get(path);
if (route) |handler| { if (route) |handler| {
handler(r); handler(r);
@ -73,5 +57,4 @@ pub const Router = struct {
r.setStatus(.not_found); r.setStatus(.not_found);
r.sendBody("404 Not Found") catch return; r.sendBody("404 Not Found") catch return;
} }
} }
};

View file

@ -1,5 +1,23 @@
const std = @import("std"); const std = @import("std");
const fio = @import("fio.zig"); const fio = @import("fio.zig");
const zap = @import("zap.zig");
/// capture self for RequestFn signature support
pub inline fn RequestHandler(self: anytype, func: *const fn (@TypeOf(self), zap.Request) void) *const fn (zap.Request) void {
return (opaque {
var hidden_self: @TypeOf(self) = undefined;
var hidden_func: *const fn (@TypeOf(self), zap.Request) void = undefined;
pub fn init(h_self: @TypeOf(self), h_func: *const fn (@TypeOf(self), zap.Request) void) *const @TypeOf(run) {
hidden_self = h_self;
hidden_func = h_func;
return &run;
}
fn run(req: zap.Request) void {
hidden_func(hidden_self, req);
}
}).init(self, func);
}
/// Used internally: convert a FIO object into its string representation. /// Used internally: convert a FIO object into its string representation.
/// note: since this is called from within request functions, we don't make /// note: since this is called from within request functions, we don't make

View file

@ -52,7 +52,7 @@ pub const Tls = @import("tls.zig");
/// ``` /// ```
pub const Endpoint = @import("endpoint.zig"); pub const Endpoint = @import("endpoint.zig");
pub usingnamespace @import("router.zig"); pub const Router = @import("router.zig");
pub usingnamespace @import("util.zig"); pub usingnamespace @import("util.zig");
pub usingnamespace @import("http.zig"); pub usingnamespace @import("http.zig");