entc/gen: generate sql builders with dialect option

Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/130

Reviewed By: alexsn

Differential Revision: D18164397

fbshipit-source-id: 2858d69d3ff85c06b51382c01c3d4369ee2c3bdb
This commit is contained in:
Ariel Mashraki
2019-10-27 21:52:31 -07:00
committed by Facebook Github Bot
parent ea479ea527
commit c259aee24b
241 changed files with 3899 additions and 2394 deletions

View File

@@ -57,15 +57,17 @@ func (gq *GroupQuery) Order(o ...Order) *GroupQuery {
// QueryUsers chains the current query on the users edge.
func (gq *GroupQuery) QueryUsers() *UserQuery {
query := &UserQuery{config: gq.config}
t1 := sql.Table(user.Table)
builder := sql.Dialect(gq.driver.Dialect())
t1 := builder.Table(user.Table)
t2 := gq.sqlQuery()
t2.Select(t2.C(group.FieldID))
t3 := sql.Table(group.UsersTable)
t4 := sql.Select(t3.C(group.UsersPrimaryKey[1])).
t3 := builder.Table(group.UsersTable)
t4 := builder.Select(t3.C(group.UsersPrimaryKey[1])).
From(t3).
Join(t2).
On(t3.C(group.UsersPrimaryKey[0]), t2.C(group.FieldID))
query.sql = sql.Select().
query.sql = builder.Select().
From(t1).
Join(t4).
On(t1.C(user.FieldID), t4.C(group.UsersPrimaryKey[1]))
@@ -333,8 +335,9 @@ func (gq *GroupQuery) sqlExist(ctx context.Context) (bool, error) {
}
func (gq *GroupQuery) sqlQuery() *sql.Selector {
t1 := sql.Table(group.Table)
selector := sql.Select(t1.Columns(group.Columns...)...).From(t1)
builder := sql.Dialect(gq.driver.Dialect())
t1 := builder.Table(group.Table)
selector := builder.Select(t1.Columns(group.Columns...)...).From(t1)
if gq.sql != nil {
selector = gq.sql
selector.Select(selector.Columns(group.Columns...)...)
@@ -603,5 +606,6 @@ func (gs *GroupSelect) sqlScan(ctx context.Context, v interface{}) error {
func (gs *GroupSelect) sqlQuery() sql.Querier {
view := "group_view"
return sql.Select(gs.fields...).From(gs.sql.As(view))
return sql.Dialect(gs.driver.Dialect()).
Select(gs.fields...).From(gs.sql.As(view))
}