fixed Join strings on func Exist
This commit is contained in:
parent
062d9960b2
commit
c3f05d347b
|
@ -25,8 +25,8 @@ func (session *Session) Exist(bean ...interface{}) (bool, error) {
|
||||||
|
|
||||||
var sqlStr string
|
var sqlStr string
|
||||||
var args []interface{}
|
var args []interface{}
|
||||||
|
var joinStr string
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
if session.statement.RawSQL == "" {
|
if session.statement.RawSQL == "" {
|
||||||
if len(bean) == 0 {
|
if len(bean) == 0 {
|
||||||
tableName := session.statement.TableName()
|
tableName := session.statement.TableName()
|
||||||
|
@ -35,6 +35,9 @@ func (session *Session) Exist(bean ...interface{}) (bool, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
tableName = session.statement.Engine.Quote(tableName)
|
tableName = session.statement.Engine.Quote(tableName)
|
||||||
|
if len(session.statement.JoinStr) > 0 {
|
||||||
|
joinStr = session.statement.JoinStr
|
||||||
|
}
|
||||||
|
|
||||||
if session.statement.cond.IsValid() {
|
if session.statement.cond.IsValid() {
|
||||||
condSQL, condArgs, err := builder.ToSQL(session.statement.cond)
|
condSQL, condArgs, err := builder.ToSQL(session.statement.cond)
|
||||||
|
@ -43,20 +46,20 @@ func (session *Session) Exist(bean ...interface{}) (bool, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if session.engine.dialect.DBType() == core.MSSQL {
|
if session.engine.dialect.DBType() == core.MSSQL {
|
||||||
sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s WHERE %s", tableName, condSQL)
|
sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s %s WHERE %s", tableName, joinStr, condSQL)
|
||||||
} else if session.engine.dialect.DBType() == core.ORACLE {
|
} else if session.engine.dialect.DBType() == core.ORACLE {
|
||||||
sqlStr = fmt.Sprintf("SELECT * FROM %s WHERE (%s) AND ROWNUM=1", tableName, condSQL)
|
sqlStr = fmt.Sprintf("SELECT * FROM %s WHERE (%s) %s AND ROWNUM=1", tableName, joinStr, condSQL)
|
||||||
} else {
|
} else {
|
||||||
sqlStr = fmt.Sprintf("SELECT * FROM %s WHERE %s LIMIT 1", tableName, condSQL)
|
sqlStr = fmt.Sprintf("SELECT * FROM %s %s WHERE %s LIMIT 1", tableName, joinStr, condSQL)
|
||||||
}
|
}
|
||||||
args = condArgs
|
args = condArgs
|
||||||
} else {
|
} else {
|
||||||
if session.engine.dialect.DBType() == core.MSSQL {
|
if session.engine.dialect.DBType() == core.MSSQL {
|
||||||
sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s", tableName)
|
sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s %s", tableName, joinStr)
|
||||||
} else if session.engine.dialect.DBType() == core.ORACLE {
|
} else if session.engine.dialect.DBType() == core.ORACLE {
|
||||||
sqlStr = fmt.Sprintf("SELECT * FROM %s WHERE ROWNUM=1", tableName)
|
sqlStr = fmt.Sprintf("SELECT * FROM %s %s WHERE ROWNUM=1", tableName, joinStr)
|
||||||
} else {
|
} else {
|
||||||
sqlStr = fmt.Sprintf("SELECT * FROM %s LIMIT 1", tableName)
|
sqlStr = fmt.Sprintf("SELECT * FROM %s %s LIMIT 1", tableName, joinStr)
|
||||||
}
|
}
|
||||||
args = []interface{}{}
|
args = []interface{}{}
|
||||||
}
|
}
|
||||||
|
|
|
@ -74,3 +74,106 @@ func TestExistStruct(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.False(t, has)
|
assert.False(t, has)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestExistStructForJoin(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
|
||||||
|
type Salary struct {
|
||||||
|
Id int64
|
||||||
|
Lid int64
|
||||||
|
}
|
||||||
|
|
||||||
|
type CheckList struct {
|
||||||
|
Id int64
|
||||||
|
Eid int64
|
||||||
|
}
|
||||||
|
|
||||||
|
type Empsetting struct {
|
||||||
|
Id int64
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(t, testEngine.Sync2(new(Salary), new(CheckList), new(Empsetting)))
|
||||||
|
|
||||||
|
var emp Empsetting
|
||||||
|
cnt, err := testEngine.Insert(&emp)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
var checklist = CheckList{
|
||||||
|
Eid: emp.Id,
|
||||||
|
}
|
||||||
|
cnt, err = testEngine.Insert(&checklist)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
var salary = Salary{
|
||||||
|
Lid: checklist.Id,
|
||||||
|
}
|
||||||
|
cnt, err = testEngine.Insert(&salary)
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.EqualValues(t, 1, cnt)
|
||||||
|
|
||||||
|
session := testEngine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid").
|
||||||
|
Where("salary.lid = ?", 1)
|
||||||
|
has, err := session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid").
|
||||||
|
Where("salary.lid = ?", 2)
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.False(t, has)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Select("check_list.id").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid").
|
||||||
|
Where("check_list.id = ?", 1)
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Select("empsetting.id").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid").
|
||||||
|
Where("empsetting.id = ?", 2)
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.False(t, has)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Select("empsetting.id").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid")
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
|
||||||
|
err = session.DropTable("check_list")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Select("empsetting.id").
|
||||||
|
Join("INNER", "check_list", "check_list.id = salary.lid").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = check_list.eid")
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.Error(t, err)
|
||||||
|
assert.False(t, has)
|
||||||
|
|
||||||
|
session.Table("salary").
|
||||||
|
Select("empsetting.id").
|
||||||
|
Join("LEFT", "empsetting", "empsetting.id = salary.lid")
|
||||||
|
has, err = session.Exist()
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.True(t, has)
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue