refactor(infra): directory structure (#4615)

2026-07-25 22:38:56 +08:00 · 2023-10-18 23:30:08 +08:00
parent 814d552be8
commit bed9310519
1150 changed files with 539 additions and 584 deletions
@@ -0,0 +1,209 @@
+export const SPAN_ID_BYTES = 8;
+export const TRACE_ID_BYTES = 16;
+export const TRACE_VERSION = '00';
+export const TRACE_FLAG = '01';
+
+const BytesBuffer = Array(32);
+
+type TraceSpan = {
+  name: string;
+  spanId: string;
+  displayName: {
+    value: string;
+    truncatedByteCount: number;
+  };
+  startTime: string;
+  endTime: string;
+  attributes: {
+    attributeMap: {
+      requestId?: {
+        stringValue: {
+          value: string;
+          truncatedByteCount: number;
+        };
+      };
+      event?: {
+        stringValue: {
+          value: string;
+          truncatedByteCount: 0;
+        };
+      };
+    };
+    droppedAttributesCount: number;
+  };
+};
+
+/**
+ * inspired by open-telemetry/opentelemetry-js
+ */
+export function generateRandUTF16Chars(bytes: number) {
+  for (let i = 0; i < bytes * 2; i++) {
+    BytesBuffer[i] = Math.floor(Math.random() * 16) + 48;
+    // valid hex characters in the range 48-57 and 97-102
+    if (BytesBuffer[i] >= 58) {
+      BytesBuffer[i] += 39;
+    }
+  }
+
+  return String.fromCharCode(...BytesBuffer.slice(0, bytes * 2));
+}
+
+export class TraceReporter {
+  static traceReportEndpoint = process.env.TRACE_REPORT_ENDPOINT;
+  static shouldReportTrace = process.env.SHOULD_REPORT_TRACE;
+
+  private spansCache = new Array<TraceSpan>();
+  private reportIntervalId: number | undefined | NodeJS.Timeout;
+  private reportInterval = 60_000;
+
+  private static instance: TraceReporter;
+
+  public static getInstance(): TraceReporter {
+    if (!TraceReporter.instance) {
+      const instance = (TraceReporter.instance = new TraceReporter());
+      instance.initTraceReport();
+    }
+
+    return TraceReporter.instance;
+  }
+
+  public cacheTrace(
+    traceId: string,
+    spanId: string,
+    startTime: string,
+    attributes: {
+      requestId?: string;
+      event?: string;
+    }
+  ) {
+    const span = TraceReporter.createTraceSpan(
+      traceId,
+      spanId,
+      startTime,
+      attributes
+    );
+    this.spansCache.push(span);
+    if (this.spansCache.length <= 1) {
+      this.initTraceReport();
+    }
+  }
+
+  public uploadTrace(
+    traceId: string,
+    spanId: string,
+    startTime: string,
+    attributes: {
+      requestId?: string;
+      event?: string;
+    }
+  ) {
+    const span = TraceReporter.createTraceSpan(
+      traceId,
+      spanId,
+      startTime,
+      attributes
+    );
+    TraceReporter.reportToTraceEndpoint(JSON.stringify({ spans: [span] }));
+  }
+
+  public static reportToTraceEndpoint(payload: string): void {
+    if (!TraceReporter.traceReportEndpoint) {
+      console.warn('No trace report endpoint found!');
+      return;
+    }
+    if (typeof navigator !== 'undefined') {
+      navigator.sendBeacon(TraceReporter.traceReportEndpoint, payload);
+    } else {
+      fetch(TraceReporter.traceReportEndpoint, {
+        method: 'POST',
+        mode: 'cors',
+        cache: 'no-cache',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: payload,
+      }).catch(console.warn);
+    }
+  }
+
+  public static createTraceSpan(
+    traceId: string,
+    spanId: string,
+    startTime: string,
+    attributes: {
+      requestId?: string;
+      event?: string;
+    }
+  ): TraceSpan {
+    const requestId = attributes.requestId;
+    const event = attributes.event;
+
+    return {
+      name: `projects/{GCP_PROJECT_ID}/traces/${traceId}/spans/${spanId}`,
+      spanId,
+      displayName: {
+        value: 'AFFiNE_REQUEST',
+        truncatedByteCount: 0,
+      },
+      startTime,
+      endTime: new Date().toISOString(),
+      attributes: {
+        attributeMap: {
+          ...(!requestId
+            ? {}
+            : {
+                requestId: {
+                  stringValue: {
+                    value: requestId,
+                    truncatedByteCount: 0,
+                  },
+                },
+              }),
+          ...(!event
+            ? {}
+            : {
+                event: {
+                  stringValue: {
+                    value: event,
+                    truncatedByteCount: 0,
+                  },
+                },
+              }),
+        },
+        droppedAttributesCount: 0,
+      },
+    };
+  }
+
+  private initTraceReport = () => {
+    if (!this.reportIntervalId && TraceReporter.shouldReportTrace) {
+      if (typeof window !== 'undefined') {
+        this.reportIntervalId = window.setInterval(
+          this.reportHandler,
+          this.reportInterval
+        );
+      } else {
+        this.reportIntervalId = setInterval(
+          this.reportHandler,
+          this.reportInterval
+        );
+      }
+    }
+  };
+
+  private reportHandler = () => {
+    if (this.spansCache.length <= 0) {
+      clearInterval(this.reportIntervalId);
+      this.reportIntervalId = undefined;
+      return;
+    }
+    TraceReporter.reportToTraceEndpoint(
+      JSON.stringify({ spans: [...this.spansCache] })
+    );
+    this.spansCache = [];
+  };
+}
+
+export const traceReporter = process.env.SHOULD_REPORT_TRACE
+  ? TraceReporter.getInstance()
+  : null;