move lambda runtime interface out of main
This commit is contained in:
parent
602a23b8f4
commit
35e09a2472
57
src/lambda.zig
Normal file
57
src/lambda.zig
Normal file
|
@ -0,0 +1,57 @@
|
||||||
|
const std = @import("std");
|
||||||
|
const requestz = @import("requestz");
|
||||||
|
|
||||||
|
pub fn run(event_handler: fn (*std.mem.Allocator, []const u8) anyerror![]const u8) !void { // TODO: remove inferred error set?
|
||||||
|
const prefix = "http://";
|
||||||
|
const postfix = "/2018-06-01/runtime/invocation";
|
||||||
|
const lambda_runtime_uri = std.os.getenv("AWS_LAMBDA_RUNTIME_API");
|
||||||
|
|
||||||
|
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
||||||
|
defer _ = gpa.deinit();
|
||||||
|
const allocator = &gpa.allocator;
|
||||||
|
|
||||||
|
const url = try std.fmt.allocPrint(allocator, "{s}{s}{s}/next", .{ prefix, lambda_runtime_uri, postfix });
|
||||||
|
defer allocator.free(url);
|
||||||
|
|
||||||
|
std.log.info("Event Url: {s}", .{url});
|
||||||
|
|
||||||
|
while (true) {
|
||||||
|
var req_alloc = std.heap.ArenaAllocator.init(allocator);
|
||||||
|
defer req_alloc.deinit();
|
||||||
|
const req_allocator = &req_alloc.allocator;
|
||||||
|
var client = try requestz.Client.init(req_allocator);
|
||||||
|
// defer client.deinit();
|
||||||
|
var response = client.get(url, .{}) catch |err| {
|
||||||
|
// TODO: report error
|
||||||
|
std.log.err("Get fail: {}", .{err});
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
defer response.deinit();
|
||||||
|
|
||||||
|
var request_id: ?[]const u8 = null;
|
||||||
|
for (response.headers.items()) |h| {
|
||||||
|
if (std.mem.indexOf(u8, h.name.value, "Lambda-Runtime-Aws-Request-Id")) |_|
|
||||||
|
request_id = h.value;
|
||||||
|
}
|
||||||
|
if (request_id == null) {
|
||||||
|
// TODO: report error
|
||||||
|
std.log.err("Could not find request id: skipping request", .{});
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const req_id = request_id.?;
|
||||||
|
|
||||||
|
const event_response = event_handler(req_allocator, response.body) catch |err| {
|
||||||
|
// TODO: report error
|
||||||
|
std.log.err("Error posting response for request id {s}: {}", .{ req_id, err });
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
const response_url = try std.fmt.allocPrint(req_allocator, "{s}{s}{s}/{s}/response", .{ prefix, lambda_runtime_uri, postfix, req_id });
|
||||||
|
// defer req_allocator.free(response_url);
|
||||||
|
var resp_resp = client.post(response_url, .{ .content = event_response }) catch |err| {
|
||||||
|
// TODO: report error
|
||||||
|
std.log.err("Error posting response for request id {s}: {}", .{ req_id, err });
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
defer resp_resp.deinit();
|
||||||
|
}
|
||||||
|
}
|
69
src/main.zig
69
src/main.zig
|
@ -1,67 +1,20 @@
|
||||||
const std = @import("std");
|
const std = @import("std");
|
||||||
const requestz = @import("requestz");
|
const lambda = @import("lambda.zig");
|
||||||
|
|
||||||
// zig build -Drelease-safe && strip zig-out/bin/bootstrap && zip -j9 function.zip zig-out/bin/bootstrap && aws lambda update-function-code --function-name zig-test --zip-file fileb://function.zip
|
// zig package -Drelease-safe does the following:
|
||||||
|
//
|
||||||
|
// zig build -Drelease-safe
|
||||||
|
// strip zig-out/bin/bootstrap # should this be stripped?
|
||||||
|
// zip -j9 zig-out/bin/function.zip zig-out/bin/bootstrap
|
||||||
|
//
|
||||||
|
// zig deploy will also do something like:
|
||||||
|
//
|
||||||
|
// aws lambda update-function-code --function-name zig-test --zip-file fileb://function.zip
|
||||||
pub fn main() anyerror!void {
|
pub fn main() anyerror!void {
|
||||||
try run(handler);
|
try lambda.run(handler);
|
||||||
}
|
}
|
||||||
|
|
||||||
fn handler(allocator: *std.mem.Allocator, event_data: []const u8) ![]const u8 {
|
fn handler(allocator: *std.mem.Allocator, event_data: []const u8) ![]const u8 {
|
||||||
_ = allocator;
|
_ = allocator;
|
||||||
return event_data;
|
return event_data;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn run(event_handler: fn (*std.mem.Allocator, []const u8) anyerror![]const u8) !void { // TODO: remove inferred error set?
|
|
||||||
const prefix = "http://";
|
|
||||||
const postfix = "/2018-06-01/runtime/invocation";
|
|
||||||
const lambda_runtime_uri = std.os.getenv("AWS_LAMBDA_RUNTIME_API");
|
|
||||||
|
|
||||||
var gpa = std.heap.GeneralPurposeAllocator(.{}){};
|
|
||||||
defer _ = gpa.deinit();
|
|
||||||
const allocator = &gpa.allocator;
|
|
||||||
|
|
||||||
const url = try std.fmt.allocPrint(allocator, "{s}{s}{s}/next", .{ prefix, lambda_runtime_uri, postfix });
|
|
||||||
defer allocator.free(url);
|
|
||||||
|
|
||||||
std.log.info("Event Url: {s}", .{url});
|
|
||||||
|
|
||||||
while (true) {
|
|
||||||
var req_alloc = std.heap.ArenaAllocator.init(allocator);
|
|
||||||
defer req_alloc.deinit();
|
|
||||||
const req_allocator = &req_alloc.allocator;
|
|
||||||
var client = try requestz.Client.init(req_allocator);
|
|
||||||
// defer client.deinit();
|
|
||||||
var response = client.get(url, .{}) catch |err| {
|
|
||||||
// TODO: report error
|
|
||||||
std.log.err("Get fail: {}", .{err});
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
defer response.deinit();
|
|
||||||
|
|
||||||
var request_id: ?[]const u8 = null;
|
|
||||||
for (response.headers.items()) |h| {
|
|
||||||
if (std.mem.indexOf(u8, h.name.value, "Lambda-Runtime-Aws-Request-Id")) |_|
|
|
||||||
request_id = h.value;
|
|
||||||
}
|
|
||||||
if (request_id == null) {
|
|
||||||
// TODO: report error
|
|
||||||
std.log.err("Could not find request id: skipping request", .{});
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
const req_id = request_id.?;
|
|
||||||
|
|
||||||
const event_response = event_handler(req_allocator, response.body) catch |err| {
|
|
||||||
// TODO: report error
|
|
||||||
std.log.err("Error posting response for request id {s}: {}", .{ req_id, err });
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
const response_url = try std.fmt.allocPrint(req_allocator, "{s}{s}{s}/{s}/response", .{ prefix, lambda_runtime_uri, postfix, req_id });
|
|
||||||
// defer req_allocator.free(response_url);
|
|
||||||
var resp_resp = client.post(response_url, .{ .content = event_response }) catch |err| {
|
|
||||||
// TODO: report error
|
|
||||||
std.log.err("Error posting response for request id {s}: {}", .{ req_id, err });
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
defer resp_resp.deinit();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user