Files
ent/entc/gen/predicate.go
Alex Snast 373769dfaf ent/gen: adding EqualFold predicate for string fields
Summary: Similar to `ContainsFold` predicate requires `--storage=sql` argument to be passed to entc.

Reviewed By: a8m

Differential Revision: D17074805

fbshipit-source-id: ced299154417fe2c9007cd6a7a504f53c8b2ef98
2019-08-27 10:48:53 -07:00

72 lines
1.9 KiB
Go

// 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
// Op is a predicate for the where clause.
type Op int
// List of all builtin predicates.
const (
EQ Op = iota // =
NEQ // <>
GT // >
GTE // >=
LT // <
LTE // <=
IsNil // IS NULL / has
NotNil // IS NOT NULL / hasNot
In // within
NotIn // without
EqualFold // equals case-insensitive
Contains // containing
ContainsFold // containing case-insensitive
HasPrefix // startingWith
HasSuffix // endingWith
)
// Name returns the string representation of an predicate.
func (o Op) Name() string {
if int(o) < len(opText) {
return opText[o]
}
return "Unknown"
}
// Variadic reports if the predicate is a variadic function.
func (o Op) Variadic() bool {
return o == In || o == NotIn
}
// Niladic reports if the predicate is a niladic predicate.
func (o Op) Niladic() bool {
return o == IsNil || o == NotNil
}
var (
// operations text.
opText = [...]string{
EQ: "EQ",
NEQ: "NEQ",
GT: "GT",
GTE: "GTE",
LT: "LT",
LTE: "LTE",
IsNil: "IsNil",
NotNil: "NotNil",
EqualFold: "EqualFold",
Contains: "Contains",
ContainsFold: "ContainsFold",
HasPrefix: "HasPrefix",
HasSuffix: "HasSuffix",
In: "In",
NotIn: "NotIn",
}
// operations per type.
boolOps = []Op{EQ, NEQ}
numericOps = append(boolOps[:], GT, GTE, LT, LTE, In, NotIn)
stringOps = append(numericOps[:], Contains, HasPrefix, HasSuffix)
nillableOps = []Op{IsNil, NotNil}
)