Make max backoff configurable

Author: jstrieb

src/argparse.zig

@@ -106,15 +106,15 @@ fn setFromCli(
                         std.process.exit(1);
                     }
                     switch (t) {
-                        // TODO
-                        .int, .float, .@"enum" => comptime unreachable,
+                        .int => @field(result, field.name) =
+                            try std.fmt.parseInt(field.type, args[i], 0),
                         .pointer => @field(
                             result,
                             field.name,
                         ) = try allocator.dupe(u8, args[i]),
                         .bool => comptime unreachable,
                         else => @compileError(
-                            "Disallowed struct field type.",
+                            "Disallowed or unimplemented struct field type.",
                         ),
                     }
                 }
@@ -153,13 +153,15 @@ fn setFromEnv(
                         @field(result, field.name) = value.len > 0 and
                             !std.ascii.eqlIgnoreCase(value, "false");
                     },
-                    // TODO
-                    .int, .float, .@"enum" => comptime unreachable,
+                    .int => @field(result, field.name) =
+                        try std.fmt.parseInt(field.type, entry.value_ptr.*, 0),
                     .pointer => @field(
                         result,
                         field.name,
                     ) = try allocator.dupe(u8, entry.value_ptr.*),
-                    else => @compileError("Disallowed struct field type."),
+                    else => @compileError(
+                        "Disallowed or unimplemented struct field type.",
+                    ),
                 }
                 seen_field.* = true;
             }

src/main.zig

@@ -44,6 +44,7 @@ const Args = struct {
     languages_output_file: ?[]const u8 = null,
     overview_template: ?[]const u8 = null,
     languages_template: ?[]const u8 = null,
+    max_backoff: usize = 600,
 
     const Self = @This();
 
@@ -212,7 +213,7 @@ pub fn main() !void {
         std.log.info("Collecting statistics from GitHub API", .{});
         var client: HttpClient = try .init(allocator, api_key);
         defer client.deinit();
-        stats = try Statistics.init(&client, allocator);
+        stats = try Statistics.init(&client, allocator, args.max_backoff);
     } else unreachable;
     defer stats.deinit(allocator);
 

src/statistics.zig

@@ -100,13 +100,17 @@ const Language = struct {
     }
 };
 
-pub fn init(client: *HttpClient, allocator: std.mem.Allocator) !Statistics {
+pub fn init(
+    client: *HttpClient,
+    allocator: std.mem.Allocator,
+    max_backoff: usize,
+) !Statistics {
     var arena = std.heap.ArenaAllocator.init(allocator);
     defer arena.deinit();
 
     var self: Statistics = try getRepos(allocator, &arena, client);
     errdefer self.deinit(allocator);
-    try self.getLinesChanged(&arena, client);
+    try self.getLinesChanged(&arena, client, max_backoff);
     return self;
 }
 
@@ -474,6 +478,7 @@ fn getLinesChanged(
     self: *Statistics,
     arena: *std.heap.ArenaAllocator,
     client: *HttpClient,
+    max_backoff: usize,
 ) !void {
     const T = struct {
         repo: *Repository,
@@ -515,7 +520,7 @@ fn getLinesChanged(
                 // Exponential backoff (in expectation) with jitter
                 item.delay +=
                     std.crypto.random.intRangeAtMost(i64, 2, item.delay);
-                item.delay = @min(item.delay, 600);
+                item.delay = @min(item.delay, max_backoff);
                 try q.add(item);
             },
             else => |status| {