1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// 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.
package gen
import (
"fmt"
"reflect"
"github.com/facebook/ent/dialect/gremlin/graph/dsl"
"github.com/facebook/ent/dialect/sql"
)
// A SchemaMode defines what type of schema feature a storage driver support.
type SchemaMode uint
const (
// Unique defines field and edge uniqueness support.
Unique SchemaMode = 1 << iota
// Indexes defines indexes support.
Indexes
// Cascade defines cascading operations (e.g. cascade deletion).
Cascade
// Migrate defines static schema and migration support (e.g. SQL-based).
Migrate
)
// Support reports whether m support the given mode.
func (m SchemaMode) Support(mode SchemaMode) bool { return m&mode != 0 }
// Storage driver type for codegen.
type Storage struct {
Name string // storage name.
Builder reflect.Type // query builder type.
Dialects []string // supported dialects.
IdentName string // identifier name (fields and funcs).
Imports []string // import packages needed.
SchemaMode SchemaMode // schema mode support.
Ops func(*Field) []Op // storage specific operations.
OpCode func(Op) string // operation code for predicates.
}
// StorageDrivers holds the storage driver options for entc.
var drivers = []*Storage{
{
Name: "sql",
IdentName: "SQL",
Builder: reflect.TypeOf(&sql.Selector{}),
Dialects: []string{"dialect.SQLite", "dialect.MySQL", "dialect.Postgres"},
Imports: []string{
"github.com/facebook/ent/dialect/sql",
"github.com/facebook/ent/dialect/sql/sqlgraph",
"github.com/facebook/ent/schema/field",
},
SchemaMode: Unique | Indexes | Cascade | Migrate,
Ops: func(f *Field) []Op {
if f.IsString() && f.ConvertedToBasic() {
return []Op{EqualFold, ContainsFold}
}
return nil
},
OpCode: opCodes(sqlCode[:]),
},
{
Name: "gremlin",
IdentName: "Gremlin",
Builder: reflect.TypeOf(&dsl.Traversal{}),
Dialects: []string{"dialect.Gremlin"},
Imports: []string{
"github.com/facebook/ent/dialect/gremlin",
"github.com/facebook/ent/dialect/gremlin/graph/dsl",
"github.com/facebook/ent/dialect/gremlin/graph/dsl/__",
"github.com/facebook/ent/dialect/gremlin/graph/dsl/g",
"github.com/facebook/ent/dialect/gremlin/graph/dsl/p",
"github.com/facebook/ent/dialect/gremlin/encoding/graphson",
},
SchemaMode: Unique,
OpCode: opCodes(gremlinCode[:]),
},
}
// NewStorage returns a the storage driver type from the given string.
// It fails if the provided string is not a valid option. this function
// is here in order to remove the validation logic from entc command line.
func NewStorage(s string) (*Storage, error) {
for _, d := range drivers {
if s == d.Name {
return d, nil
}
}
return nil, fmt.Errorf("entc/gen: invalid storage driver %q", s)
}
// String implements the fmt.Stringer interface for template usage.
func (s *Storage) String() string { return s.Name }
var (
// exceptional operation names in sql.
sqlCode = [...]string{
IsNil: "IsNull",
NotNil: "NotNull",
}
// exceptional operation names in gremlin.
gremlinCode = [...]string{
IsNil: "HasNot",
NotNil: "Has",
In: "Within",
NotIn: "Without",
Contains: "Containing",
HasPrefix: "StartingWith",
HasSuffix: "EndingWith",
}
)
func opCodes(codes []string) func(Op) string {
return func(o Op) string {
if int(o) < len(codes) && codes[o] != "" {
return codes[o]
}
return o.Name()
}
}
|
