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
|
// 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 integration
import (
"context"
"testing"
"github.com/google/uuid"
"github.com/facebook/ent/entc/integration/ent"
"github.com/facebook/ent/entc/integration/ent/pet"
"github.com/facebook/ent/entc/integration/ent/user"
"github.com/facebook/ent/entql"
"github.com/stretchr/testify/require"
)
func EntQL(t *testing.T, client *ent.Client) {
require := require.New(t)
ctx := context.Background()
a8m := client.User.Create().SetName("a8m").SetAge(30).SaveX(ctx)
nati := client.User.Create().SetName("nati").SetAge(30).AddFriends(a8m).SaveX(ctx)
uq := client.User.Query()
uq.Filter().Where(entql.HasEdge("friends"))
require.Equal(2, uq.CountX(ctx))
uq = client.User.Query()
uq.Filter().Where(
entql.And(
entql.FieldEQ("name", "nati"),
entql.HasEdge("friends"),
),
)
require.Equal(nati.ID, uq.OnlyIDX(ctx))
u1, u2 := uuid.New(), uuid.New()
xabi := client.Pet.Create().SetName("xabi").SetOwner(a8m).SetUUID(u1).SaveX(ctx)
luna := client.Pet.Create().SetName("luna").SetOwner(nati).SetUUID(u2).SaveX(ctx)
uq = client.User.Query()
uq.Filter().Where(
entql.And(
entql.HasEdge("pets"),
entql.HasEdgeWith("friends", entql.FieldEQ("name", "nati")),
),
)
require.Equal(a8m.ID, uq.OnlyIDX(ctx))
uq = client.User.Query()
uq.Filter().Where(
entql.And(
entql.HasEdgeWith("pets", entql.FieldEQ("name", "luna")),
entql.HasEdge("friends"),
),
)
require.Equal(nati.ID, uq.OnlyIDX(ctx))
pq := client.Pet.Query()
pq.Filter().WhereUUID(entql.ValueEQ(u1))
require.Equal(xabi.ID, pq.OnlyIDX(ctx))
pq = client.Pet.Query()
pq.Filter().WhereUUID(entql.ValueEQ(u2))
require.Equal(luna.ID, pq.OnlyIDX(ctx))
uq = client.User.Query()
uq.Filter().WhereName(entql.StringEQ("a8m"))
require.Equal(a8m.ID, uq.OnlyIDX(ctx))
pq = client.Pet.Query()
pq.Filter().WhereName(entql.StringOr(entql.StringEQ("xabi"), entql.StringEQ("luna")))
require.Equal([]int{luna.ID, xabi.ID}, pq.Order(ent.Asc(pet.FieldName)).IDsX(ctx))
pq = client.Pet.Query()
pq.Where(pet.Name(luna.Name)).Filter().WhereID(entql.IntEQ(luna.ID))
require.Equal(luna.ID, pq.Order(ent.Asc(pet.FieldName)).OnlyIDX(ctx))
pq = client.Pet.Query()
pq.Where(pet.Name(luna.Name)).Filter().WhereID(entql.IntEQ(xabi.ID))
require.False(pq.ExistX(ctx))
update := client.User.Update().SetRole(user.RoleAdmin)
update.Mutation().Filter().WhereName(entql.StringEQ(a8m.Name))
updated := update.SaveX(ctx)
require.Equal(1, updated)
uq = client.User.Query()
uq.Filter().WhereRole(entql.StringEQ(string(user.RoleAdmin)))
require.Equal(a8m.ID, uq.OnlyIDX(ctx))
}
|
