golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-28 09:49:08 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen: remove multi storage support

Browse Source
This commit is contained in:
Ariel Mashraki
2019-12-13 17:20:06 +02:00
parent 1311f5f2f2
commit e85b10be36
226 changed files with 35018 additions and 15853 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
entc/integration/ent/comment_create.go
View File

@@ -11,12 +11,6 @@ import (
"errors"
"strconv"
"github.com/facebookincubator/ent/dialect"
"github.com/facebookincubator/ent/dialect/gremlin"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/__"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/g"
"github.com/facebookincubator/ent/dialect/gremlin/graph/dsl/p"
"github.com/facebookincubator/ent/dialect/sql"
"github.com/facebookincubator/ent/entc/integration/ent/comment"
)
@@ -63,14 +57,7 @@ func (cc *CommentCreate) Save(ctx context.Context) (*Comment, error) {
if cc.unique_float == nil {
return nil, errors.New("ent: missing required field \"unique_float\"")
}
switch cc.driver.Dialect() {
case dialect.MySQL, dialect.Postgres, dialect.SQLite:
return cc.sqlSave(ctx)
case dialect.Gremlin:
return cc.gremlinSave(ctx)
default:
return nil, errors.New("ent: unsupported dialect")
}
return cc.sqlSave(ctx)
}
// SaveX calls Save and panics if Save returns an error.
@@ -115,53 +102,3 @@ func (cc *CommentCreate) sqlSave(ctx context.Context) (*Comment, error) {
}
return c, nil
}
func (cc *CommentCreate) gremlinSave(ctx context.Context) (*Comment, error) {
res := &gremlin.Response{}
query, bindings := cc.gremlin().Query()
if err := cc.driver.Exec(ctx, query, bindings, res); err != nil {
return nil, err
}
if err, ok := isConstantError(res); ok {
return nil, err
}
c := &Comment{config: cc.config}
if err := c.FromResponse(res); err != nil {
return nil, err
}
return c, nil
}
func (cc *CommentCreate) gremlin() *dsl.Traversal {
type constraint struct {
pred *dsl.Traversal // constraint predicate.
test *dsl.Traversal // test matches and its constant.
}
constraints := make([]*constraint, 0, 2)
v := g.AddV(comment.Label)
if cc.unique_int != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(comment.Label, comment.FieldUniqueInt, *cc.unique_int).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(comment.Label, comment.FieldUniqueInt, *cc.unique_int)),
})
v.Property(dsl.Single, comment.FieldUniqueInt, *cc.unique_int)
}
if cc.unique_float != nil {
constraints = append(constraints, &constraint{
pred: g.V().Has(comment.Label, comment.FieldUniqueFloat, *cc.unique_float).Count(),
test: __.Is(p.NEQ(0)).Constant(NewErrUniqueField(comment.Label, comment.FieldUniqueFloat, *cc.unique_float)),
})
v.Property(dsl.Single, comment.FieldUniqueFloat, *cc.unique_float)
}
if cc.nillable_int != nil {
v.Property(dsl.Single, comment.FieldNillableInt, *cc.nillable_int)
}
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
}