mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
95 lines
3.0 KiB
Cheetah
95 lines
3.0 KiB
Cheetah
{{/*
|
|
Copyright 2019-present Facebook Inc. All rights reserved.
|
|
This source code is licensed under the Apache 2.0 license found
|
|
in the LICENSE file in the root directory of this source tree.
|
|
*/}}
|
|
|
|
{{/* gotype: entgo.io/ent/entc/gen.typeScope */}}
|
|
|
|
{{ define "dialect/gremlin/create" }}
|
|
{{ $builder := pascal $.Scope.Builder }}
|
|
{{ $receiver := $.Scope.Receiver }}
|
|
{{ $mutation := print $receiver ".mutation" }}
|
|
|
|
func ({{ $receiver }} *{{ $builder }}) gremlinSave(ctx context.Context) (*{{ $.Name }}, error) {
|
|
if err := {{ $receiver }}.check(); err != nil {
|
|
return nil, err
|
|
}
|
|
res := &gremlin.Response{}
|
|
query, bindings := {{ $receiver }}.gremlin().Query()
|
|
if err := {{ $receiver }}.driver.Exec(ctx, query, bindings, res); err != nil {
|
|
return nil, err
|
|
}
|
|
if err, ok := isConstantError(res); ok {
|
|
return nil, err
|
|
}
|
|
rnode := &{{ $.Name }}{config: {{ $receiver }}.config}
|
|
if err := rnode.FromResponse(res); err != nil {
|
|
return nil, err
|
|
}
|
|
{{- if $.HasOneFieldID }}
|
|
{{ $mutation }}.{{ $.ID.BuilderField }} = &rnode.{{ $.ID.StructField }}
|
|
{{ $mutation }}.done = true
|
|
{{- end }}
|
|
return rnode, nil
|
|
}
|
|
|
|
func ({{ $receiver }} *{{ $builder }}) gremlin() *dsl.Traversal {
|
|
{{- with .NumConstraint }}
|
|
type constraint struct {
|
|
pred *dsl.Traversal // constraint predicate.
|
|
test *dsl.Traversal // test matches and its constant.
|
|
}
|
|
constraints := make([]*constraint, 0, {{ . }})
|
|
{{- end }}
|
|
v := g.AddV({{ $.Package }}.Label)
|
|
{{- if $.ID.UserDefined }}
|
|
if id, ok := {{ $mutation }}.{{ $.ID.MutationGet }}(); ok {
|
|
v.Property(dsl.ID, id)
|
|
}
|
|
{{- end }}
|
|
{{- range $f := $.MutationFields }}
|
|
if value, ok := {{ $mutation }}.{{ $f.MutationGet }}(); ok {
|
|
{{- if $f.Unique }}
|
|
constraints = append(constraints, &constraint{
|
|
pred: g.V().Has({{ $.Package }}.Label, {{ $.Package }}.{{ $f.Constant }}, value).Count(),
|
|
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField({{ $.Package }}.Label, {{ $.Package }}.{{ $f.Constant }}, value)),
|
|
})
|
|
{{- end }}
|
|
v.Property(dsl.Single, {{ $.Package }}.{{ $f.Constant }}, value)
|
|
}
|
|
{{- end }}
|
|
{{- range $e := $.Edges }}
|
|
{{- $direction := "In" }}
|
|
{{- $name := printf "%s.%s" $.Package $e.LabelConstant }}
|
|
for _, id := range {{ $mutation }}.{{ $e.StructField }}IDs() {
|
|
{{- if $e.IsInverse }}
|
|
{{- $direction = "Out" }}
|
|
{{- $name = printf "%s.%s" $e.Type.Package $e.LabelConstant }}
|
|
v.AddE({{ $name }}).From(g.V(id)).InV()
|
|
{{- else }}
|
|
v.AddE({{ $name }}).To(g.V(id)).OutV()
|
|
{{- end }}
|
|
{{- if $e.HasConstraint }}
|
|
constraints = append(constraints, &constraint{
|
|
pred: g.E().HasLabel({{ $name }}).{{ $direction }}V().HasID(id).Count(),
|
|
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueEdge({{ $.Package }}.Label, {{ $name }}, id)),
|
|
})
|
|
{{- end }}
|
|
}
|
|
{{- end }}
|
|
{{- with .NumConstraint }}
|
|
if len(constraints) == 0 {
|
|
return v.ValueMap(true)
|
|
}
|
|
tr := constraints[0].pred.Coalesce(constraints[0].test, v.ValueMap(true))
|
|
for _, cr := range constraints[1:] {
|
|
tr = cr.pred.Coalesce(cr.test, tr)
|
|
}
|
|
return tr
|
|
{{- else }}
|
|
return v.ValueMap(true)
|
|
{{- end }}
|
|
}
|
|
{{ end }}
|