-
-
Notifications
You must be signed in to change notification settings - Fork 78
/
simple_router.zig
110 lines (83 loc) · 2.69 KB
/
simple_router.zig
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
const std = @import("std");
const zap = @import("zap");
const Allocator = std.mem.Allocator;
fn on_request_verbose(r: zap.Request) void {
if (r.path) |the_path| {
std.debug.print("PATH: {s}\n", .{the_path});
}
if (r.query) |the_query| {
std.debug.print("QUERY: {s}\n", .{the_query});
}
r.sendBody("<html><body><h1>Hello from ZAP!!!</h1></body></html>") catch return;
}
pub const SomePackage = struct {
const Self = @This();
allocator: Allocator,
a: i8,
b: i8,
pub fn init(allocator: Allocator, a: i8, b: i8) Self {
return .{
.allocator = allocator,
.a = a,
.b = b,
};
}
pub fn getA(self: *Self, req: zap.Request) void {
std.log.warn("get_a_requested", .{});
const string = std.fmt.allocPrint(
self.allocator,
"A value is {d}\n",
.{self.a},
) catch return;
defer self.allocator.free(string);
req.sendBody(string) catch return;
}
pub fn getB(self: *Self, req: zap.Request) void {
std.log.warn("get_b_requested", .{});
const string = std.fmt.allocPrint(
self.allocator,
"B value is {d}\n",
.{self.b},
) catch return;
defer self.allocator.free(string);
req.sendBody(string) catch return;
}
pub fn incrementA(self: *Self, req: zap.Request) void {
std.log.warn("increment_a_requested", .{});
self.a += 1;
req.sendBody("incremented A") catch return;
}
};
fn not_found(req: zap.Request) void {
std.debug.print("not found handler", .{});
req.sendBody("Not found") catch return;
}
pub fn main() !void {
var gpa = std.heap.GeneralPurposeAllocator(.{
.thread_safe = true,
}){};
const allocator = gpa.allocator();
var simpleRouter = zap.Router.init(allocator, .{
.not_found = not_found,
});
defer simpleRouter.deinit();
var somePackage = SomePackage.init(allocator, 1, 2);
try simpleRouter.handle_func_unbound("/", on_request_verbose);
try simpleRouter.handle_func("/geta", &somePackage, &SomePackage.getA);
try simpleRouter.handle_func("/getb", &somePackage, &SomePackage.getB);
try simpleRouter.handle_func("/inca", &somePackage, &SomePackage.incrementA);
var listener = zap.HttpListener.init(.{
.port = 3000,
.on_request = simpleRouter.on_request_handler(),
.log = true,
.max_clients = 100000,
});
try listener.listen();
std.debug.print("Listening on 0.0.0.0:3000\n", .{});
// start worker threads
zap.start(.{
.threads = 2,
// Must be 1 if state is shared
.workers = 1,
});
}