github.com/iasthc/atlas/cmd/atlas@v0.0.0-20230523071841-73246df3f88d/internal/migrate/ent/template/convert.tmpl (about) 1 {{/* gotype: entgo.io/ent/entc/gen.Graph */}} 2 3 {{ define "convert" }} 4 5 {{ $pkg := base $.Config.Package }} 6 {{ template "header" $ }} 7 8 import "github.com/iasthc/atlas/sql/migrate" 9 10 {{ range $n := $.Nodes }} 11 {{ if eq $n.Name "Revision" }} 12 {{ $builder := $n.CreateName }} 13 {{ $receiver := receiver $builder }} 14 15 // SetRevision takes the values for each field from the given migrate.Revision. 16 func ({{ $receiver }} *{{ $builder }}) SetRevision(rev *migrate.Revision) *{{ $builder }} { 17 {{ $receiver }}.SetID(rev.Version) 18 {{- range $f := $n.Fields }} 19 {{ $receiver }}.Set{{ $f.StructField }}(rev.{{ $f.StructField }}) 20 {{- end }} 21 return {{ $receiver }} 22 } 23 24 // AtlasRevision returns an migrate.Revision from the current Revision. 25 func({{ $n.Receiver}} *Revision) AtlasRevision() *migrate.Revision { 26 return &migrate.Revision{ 27 Version: r.ID, 28 {{- range $f := $n.Fields }} 29 {{ $f.StructField }}: r.{{ $f.StructField }}, 30 {{- end }} 31 } 32 } 33 {{ end }} 34 {{ end }} 35 36 {{ end }}