feat: add index on state

sev-2 · Oct 25, 2024 · 3e9f8c3 · 3e9f8c3
1 parent bfbb805
commit 3e9f8c3
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/pkg/generator/model.go b/pkg/generator/model.go
@@ -275,7 +275,7 @@ func BuildRelationTag(r *state.Relation) string {
 	tags = append(tags, jsonTag)
 
 	if r.Index != nil {
-		tags = append(tags, "indexed")
+		tags = append(tags, "indexed:true")
 	}
 
 	if r.Action != nil {

diff --git a/pkg/state/table.go b/pkg/state/table.go
@@ -172,6 +172,17 @@ func buildTableFromModel(model any) (ei ExtractTableItem) {
 						}
 					}
 
+					// check index field
+					indexed := field.Tag.Get("indexed")
+					if len(indexed) > 0 {
+						rel.Index = &objects.Index{
+							Schema:     ei.Table.Schema,
+							Table:      rel.TargetTableName,
+							Name:       fmt.Sprintf("ix_%s_%s", rel.TargetTableName, rel.TargetColumnName),
+							Definition: fmt.Sprintf("CREATE INDEX ix_%s_%s ON %s(%s);", rel.TargetTableName, rel.TargetColumnName, rel.TargetTableName, rel.TargetColumnName),
+						}
+					}
+
 					ei.Table.Relationships = append(ei.Table.Relationships, rel)
 				}
 			}

diff --git a/pkg/state/table_test.go b/pkg/state/table_test.go
@@ -24,7 +24,7 @@ type Submission struct {
 	Acl string `json:"-" read:"anon" write:"anon"`
 
 	// Relations
-	Candidate *Candidate `json:"candidate,omitempty" join:"joinType:hasOne;primaryKey:id;foreignKey:candidate_id"  onUpdate:"cascade" onDelete:"cascade"`
+	Candidate *Candidate `json:"candidate,omitempty" join:"joinType:hasOne;primaryKey:id;foreignKey:candidate_id"  onUpdate:"cascade" onDelete:"cascade" indexed:"true"`
 }
 
 type Candidate struct {