fix: index, multiple migrations

Author: mrkaye97

cmd/hatchet-migrate/migrate/migrations/20260428204451_v1_0_103.sql

@@ -1,9 +1,9 @@
 -- +goose Up
 -- +goose StatementBegin
-ALTER TABLE "Worker" ADD COLUMN "actionsHash" BYTEA;
+ALTER TABLE "Worker" ADD COLUMN "actionHash" BYTEA;
 -- +goose StatementEnd
 
 -- +goose Down
 -- +goose StatementBegin
-ALTER TABLE "Worker" DROP COLUMN "actionsHash";
+ALTER TABLE "Worker" DROP COLUMN "actionHash";
 -- +goose StatementEnd

cmd/hatchet-migrate/migrate/migrations/20260428213451_v1_0_104.sql

@@ -0,0 +1,11 @@
+-- +goose no transaction
+
+-- +goose Up
+-- +goose StatementBegin
+CREATE INDEX CONCURRENTLY IF NOT EXISTS "Worker_tenantId_actionHash_idx" ON "Worker" ("tenantId", "actionHash");
+-- +goose StatementEnd
+
+-- +goose Down
+-- +goose StatementBegin
+DROP INDEX CONCURRENTLY IF EXISTS "Worker_tenantId_actionHash_idx";
+-- +goose StatementEnd

pkg/repository/sqlcv1/models.go

@@ -435,7 +435,7 @@ INSERT INTO "Worker" (
     "languageVersion",
     "os",
     "runtimeExtra",
-    "actionsHash"
+    "actionHash"
 ) VALUES (
     gen_random_uuid(),
     CURRENT_TIMESTAMP,
@@ -449,7 +449,7 @@ INSERT INTO "Worker" (
     sqlc.narg('languageVersion')::text,
     sqlc.narg('os')::text,
     sqlc.narg('runtimeExtra')::text,
-    @actionsHash::bytea
+    @actionHash::bytea
 ) RETURNING *;
 
 -- name: LinkServicesToWorker :exec

pkg/repository/sqlcv1/workers.sql

@@ -89,7 +89,7 @@ type WorkerRepository interface {
 	ListActiveSDKsPerTenant(ctx context.Context) (map[TenantIdSDKTuple]int64, error)
 
 	// GetWorkerActionsByWorkerId returns a list of actions for a worker
-	GetWorkerActionsByWorkerId(ctx context.Context, tenantId uuid.UUID, workerId []uuid.UUID) (map[string][]string, error)
+	GetWorkerActionsForWorkers(ctx context.Context, tenantId uuid.UUID, workers []sqlcv1.Worker) (map[string][]string, error)
 
 	// GetWorkerWorkflowsByWorkerId returns a list of workflows for a worker
 	GetWorkerWorkflowsByWorkerId(ctx context.Context, tenantId uuid.UUID, workerId uuid.UUID) ([]*sqlcv1.Workflow, error)
@@ -276,7 +276,13 @@ func (w *workerRepository) CountActiveWorkersPerTenant(ctx context.Context) (map
 	return tenantToWorkers, nil
 }
 
-func (w *workerRepository) GetWorkerActionsByWorkerId(ctx context.Context, tenantId uuid.UUID, workerIds []uuid.UUID) (map[string][]string, error) {
+func (w *workerRepository) GetWorkerActionsForWorkers(ctx context.Context, tenantId uuid.UUID, workers []sqlcv1.Worker) (map[string][]string, error) {
+	actionHashToWorkerId := make(map[uuid.UUID][]byte)
+
+	for _, worker := range workers {
+		actionHashToWorkerId[worker.ID] = worker.ActionsHash
+	}
+
 	records, err := w.queries.GetWorkerActionsByWorkerId(ctx, w.pool, sqlcv1.GetWorkerActionsByWorkerIdParams{
 		Workerids: workerIds,
 		Tenantid:  tenantId,

pkg/repository/sqlcv1/workers.sql.go

@@ -864,7 +864,7 @@ CREATE TABLE "Worker" (
     "runtimeExtra" TEXT,
     "sdkVersion" TEXT,
     "durableTaskDispatcherId" UUID,
-    "actionsHash" BYTEA,
+    "actionHash" BYTEA,
 
     CONSTRAINT "Worker_pkey" PRIMARY KEY ("id")
 );
@@ -1442,6 +1442,9 @@ CREATE UNIQUE INDEX "Worker_webhookId_key" ON "Worker" ("webhookId" ASC);
 
 CREATE INDEX "Worker_tenantId_lastHeartbeatAt_idx" ON "Worker" ("tenantId", "lastHeartbeatAt");
 
+-- CreateIndex
+CREATE INDEX "Worker_tenantId_actionHash_idx" ON "Worker" ("tenantId", "actionHash");
+
 -- CreateIndex
 CREATE INDEX "WorkerAssignEvent_workerId_id_idx" ON "WorkerAssignEvent" ("workerId" ASC, "id" ASC);