chore: update indexes for workflow runs pagination (#1609)

Signed-off-by: Miguel Martinez <[email protected]>

Author: migmartri

app/controlplane/pkg/data/ent/migrate/migrations/20241129164322.sql

@@ -0,0 +1,12 @@
+-- Drop index "workflowrun_created_at_id" from table: "workflow_runs"
+DROP INDEX "workflowrun_created_at_id";
+-- Drop index "workflowrun_created_at_state" from table: "workflow_runs"
+DROP INDEX "workflowrun_created_at_state";
+-- Drop index "workflowrun_created_at_workflow_id" from table: "workflow_runs"
+DROP INDEX "workflowrun_created_at_workflow_id";
+-- Create index "workflowrun_created_at_id" to table: "workflow_runs"
+CREATE INDEX "workflowrun_created_at_id" ON "workflow_runs" ("created_at" DESC, "id");
+-- Create index "workflowrun_created_at_state" to table: "workflow_runs"
+CREATE INDEX "workflowrun_created_at_state" ON "workflow_runs" ("created_at" DESC, "state");
+-- Create index "workflowrun_created_at_workflow_id" to table: "workflow_runs"
+CREATE INDEX "workflowrun_created_at_workflow_id" ON "workflow_runs" ("created_at" DESC, "workflow_id");

app/controlplane/pkg/data/ent/migrate/migrations/20241129170353.sql

@@ -0,0 +1,2 @@
+-- Create index "workflow_organization_id" to table: "workflows"
+CREATE INDEX "workflow_organization_id" ON "workflows" ("organization_id") WHERE (deleted_at IS NULL);

app/controlplane/pkg/data/ent/migrate/migrations/atlas.sum

@@ -1,4 +1,4 @@
-h1:cd7peHu3MdqEo1c6vGrDGeDjSLGEFrPqrrEiSMDZJp0=
+h1:Igi2AdQrKbr6iehJdtxkal0vZYJOo0Z7OpmB0/T9rk0=
 20230706165452_init-schema.sql h1:VvqbNFEQnCvUVyj2iDYVQQxDM0+sSXqocpt/5H64k8M=
 20230710111950-cas-backend.sql h1:A8iBuSzZIEbdsv9ipBtscZQuaBp3V5/VMw7eZH6GX+g=
 20230712094107-cas-backends-workflow-runs.sql h1:a5rzxpVGyd56nLRSsKrmCFc9sebg65RWzLghKHh5xvI=
@@ -69,3 +69,5 @@ h1:cd7peHu3MdqEo1c6vGrDGeDjSLGEFrPqrrEiSMDZJp0=
 20241123163942.sql h1:s+ImKf84B5s3Hp9Q7ILTcbNkQFIveZapu8fSe3X0eLY=
 20241126164629.sql h1:TrYF9F2RmBUVVB/9a3XFlintq94b2/BaJLYjv80TXOQ=
 20241126182209.sql h1:01eeRbQxKYsjhEvafp1fL4kYJx4NoidwjcxWbWLlNak=
+20241129164322.sql h1:QYuxTg1P//afuTEzzopH/D6mexMm3UVuIIwlaxCWGOA=
+20241129170353.sql h1:9RmXI0seY855RHiJx1+tfd3iwf9R5HR+3EsWmZdH6nw=

app/controlplane/pkg/data/ent/migrate/schema.go

@@ -471,6 +471,14 @@ var (
 					Where: "deleted_at IS NULL",
 				},
 			},
+			{
+				Name:    "workflow_organization_id",
+				Unique:  false,
+				Columns: []*schema.Column{WorkflowsColumns[10]},
+				Annotation: &entsql.IndexAnnotation{
+					Where: "deleted_at IS NULL",
+				},
+			},
 			{
 				Name:    "workflow_workflow_contract",
 				Unique:  false,
@@ -593,11 +601,21 @@ var (
 				Name:    "workflowrun_created_at_id",
 				Unique:  false,
 				Columns: []*schema.Column{WorkflowRunsColumns[1], WorkflowRunsColumns[0]},
+				Annotation: &entsql.IndexAnnotation{
+					DescColumns: map[string]bool{
+						WorkflowRunsColumns[1].Name: true,
+					},
+				},
 			},
 			{
 				Name:    "workflowrun_created_at_state",
 				Unique:  false,
 				Columns: []*schema.Column{WorkflowRunsColumns[1], WorkflowRunsColumns[3]},
+				Annotation: &entsql.IndexAnnotation{
+					DescColumns: map[string]bool{
+						WorkflowRunsColumns[1].Name: true,
+					},
+				},
 			},
 			{
 				Name:    "workflowrun_attestation_digest",
@@ -613,6 +631,11 @@ var (
 				Name:    "workflowrun_created_at_workflow_id",
 				Unique:  false,
 				Columns: []*schema.Column{WorkflowRunsColumns[1], WorkflowRunsColumns[13]},
+				Annotation: &entsql.IndexAnnotation{
+					DescColumns: map[string]bool{
+						WorkflowRunsColumns[1].Name: true,
+					},
+				},
 			},
 		},
 	}

app/controlplane/pkg/data/ent/schema/workflow.go

@@ -86,6 +86,9 @@ func (Workflow) Indexes() []ent.Index {
 		index.Fields("organization_id", "id").Unique().Annotations(
 			entsql.IndexWhere("deleted_at IS NULL"),
 		),
+		index.Fields("organization_id").Annotations(
+			entsql.IndexWhere("deleted_at IS NULL"),
+		),
 		index.Edges("contract").Annotations(
 			entsql.IndexWhere("deleted_at IS NULL"),
 		),

app/controlplane/pkg/data/ent/schema/workflowrun.go

@@ -79,12 +79,12 @@ func (WorkflowRun) Edges() []ent.Edge {
 func (WorkflowRun) Indexes() []ent.Index {
 	return []ent.Index{
 		// Cursor pagination support
-		index.Fields("created_at", "id"),
+		index.Fields("created_at", "id").Annotations(entsql.DescColumns("created_at")),
 		// Expiration job
-		index.Fields("created_at", "state"),
+		index.Fields("created_at", "state").Annotations(entsql.DescColumns("created_at")),
 		index.Fields("attestation_digest"),
 		index.Edges("workflow"),
 		// DailyRun counts
-		index.Edges("workflow").Fields("created_at"),
+		index.Edges("workflow").Fields("created_at").Annotations(entsql.DescColumns("created_at")),
 	}
 }