depends protobuf defs only in api/trace/ structure

Signed-off-by: drumato <[email protected]>

Author: Drumato

src/api/trace.zig

@@ -4,6 +4,24 @@ pub const Tracer = @import("trace/tracer.zig").Tracer;
 pub const TracerProvider = @import("trace/provider.zig").TracerProvider;
 pub const TracerConfig = @import("trace/config.zig").TracerConfig;
 
+const span = @import("trace/span.zig");
+pub const Span = span.Span;
+pub const SpanKind = span.SpanKind;
+pub const Status = span.Status;
+pub const Event = @import("trace/event.zig").Event;
+pub const Code = @import("trace/code.zig").Code;
+pub const Link = @import("trace/link.zig").Link;
+
+test {
+    _ = @import("trace/code.zig");
+    _ = @import("trace/config.zig");
+    _ = @import("trace/event.zig");
+    _ = @import("trace/link.zig");
+    _ = @import("trace/provider.zig");
+    _ = @import("trace/span.zig");
+    _ = @import("trace/tracer.zig");
+}
+
 pub const TraceID = struct {
     value: [16]u8,
 
@@ -15,6 +33,27 @@ pub const TraceID = struct {
         };
     }
 
+    pub fn zero() Self {
+        return init([16]u8{
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+            0,
+        });
+    }
+
     pub fn isValid(self: Self) bool {
         for (self.value) |item| {
             if (item != 0) {
@@ -37,6 +76,10 @@ pub const SpanID = struct {
         };
     }
 
+    pub fn zero() Self {
+        return init([8]u8{ 0, 0, 0, 0, 0, 0, 0, 0 });
+    }
+
     pub fn isValid(self: Self) bool {
         for (self.value) |item| {
             if (item != 0) {
@@ -48,12 +91,6 @@ pub const SpanID = struct {
     }
 };
 
-test {
-    _ = @import("trace/config.zig");
-    _ = @import("trace/provider.zig");
-    _ = @import("trace/tracer.zig");
-}
-
 test "TraceID isValid" {
     try std.testing.expect(TraceID.init([16]u8{ 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }).isValid());
     try std.testing.expect(!TraceID.init([16]u8{ 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }).isValid());

src/api/trace/code.zig

@@ -0,0 +1,5 @@
+pub const Code = enum {
+    Unset,
+    Error,
+    Ok,
+};

src/api/trace/event.zig

@@ -0,0 +1,7 @@
+const attributes = @import("../../attributes.zig");
+
+pub const Event = struct {
+    name: []const u8,
+    attributes: []attributes.Attribute,
+    time_uniX_nano: u64 = 0,
+};

src/api/trace/link.zig

@@ -0,0 +1,7 @@
+const attributes = @import("../../attributes.zig");
+pub const Link = struct {
+    // TODO
+    // span_context: trace.SpanContext,
+
+    attributes: []attributes.Attribute,
+};

src/api/trace/span.zig

@@ -0,0 +1,41 @@
+const std = @import("std");
+
+const attribute = @import("../../attributes.zig");
+const trace = @import("../trace.zig");
+const pb = @import("../../opentelemetry/proto/trace/v1.pb.zig");
+
+/// Span represents a span of a trace.
+/// It is the only component that depends protobuf definition.
+pub const Span = struct {
+    trace_id: trace.TraceID = trace.TraceID.zero(),
+    span_id: trace.SpanID = trace.SpanID.zero(),
+    // TODO: maybe not string
+    trace_state: []const u8 = "",
+    parent_span_id: ?trace.SpanID = null,
+    flags: u32 = 0,
+    name: []const u8 = "",
+    kind: SpanKind = SpanKind.Unspecified,
+    start_time_unix_nano: u64 = 0,
+    end_time_unix_nano: u64 = 0,
+    attributes: []attribute.Attribute = undefined,
+    // dropped_attributes_count
+    events: []trace.Event,
+    // dropped_events_count
+    links: []trace.Link = undefined,
+    // dropped_links_count
+    status: ?Status = null,
+};
+
+pub const SpanKind = enum {
+    Unspecified,
+    Internal,
+    Server,
+    Client,
+    Producer,
+    Consumer,
+};
+
+pub const Status = struct {
+    code: trace.Code,
+    description: []const u8,
+};

src/sdk/trace/exporters/stdout_exporter.zig

@@ -1,6 +1,6 @@
 const std = @import("std");
-const pb = @import("../../../opentelemetry/proto/trace/v1.pb.zig");
-const protobuf = @import("protobuf");
+
+const trace = @import("../../../api/trace.zig");
 const SpanExporter = @import("../span_exporter.zig").SpanExporter;
 
 /// GenericWriterExporter is the generic SpanExporter that outputs spans to the given writer.
@@ -18,7 +18,7 @@ fn GenericWriterExporter(
             };
         }
 
-        pub fn exportSpans(ctx: *anyopaque, spans: []pb.Span) anyerror!void {
+        pub fn exportSpans(ctx: *anyopaque, spans: []trace.Span) anyerror!void {
             const self: *Self = @ptrCast(@alignCast(ctx));
             try std.json.stringify(spans, .{}, self.writer);
         }
@@ -51,8 +51,9 @@ test "GenericWriterExporter" {
     var inmemory_exporter = InmemoryExporter.init(out_buf.writer());
     var exporter = inmemory_exporter.asSpanExporter();
 
-    var spans = [_]pb.Span{};
+    var spans = [_]trace.Span{};
     try exporter.exportSpans(spans[0..spans.len]);
 
+    std.debug.print("{s}", .{out_buf.items});
     // TODO: detailed output test
 }

src/sdk/trace/span_exporter.zig

@@ -1,4 +1,4 @@
-const pb = @import("../../opentelemetry/proto/trace/v1.pb.zig");
+const trace = @import("../../api/trace.zig");
 
 /// SpanExporter is the interface that provides an
 pub const SpanExporter = struct {
@@ -15,15 +15,15 @@ pub const SpanExporter = struct {
         /// but it is not defined in the OpenTelemetry specification, so for now we don't use it.
         exportSpansFn: *const fn (
             ctx: *anyopaque,
-            spans: []pb.Span,
+            spans: []trace.Span,
         ) anyerror!void,
 
         shutdownFn: *const fn (ctx: *anyopaque) anyerror!void,
     };
 
     pub fn exportSpans(
         self: Self,
-        spans: []pb.Span,
+        spans: []trace.Span,
     ) anyerror!void {
         return self.vtable.exportSpansFn(self.ptr, spans);
     }