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 }}