feat: introduce indexed tag

sev-2 · Oct 25, 2024 · bfbb805 · bfbb805
1 parent 14155f8
commit bfbb805
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pkg/generator/model.go b/pkg/generator/model.go
@@ -274,6 +274,10 @@ func BuildRelationTag(r *state.Relation) string {
 	jsonTag := fmt.Sprintf("json:%q", utils.ToSnakeCase(r.Table)+",omitempty")
 	tags = append(tags, jsonTag)
 
+	if r.Index != nil {
+		tags = append(tags, "indexed")
+	}
+
 	if r.Action != nil {
 
 		onUpdate, onDelete := objects.RelationActionDefaultLabel, objects.RelationActionDefaultLabel

diff --git a/pkg/generator/model_test.go b/pkg/generator/model_test.go
@@ -48,6 +48,7 @@ func TestGenerateModels(t *testing.T) {
 					ForeignKey:   "test_table_id",
 					PrimaryKey:   "id",
 					Action:       &relationshipAction,
+					Index:        &objects.Index{Schema: "public", Table: "related_table", Name: "test_table_id", Definition: "CREATE INDEX test_table_id ON related_table(test_table_id);"},
 				},
 			},
 			Policies: objects.Policies{},