hax/fast-cli
1
0
Fork 0
mirror of https://github.com/mikkelam/fast-cli.git synced 2025-12-18 04:44:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

simplify latency tester

Browse source
This commit is contained in:
mikkelam 2025-11-17 15:58:58 +01:00
parent 898b3b52c6
commit 6725587bfc
1 changed files with 19 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
src/lib/http_latency_tester.zig
View file

@ -1,5 +1,6 @@
const std = @import("std"); const std = @import("std");
const http = std.http; const http = std.http;
const log = std.log.scoped(.cli);
pub const HttpLatencyTester = struct { pub const HttpLatencyTester = struct {
allocator: std.mem.Allocator, allocator: std.mem.Allocator,
@ -18,15 +19,21 @@ pub const HttpLatencyTester = struct {
/// Measure latency to multiple URLs using HEAD requests /// Measure latency to multiple URLs using HEAD requests
/// Returns median latency in milliseconds, or null if all requests failed /// Returns median latency in milliseconds, or null if all requests failed
/// Zig's http client seems to be ~20ms slower than curl.
/// Let's not worry about that misreporting for now
pub fn measureLatency(self: *Self, urls: []const []const u8) !?f64 { pub fn measureLatency(self: *Self, urls: []const []const u8) !?f64 {
if (urls.len == 0) return null; if (urls.len == 0) return null;
var latencies = std.ArrayList(f64).empty; var latencies: std.ArrayList(f64) = .{};
defer latencies.deinit(self.allocator); defer latencies.deinit(self.allocator);
// HTTP client for all requests
var client = http.Client{ .allocator = self.allocator };
defer client.deinit();
// Test each URL // Test each URL
for (urls) |url| { for (urls) |url| {
if (self.measureSingleUrl(url)) |latency_ms| { if (self.measureSingleUrl(url, &client)) |latency_ms| {
try latencies.append(self.allocator, latency_ms); try latencies.append(self.allocator, latency_ms);
} else |_| { } else |_| {
// Ignore errors, continue with other URLs // Ignore errors, continue with other URLs
@ -36,42 +43,26 @@ pub const HttpLatencyTester = struct {
if (latencies.items.len == 0) return null; if (latencies.items.len == 0) return null;
log.info("Latencies: {any}", .{latencies.items});
// Return median latency // Return median latency
return self.calculateMedian(latencies.items); return self.calculateMedian(latencies.items);
} }
/// Measure latency to a single URL using connection reuse method /// Measure latency to a single URL using HEAD request
/// First request establishes HTTPS connection, second request measures pure RTT fn measureSingleUrl(self: *Self, url: []const u8, client: *http.Client) !f64 {
fn measureSingleUrl(self: *Self, url: []const u8) !f64 { _ = self;
var client = http.Client{ .allocator = self.allocator };
defer client.deinit();
// Parse URL // Parse URL
const uri = try std.Uri.parse(url); const uri = try std.Uri.parse(url);
// First request: Establish HTTPS connection (ignore timing) // Measure request/response timing
{
const server_header_buffer = try self.allocator.alloc(u8, 4096);
defer self.allocator.free(server_header_buffer);
var req = try client.request(.HEAD, uri, .{});
defer req.deinit();
try req.sendBodiless();
}
// Second request: Reuse connection and measure pure HTTP RTT
const start_time = std.time.nanoTimestamp(); const start_time = std.time.nanoTimestamp();
{ _ = try client.fetch(.{
const server_header_buffer = try self.allocator.alloc(u8, 4096); .method = .HEAD,
defer self.allocator.free(server_header_buffer); .location = .{ .uri = uri },
});
var req = try client.request(.HEAD, uri, .{});
defer req.deinit();
try req.sendBodiless();
}
const end_time = std.time.nanoTimestamp(); const end_time = std.time.nanoTimestamp();