Merge pull request #130 from stephenafamo/fix-setter-if-relationships

Generate setter if a table has relationships
stephenafamo · Nov 18, 2023 · 68600c9 · 68600c9
2 parents e3353da + 12381f4
commit 68600c9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/gen/templates/models/01_types.go.tpl b/gen/templates/models/01_types.go.tpl
@@ -67,7 +67,7 @@ type {{$tAlias.DownSingular}}R struct {
 }
 {{- end}}
 
-{{if .Table.PKey -}}
+{{if or .Table.PKey .Table.Relationships -}}
 // {{$tAlias.UpSingular}}Setter is used for insert/upsert/update operations
 // All values are optional, and do not have to be set
 // Generated columns are not included
@@ -223,4 +223,3 @@ func {{$tAlias.UpSingular}}Where[Q {{$.Dialect}}.Filterable]() {{$tAlias.DownSin
 			{{end -}}
 	}
 }
-
diff --git a/gen/templates/models/11_rel_ops.go.tpl b/gen/templates/models/11_rel_ops.go.tpl
@@ -1,3 +1,4 @@
+{{if .Table.PKey -}}
 {{$table := .Table}}
 {{$tAlias := .Aliases.Table $table.Key -}}
 
@@ -320,3 +321,4 @@
 
 {{end -}}{{end -}}
 
+{{- end}}