Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-16 01:26:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: auth metric and trace (#4063)

Browse Source
This commit is contained in:
X1a0t
2023-09-06 12:20:06 +08:00
committed by GitHub
parent d29514c995
commit ef3d3a34e2
9 changed files with 143 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/graphql/src/utils.ts
+50 -18
View File
@@ -16,12 +16,18 @@ type TraceSpan = {
endTime: string;
attributes: {
attributeMap: {
requestId: {
requestId?: {
stringValue: {
value: string;
truncatedByteCount: number;
};
};
event?: {
stringValue: {
value: string;
truncatedByteCount: 0;
};
};
};
droppedAttributesCount: number;
};
@@ -65,14 +71,17 @@ export class TraceReporter {
public cacheTrace(
traceId: string,
spanId: string,
requestId: string,
startTime: string
startTime: string,
attributes: {
requestId?: string;
event?: string;
}
) {
const span = TraceReporter.createTraceSpan(
traceId,
spanId,
requestId,
startTime
startTime,
attributes
);
this.spansCache.push(span);
if (this.spansCache.length <= 1) {
@@ -83,14 +92,17 @@ export class TraceReporter {
public uploadTrace(
traceId: string,
spanId: string,
requestId: string,
startTime: string
startTime: string,
attributes: {
requestId?: string;
event?: string;
}
) {
const span = TraceReporter.createTraceSpan(
traceId,
spanId,
requestId,
startTime
startTime,
attributes
);
TraceReporter.reportToTraceEndpoint(JSON.stringify({ spans: [span] }));
}
@@ -114,26 +126,46 @@ export class TraceReporter {
public static createTraceSpan(
traceId: string,
spanId: string,
requestId: string,
startTime: 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: 'fetch',
value: 'AFFiNE_REQUEST',
truncatedByteCount: 0,
},
startTime,
endTime: new Date().toISOString(),
attributes: {
attributeMap: {
requestId: {
stringValue: {
value: requestId,
truncatedByteCount: 0,
},
},
...(!requestId
? {}
: {
requestId: {
stringValue: {
value: requestId,
truncatedByteCount: 0,
},
},
}),
...(!event
? {}
: {
event: {
stringValue: {
value: event,
truncatedByteCount: 0,
},
},
}),
},
droppedAttributesCount: 0,
},