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

dialect/sql/sqlgraph: add function to order by edge count (#3422)

Browse Source
This commit is contained in:
Ariel Mashraki
2023-03-30 10:10:59 +03:00
committed by GitHub
parent 651a2a166e
commit e3cee0adc2
6 changed files with 331 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
dialect/sql/builder.go
View File

@@ -3330,12 +3330,12 @@ func (b *Builder) Quote(ident string) string {
func (b *Builder) Ident(s string) *Builder {
switch {
case len(s) == 0:
case !strings.HasSuffix(s, "*") && !b.isIdent(s) && !isFunc(s) && !isModifier(s):
case !strings.HasSuffix(s, "*") && !b.isIdent(s) && !isFunc(s) && !isModifier(s) && !isAlias(s):
if b.qualifier != "" {
b.WriteString(b.Quote(b.qualifier)).WriteByte('.')
}
b.WriteString(b.Quote(s))
case (isFunc(s) || isModifier(s)) && b.postgres():
case (isFunc(s) || isModifier(s) || isAlias(s)) && b.postgres():
// Modifiers and aggregation functions that
// were called without dialect information.
b.WriteString(strings.ReplaceAll(s, "`", `"`))
@@ -3432,8 +3432,8 @@ func (b *Builder) Err() error {
// An Op represents an operator.
type Op int
// Predicate and arithmetic operators.
const (
// Predicate operators.
OpEQ Op = iota // =
OpNEQ // <>
OpGT // >
@@ -3445,13 +3445,11 @@ const (
OpLike // LIKE
OpIsNull // IS NULL
OpNotNull // IS NOT NULL
// Arithmetic operators.
OpAdd // +
OpSub // -
OpMul // *
OpDiv // / (Quotient)
OpMod // % (Reminder)
OpAdd // +
OpSub // -
OpMul // *
OpDiv // / (Quotient)
OpMod // % (Reminder)
)
var ops = [...]string{
@@ -3713,6 +3711,19 @@ func Dialect(name string) *DialectBuilder {
return &DialectBuilder{name}
}
// String builds a dialect-aware expression string from the given callback.
func (d *DialectBuilder) String(f func(*Builder)) string {
b := &Builder{}
b.SetDialect(d.dialect)
f(b)
return b.String()
}
// Expr builds a dialect-aware expression from the given callback.
func (d *DialectBuilder) Expr(f func(*Builder)) Querier {
return Expr(d.String(f))
}
// Describe creates a DescribeBuilder for the configured dialect.
//
// Dialect(dialect.Postgres).
@@ -3870,6 +3881,10 @@ func (d *DialectBuilder) DropIndex(name string) *DropIndexBuilder {
return b
}
func isAlias(s string) bool {
return strings.Contains(s, " AS ") || strings.Contains(s, " as ")
}
func isFunc(s string) bool {
return strings.Contains(s, "(") && strings.Contains(s, ")")
}