docs: 添加对 count(case ...) 语句的说明

Author: caixw

docs/sqlbuilder.md

@@ -240,4 +240,8 @@ list := make([]*User, 0, 10);
 // 查询符合以上条件的所有记录数
 builder.Count("count(*) AS cnt")
     count, err := builder.QueryInt("cnt")
+
+// 也支持 case 作为 count 内容
+builder.Count("count(CASE WHEN age>18 THEN age ELSE NULL END) AS cnt")
+	count, err := builder.QueryInt("cnt")
 ```

fetch/object.go

@@ -60,7 +60,7 @@ func convertError(field string, message error) error {
 //	    Count int `orm:"-"`         // 不会匹配与该字段对应的列。
 //	}
 //
-// 第一个参数用于表示有多少数据被正确导入到 obj 中
+// 第一个返回参数用于表示有多少数据被正确导入到 obj 中
 func Object(strict bool, rows *sql.Rows, obj any) (int, error) {
 	val := reflect.ValueOf(obj)
 
@@ -183,7 +183,7 @@ func getColumns(v reflect.Value, cols []string) ([]any, error) {
 }
 
 // 将 rows 中的一条记录写入到 val 中，必须保证 val 的类型为 reflect.Struct。
-// 仅供 Obj() 调用。
+// 仅供 Object() 调用。
 func fetchOnceObj(strict bool, val reflect.Value, rows *sql.Rows) (int, error) {
 	if !strict {
 		return fetchOnceObjNoStrict(val, rows)

go.mod

@@ -9,7 +9,7 @@ require (
 	github.com/lib/pq v1.10.9
 	github.com/mattn/go-sqlite3 v1.14.24
 	github.com/shopspring/decimal v1.4.0
-	modernc.org/sqlite v1.33.1
+	modernc.org/sqlite v1.34.4
 )
 
 require (

go.sum

@@ -59,8 +59,8 @@ modernc.org/opt v0.1.3 h1:3XOZf2yznlhC+ibLltsDGzABUGVx8J6pnFMS3E4dcq4=
 modernc.org/opt v0.1.3/go.mod h1:WdSiB5evDcignE70guQKxYUl14mgWtbClRi5wmkkTX0=
 modernc.org/sortutil v1.2.0 h1:jQiD3PfS2REGJNzNCMMaLSp/wdMNieTbKX920Cqdgqc=
 modernc.org/sortutil v1.2.0/go.mod h1:TKU2s7kJMf1AE84OoiGppNHJwvB753OYfNl2WRb++Ss=
-modernc.org/sqlite v1.33.1 h1:trb6Z3YYoeM9eDL1O8do81kP+0ejv+YzgyFo+Gwy0nM=
-modernc.org/sqlite v1.33.1/go.mod h1:pXV2xHxhzXZsgT/RtTFAPY6JJDEvOTcTdwADQCCWD4k=
+modernc.org/sqlite v1.34.4 h1:sjdARozcL5KJBvYQvLlZEmctRgW9xqIZc2ncN7PU0P8=
+modernc.org/sqlite v1.34.4/go.mod h1:3QQFCG2SEMtc2nv+Wq4cQCH7Hjcg+p/RMlS1XK+zwbk=
 modernc.org/strutil v1.2.0 h1:agBi9dp1I+eOnxXeiZawM8F4LawKv4NzGWSaLfyeNZA=
 modernc.org/strutil v1.2.0/go.mod h1:/mdcBmfOibveCTBxUl5B5l6W+TTH1FXPLHZE6bTosX0=
 modernc.org/token v1.1.0 h1:Xl7Ap9dKaEs5kLoOQeQmPWevfnk/DM5qcLcYlA8ys6Y=

sqlbuilder/select.go

@@ -401,6 +401,10 @@ func (stmt *SelectStmt) Limit(limit any, offset ...any) *SelectStmt {
 //
 // 如果指定了 count 表达式，则会造成 limit 失效，
 // 如果设置为空值，则取消 count，恢复普通的 select 。
+//
+//	stmt := NewSelectStmt()
+//	stmt.Count("count(*) as cnt")
+//	stmt.Count("count(CASE WHEN xx) as cnt1, count(CASE WHEN yy) as cnt2")
 func (stmt *SelectStmt) Count(expr string) *SelectStmt {
 	stmt.countExpr = expr
 

sqlbuilder/select_test.go

@@ -31,12 +31,10 @@ func TestSelect(t *testing.T) {
 			Desc("id")
 
 		id, err := stmt.QueryInt("id")
-		a.NotError(err).
-			Equal(id, 4)
+		a.NotError(err).Equal(id, 4)
 
 		f, err := stmt.QueryFloat("id")
-		a.NotError(err).
-			Equal(f, 4.0)
+		a.NotError(err).Equal(f, 4.0)
 
 		// 不存在的列
 		f, err = stmt.QueryFloat("id_not_exists")
@@ -57,6 +55,17 @@ func TestSelect(t *testing.T) {
 		cnt, err := stmt.Count("count(*) as cnt").QueryInt("cnt")
 		a.NotError(err).Equal(cnt, 4)
 
+		type counter struct {
+			Cnt1 int `orm:"name(cnt1)"`
+			Cnt2 int `orm:"name(cnt2)"`
+		}
+		cntr := &counter{}
+		size, err = stmt.Count("COUNT(CASE WHEN id>3 THEN id END) AS cnt1, COUNT(CASE WHEN id<3 THEN id END) AS cnt2").
+			QueryObject(true, cntr)
+		a.NotError(err).Equal(size, 1).
+			Equal(cntr.Cnt1, 1). // stmt 中的条件是 id<5，所以只有一个符合条件的数据
+			Equal(cntr.Cnt2, 2)
+
 		// 没有符合条件的数据
 		stmt.Reset()
 		stmt.Column("*").