feat!: affine cloud support (#3813)

Co-authored-by: Hongtao Lye <codert.sn@gmail.com>
Co-authored-by: liuyi <forehalo@gmail.com>
Co-authored-by: LongYinan <lynweklm@gmail.com>
Co-authored-by: X1a0t <405028157@qq.com>
Co-authored-by: JimmFly <yangjinfei001@gmail.com>
Co-authored-by: Peng Xiao <pengxiao@outlook.com>
Co-authored-by: xiaodong zuo <53252747+zuoxiaodong0815@users.noreply.github.com>
Co-authored-by: DarkSky <25152247+darkskygit@users.noreply.github.com>
Co-authored-by: Qi <474021214@qq.com>
Co-authored-by: danielchim <kahungchim@gmail.com>

packages/graphql/src/__tests__/fetcher.spec.ts

@@ -1,8 +1,15 @@
+import { nanoid } from 'nanoid';
 import type { Mock } from 'vitest';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import { gqlFetcherFactory } from '../fetcher';
 import type { GraphQLQuery } from '../graphql';
+import {
+  generateRandUTF16Chars,
+  SPAN_ID_BYTES,
+  TRACE_ID_BYTES,
+  TraceReporter,
+} from '../utils';
 
 const query: GraphQLQuery = {
   id: 'query',
@@ -51,16 +58,18 @@ describe('GraphQL fetcher', () => {
       variables: { a: 1, b: '2', c: { d: false } },
     });
 
-    expect(fetch.mock.lastCall[1]).toMatchInlineSnapshot(`
-      {
-        "body": "{\\"query\\":\\"query { field }\\",\\"variables\\":{\\"a\\":1,\\"b\\":\\"2\\",\\"c\\":{\\"d\\":false}},\\"operationName\\":\\"query\\"}",
-        "headers": {
-          "x-definition-name": "query",
-          "x-operation-name": "query",
-        },
-        "method": "POST",
-      }
-    `);
+    expect(fetch.mock.lastCall[1]).toEqual(
+      expect.objectContaining({
+        body: '{"query":"query { field }","variables":{"a":1,"b":"2","c":{"d":false}},"operationName":"query"}',
+        headers: expect.objectContaining({
+          'content-type': 'application/json',
+          'x-definition-name': 'query',
+          'x-operation-name': 'query',
+          'x-request-id': expect.any(String),
+        }),
+        method: 'POST',
+      })
+    );
   });
 
   it('should correctly ignore nil variables', async () => {
@@ -110,3 +119,41 @@ describe('GraphQL fetcher', () => {
     `);
   });
 });
+
+describe('Trace Reporter', () => {
+  const startTime = new Date().toISOString();
+  const traceId = generateRandUTF16Chars(TRACE_ID_BYTES);
+  const spanId = generateRandUTF16Chars(SPAN_ID_BYTES);
+  const requestId = nanoid();
+
+  it('spanId, traceId should be right format', () => {
+    expect(
+      new RegExp(`^[0-9a-f]{${SPAN_ID_BYTES * 2}}$`).test(
+        generateRandUTF16Chars(SPAN_ID_BYTES)
+      )
+    ).toBe(true);
+    expect(
+      new RegExp(`^[0-9a-f]{${TRACE_ID_BYTES * 2}}$`).test(
+        generateRandUTF16Chars(TRACE_ID_BYTES)
+      )
+    ).toBe(true);
+  });
+
+  it('test createTraceSpan', () => {
+    const traceSpan = TraceReporter.createTraceSpan(
+      traceId,
+      spanId,
+      requestId,
+      startTime
+    );
+    expect(traceSpan.startTime).toBe(startTime);
+    expect(
+      traceSpan.name ===
+        `projects/{GCP_PROJECT_ID}/traces/${traceId}/spans/${spanId}`
+    ).toBe(true);
+    expect(traceSpan.spanId).toBe(spanId);
+    expect(traceSpan.attributes.attributeMap.requestId.stringValue.value).toBe(
+      requestId
+    );
+  });
+});