feature: tag[0] 기반 파티셔닝 전략 도입

Author: toothlessdev

packages/patchlogr-core/src/partition/__tests__/partitionByTag.test.ts

@@ -0,0 +1,69 @@
+import type { CanonicalSpec } from "@patchlogr/types";
+import { describe, expect, test } from "vitest";
+import { partitionByTag } from "../partitionByTag";
+
+describe("partitionByTag", () => {
+    test("should group by first tag", () => {
+        const spec: CanonicalSpec = {
+            operations: {
+                "GET /user": {
+                    key: "GET /user",
+                    doc: { tags: ["user"] },
+                    method: "GET",
+                    path: "/user",
+                    request: { params: [] },
+                    responses: {},
+                },
+                "GET /user/{userId}": {
+                    key: "GET /user/{userId}",
+                    doc: { tags: ["user"] },
+                    method: "GET",
+                    path: "/user/{userId}",
+                    request: { params: [] },
+                    responses: {},
+                },
+            },
+        };
+
+        const partitions = partitionByTag(spec).partitions;
+        expect(partitions).toHaveLength(1);
+        expect(partitions.get("user")).toHaveLength(2);
+        expect(partitions.get("user")?.[0]?.operationKey).toBe("GET /user");
+        expect(partitions.get("user")?.[1]?.operationKey).toBe(
+            "GET /user/{userId}",
+        );
+    });
+
+    test("should group by multiple tags", () => {
+        const spec: CanonicalSpec = {
+            operations: {
+                "GET /user": {
+                    key: "GET /user",
+                    doc: { tags: ["user"] },
+                    method: "GET",
+                    path: "/user",
+                    request: { params: [] },
+                    responses: {},
+                },
+                "POST /auth/login": {
+                    key: "POST /auth/login",
+                    doc: { tags: ["auth"] },
+                    method: "POST",
+                    path: "/auth/login",
+                    request: { params: [] },
+                    responses: {},
+                },
+            },
+        };
+
+        const partitions = partitionByTag(spec).partitions;
+
+        expect(partitions).toHaveLength(2);
+        expect(partitions.get("user")).toHaveLength(1);
+        expect(partitions.get("auth")).toHaveLength(1);
+        expect(partitions.get("user")?.[0]?.operationKey).toBe("GET /user");
+        expect(partitions.get("auth")?.[0]?.operationKey).toBe(
+            "POST /auth/login",
+        );
+    });
+});

packages/patchlogr-core/src/partition/partition.ts

@@ -0,0 +1,14 @@
+export type PartitionManifest = {
+    key: string;
+    hash: string;
+};
+
+export type Partition = {
+    hash: string;
+    operationKey: string;
+};
+
+export type PartitionedSpec = {
+    metadata: Record<string, unknown>;
+    partitions: Map<string, Partition[]>;
+};

packages/patchlogr-core/src/partition/partitionByTag.ts

@@ -0,0 +1,28 @@
+import type { CanonicalSpec } from "@patchlogr/types";
+import type { Partition, PartitionedSpec } from "./partition";
+
+import { createSHA256Hash } from "../utils/createHash";
+import { stableStringify } from "../utils/stableStringify";
+
+const DEFAULT_TAG = "__DEFAULT__";
+
+export function partitionByTag(spec: CanonicalSpec): PartitionedSpec {
+    const partitions = new Map<string, Partition[]>();
+
+    Object.entries(spec.operations).forEach(([key, operation]) => {
+        const tag = operation.doc?.tags?.[0] || DEFAULT_TAG;
+        const hash = createSHA256Hash(stableStringify(operation));
+
+        if (!partitions.has(tag))
+            partitions.set(tag, [{ hash, operationKey: key }]);
+        else partitions.get(tag)?.push({ hash, operationKey: key });
+    });
+
+    return {
+        metadata: {
+            ...spec.info,
+            ...spec.security,
+        },
+        partitions,
+    };
+}