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: configure contains-fold per dialect (#666)

Browse Source
This commit is contained in:
Ariel Mashraki
2020-08-13 17:26:19 +03:00
committed by GitHub
parent 07c38e8e8f
commit d9c8130d08
2 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
dialect/sql/builder.go
View File

@@ -1138,13 +1138,20 @@ func (p *Predicate) Contains(col, sub string) *Predicate {
func ContainsFold(col, sub string) *Predicate { return (&Predicate{}).ContainsFold(col, sub) }
// ContainsFold is a helper predicate that applies the LIKE predicate with case-folding.
// The recommendation is to avoid using it, and to use a dialect specific feature, like
// `ILIKE` in PostgreSQL, and `COLLATE` clause in MySQL.
func (p *Predicate) ContainsFold(col, sub string) *Predicate {
return p.append(func(b *Builder) {
f := &Func{}
f.SetDialect(b.dialect)
b.Ident(f.Lower(col)).WriteString(" LIKE ")
switch b.dialect {
case dialect.MySQL:
// We assume the CHARACTER SET is configured to utf8mb4,
// because this how it is defined in dialect/sql/schema.
b.Ident(col).WriteString(" COLLATE utf8mb4_general_ci LIKE ")
case dialect.Postgres:
b.Ident(col).WriteString(" ILIKE ")
default: // SQLite.
b.Ident(f.Lower(col)).WriteString(" LIKE ")
}
b.Arg("%" + strings.ToLower(sub) + "%")
})
}

13
dialect/sql/builder_test.go
View File

@@ -802,7 +802,8 @@ func TestBuilder(t *testing.T) {
wantArgs: []interface{}{"bar", "baz"},
},
{
input: Select().
input: Dialect(dialect.SQLite).
Select().
From(Table("users")).
Where(ContainsFold("name", "Ariel").And().ContainsFold("nick", "Bar")),
wantQuery: "SELECT * FROM `users` WHERE LOWER(`name`) LIKE ? AND LOWER(`nick`) LIKE ?",
@@ -813,7 +814,15 @@ func TestBuilder(t *testing.T) {
Select().
From(Table("users")).
Where(ContainsFold("name", "Ariel").And().ContainsFold("nick", "Bar")),
wantQuery: `SELECT * FROM "users" WHERE LOWER("name") LIKE $1 AND LOWER("nick") LIKE $2`,
wantQuery: `SELECT * FROM "users" WHERE "name" ILIKE $1 AND "nick" ILIKE $2`,
wantArgs: []interface{}{"%ariel%", "%bar%"},
},
{
input: Dialect(dialect.MySQL).
Select().
From(Table("users")).
Where(ContainsFold("name", "Ariel").And().ContainsFold("nick", "Bar")),
wantQuery: "SELECT * FROM `users` WHERE `name` COLLATE utf8mb4_general_ci LIKE ? AND `nick` COLLATE utf8mb4_general_ci LIKE ?",
wantArgs: []interface{}{"%ariel%", "%bar%"},
},
{