golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen/template: allow mutating the sqlgraph.CreateSpec by ent extensions or user templates

Browse Source
This commit is contained in:
Ariel Mashraki
2021-01-19 09:34:40 +02:00
committed by Ariel Mashraki
parent aa73b9067f
commit 63eb68a269
3 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
entc/gen/internal/bindata.go
View File

File diff suppressed because one or more lines are too long

1
entc/gen/template.go
View File

@@ -167,6 +167,7 @@ var (
partialPatterns = [...]string{
"dialect/additional/*",
"dialect/*/import/additional/*",
"dialect/*/*/spec/*",
}
// templates holds the Go templates for the code generation.
templates *Template

6
entc/gen/template/dialect/sql/create.tmpl
View File

@@ -43,6 +43,12 @@ func ({{ $receiver }} *{{ $builder }}) createSpec() (*{{ $.Name }}, *sqlgraph.Cr
},
}
)
{{- /* Allow mutating the sqlgraph.CreateSpec by ent extensions or user templates.*/}}
{{- with $tmpls := matchTemplate "dialect/sql/create/spec/*" }}
{{- range $tmpl := $tmpls }}
{{- xtemplate $tmpl $ }}
{{- end }}
{{- end }}
{{- if $.ID.UserDefined }}
if id, ok := {{ $mutation }}.{{ $.ID.MutationGet }}(); ok {
_node.ID = id