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

Merge 1537cf3d60 into 33fc33b2f5

This commit is contained in:
jiangyanfeng 2019-10-08 13:10:57 +00:00 committed by GitHub
parent 33fc33b2f5 1537cf3d60
commit 0d33408950
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
session_stats.go
View File

@ -30,6 +30,10 @@ func (session *Session) Count(bean ...interface{}) (int64, error) {
args = session.statement.RawParams
}
if len(session.statement.selectStr) > 0 {
sqlStr = "SELECT COUNT(*) FROM ("+sqlStr+") _TEMP_"
}
var total int64
err = session.queryRow(sqlStr, args...).Scan(&total)
if err == sql.ErrNoRows || err == nil {

4
session_stats_test.go
View File

@ -182,6 +182,10 @@ func TestCount(t *testing.T) {
assert.NoError(t, err)
assert.EqualValues(t, 1, cnt)
total, err = testEngine.Where(cond).Select(colName).Count(new(UserinfoCount))
assert.NoError(t, err)
assert.EqualValues(t, 1, total)
total, err = testEngine.Where(cond).Count(new(UserinfoCount))
assert.NoError(t, err)
assert.EqualValues(t, 1, total)