mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
cef32f1fb4
Summary:
Set the standard header ("Code generated by entc, DO NOT EDIT.") as default, and override it using option in graph.
No changes to graph except the `generate.go` file.
Reviewed By: idoshveki
Differential Revision: D16642348
fbshipit-source-id: d9fd1d2046e2fd96acbb100ef061fda75d99ce52
34 lines
967 B
Cheetah
34 lines
967 B
Cheetah
{{ define "predicate" }}
|
|
|
|
{{- with extend $ "Package" "predicate" -}}
|
|
{{ template "header" . }}
|
|
{{ end }}
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"fbc/ent/dialect/gremlin/graph/dsl"
|
|
"fbc/ent/dialect/sql"
|
|
)
|
|
|
|
{{ range $_, $n := $.Nodes -}}
|
|
// {{ $n.Name }} is the predicate function for {{ $n.Package }} builders.
|
|
type {{ $n.Name }} func({{ if gt (len $.Storage) 1 }}interface{}{{ else }}{{ (index $.Storage 0).Builder }}{{ end }})
|
|
|
|
{{ if gt (len $.Storage) 1 }}
|
|
// {{ $n.Name }}PerDialect construct a predicate for graph traversals based on dialect type.
|
|
func {{ $n.Name }}PerDialect({{ range $i, $storage := $.Storage }}{{ if $i }},{{ end }}f{{ $i }} func({{ $storage.Builder }}){{ end }}) {{ $n.Name }} {
|
|
return {{ $n.Name }}(func (v interface{}) {
|
|
switch v := v.(type) {
|
|
{{- range $i, $storage := $.Storage }}
|
|
case {{ $storage.Builder }}:
|
|
f{{ $i }}(v)
|
|
{{- end }}
|
|
default:
|
|
panic(fmt.Sprintf("unknown type for predicate: %T", v))
|
|
}
|
|
})
|
|
}
|
|
{{ end }}
|
|
{{ end }}
|
|
{{ end }} |