HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复statement GenExistSQL 只有表名，没有RefTable时，未拼接别名Alias的BUG

This commit is contained in:
hsfish 2025-02-19 18:31:43 +08:00
parent 7654b7b749
commit 0f6fb5409d
2 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
internal/statements/query.go
View File

@ -318,7 +318,10 @@ func (statement *Statement) GenExistSQL(bean ...interface{}) (string, []interfac
buf := builder.NewWriter() buf := builder.NewWriter()
if statement.dialect.URI().DBType == schemas.MSSQL { if statement.dialect.URI().DBType == schemas.MSSQL {
if _, err := fmt.Fprintf(buf, "SELECT TOP 1 * FROM %s", tableName); err != nil { if _, err := fmt.Fprintf(buf, "SELECT TOP 1 *"); err != nil {
return "", nil, err
}
if err := statement.writeFrom(buf); err != nil {
return "", nil, err return "", nil, err
} }
if err := statement.writeJoins(buf); err != nil { if err := statement.writeJoins(buf); err != nil {
@ -328,7 +331,10 @@ func (statement *Statement) GenExistSQL(bean ...interface{}) (string, []interfac
return "", nil, err return "", nil, err
} }
} else if statement.dialect.URI().DBType == schemas.ORACLE { } else if statement.dialect.URI().DBType == schemas.ORACLE {
if _, err := fmt.Fprintf(buf, "SELECT * FROM %s", tableName); err != nil { if _, err := fmt.Fprintf(buf, "SELECT *"); err != nil {
return "", nil, err
}
if err := statement.writeFrom(buf); err != nil {
return "", nil, err return "", nil, err
} }
if err := statement.writeJoins(buf); err != nil { if err := statement.writeJoins(buf); err != nil {
@ -349,7 +355,10 @@ func (statement *Statement) GenExistSQL(bean ...interface{}) (string, []interfac
return "", nil, err return "", nil, err
} }
} else { } else {
if _, err := fmt.Fprintf(buf, "SELECT 1 FROM %s", tableName); err != nil { if _, err := fmt.Fprintf(buf, "SELECT 1"); err != nil {
return "", nil, err
}
if err := statement.writeFrom(buf); err != nil {
return "", nil, err return "", nil, err
} }
if err := statement.writeJoins(buf); err != nil { if err := statement.writeJoins(buf); err != nil {

19
internal/statements/query_test.go Normal file
View File

@ -0,0 +1,19 @@
package statements
import (
"github.com/stretchr/testify/assert"
"testing"
)
func TestGenExistSQL(t *testing.T) {
statement, err := createTestStatement()
assert.NoError(t, err)
statement.RefTable = nil
statement.SetTable("testDB")
statement.Alias("tdb")
statement.Where(`tdb.id=1`)
sql, _, err := statement.GenExistSQL()
assert.NoError(t, err)
assert.Equal(t, "SELECT 1 FROM `testDB` AS `tdb` WHERE (tdb.id=1) LIMIT 1", sql)
}