add migrations for the updated spans table with project id

frontend/lib/db/migrations/0002_next_tusk.sql → frontend/lib/db/migrations/0002_cold_romulus.sql

@@ -2,8 +2,8 @@ CREATE TABLE IF NOT EXISTS "labeling_queue_data" (
 	"id" uuid PRIMARY KEY DEFAULT gen_random_uuid() NOT NULL,
 	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
 	"queue_id" uuid DEFAULT gen_random_uuid() NOT NULL,
-	"data" jsonb NOT NULL,
-	"action" jsonb NOT NULL
+	"action" jsonb NOT NULL,
+	"span_id" uuid NOT NULL
 );
 --> statement-breakpoint
 ALTER TABLE "dataset_datapoints" ALTER COLUMN "target" DROP NOT NULL;--> statement-breakpoint

frontend/lib/db/migrations/0003_cultured_tinkerer.sql

@@ -0,0 +1,29 @@
+ALTER TABLE "labels" DROP CONSTRAINT "trace_tags_span_id_fkey";
+--> statement-breakpoint
+/* 
+    [SOLVED]
+    Unfortunately in current drizzle-kit version we can't automatically get name for primary key.
+    We are working on making it available!
+
+    Meanwhile you can:
+        1. Check pk name in your database, by running
+            SELECT constraint_name FROM information_schema.table_constraints
+            WHERE table_schema = 'public'
+                AND table_name = 'spans'
+                AND constraint_type = 'PRIMARY KEY';
+        2. Uncomment code below and paste pk name manually
+
+    Hope to release this update as soon as possible
+*/
+
+ALTER TABLE "spans" DROP CONSTRAINT "spans_pkey";--> statement-breakpoint
+ALTER TABLE "spans" ADD COLUMN "project_id" uuid NOT NULL;--> statement-breakpoint
+ALTER TABLE "spans" ADD CONSTRAINT "spans_pkey" PRIMARY KEY("span_id","project_id");--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "spans" ADD CONSTRAINT "spans_project_id_fkey" FOREIGN KEY ("project_id") REFERENCES "public"."projects"("id") ON DELETE no action ON UPDATE cascade;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;
+--> statement-breakpoint
+CREATE INDEX IF NOT EXISTS "spans_project_id_idx" ON "spans" USING btree ("project_id");--> statement-breakpoint
+ALTER TABLE "spans" ADD CONSTRAINT "unique_span_id_project_id" UNIQUE("span_id","project_id");

frontend/lib/db/migrations/meta/0002_snapshot.json

@@ -1,5 +1,5 @@
 {
-  "id": "4bf43179-5f48-4c3e-b1f0-033007c3a0c9",
+  "id": "326ba2ca-6925-4d69-8416-1a7691a8bcfd",
   "prevId": "a7f4ca0c-430b-49f5-93d4-09d1ab968dac",
   "version": "7",
   "dialect": "postgresql",
@@ -742,15 +742,15 @@
           "notNull": true,
           "default": "gen_random_uuid()"
         },
-        "data": {
-          "name": "data",
+        "action": {
+          "name": "action",
           "type": "jsonb",
           "primaryKey": false,
           "notNull": true
         },
-        "action": {
-          "name": "action",
-          "type": "jsonb",
+        "span_id": {
+          "name": "span_id",
+          "type": "uuid",
           "primaryKey": false,
           "notNull": true
         }