fix id template and generate cmd

This commit is contained in:
Hayden 2022-12-05 13:29:00 -09:00
parent 8d456eaef5
commit 7cc3ad6cea
No known key found for this signature in database
GPG key ID: 17CF79474E257545
2 changed files with 7 additions and 2 deletions

View file

@ -1,3 +1,3 @@
package ent
//go:generate go run -mod=mod entgo.io/ent/cmd/ent generate --feature sql/versioned-migration ./schema
//go:generate go run -mod=mod entgo.io/ent/cmd/ent generate --feature sql/versioned-migration ./schema --template=./schema/templates/has_id.tmpl

View file

@ -9,8 +9,13 @@
import "github.com/google/uuid"
{{/* Loop over all nodes and implement the "HasID" interface */}}
{{ range $n := $.Nodes }}
{{ if not $n.ID }}
{{/* If the node doesn't have an ID field, we skip it. */}}
{{ continue }}
{{ end }}
{{/* The "HasID" interface is implemented by the "ID" method. */}}
{{ $receiver := $n.Receiver }}
func ({{ $receiver }} *{{ $n.Name }}) GetID() uuid.UUID {
func ({{ $receiver }} *{{ $n.Name }}) GetID() {{ $n.ID.Type }} {
return {{ $receiver }}.ID
}
{{ end }}