add sqlbuilder

Author: free1139

api.go

@@ -148,6 +148,16 @@ func QueryPageMapContext(queryer Queryer, ctx context.Context, querySql string,
 	return queryPageMap(queryer, ctx, querySql, args...)
 }
 
-func StmtWhereIn(drvName string, paramStartIdx, paramsLen int) string {
-	return stmtWhereIn(drvName, paramStartIdx, paramsLen)
+// Extend stmt for the where in
+//
+// Example for the first input:
+// fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args))
+// Or
+// fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args), qsql.DRV_NAME_MYSQL)
+//
+// Example for the second input:
+// fmt.Sprintf("select * from table_name where id=? in (%s)", qsql.StmtWhereIn(1,len(args))
+
+func StmtIn(paramStartIdx, paramsLen int, drvName ...string) string {
+	return stmtIn(paramStartIdx, paramsLen, drvName...)
 }

db.go

@@ -107,8 +107,10 @@ func (db *DB) QueryPageMap(querySql string, args ...interface{}) (titles []strin
 func (db *DB) QueryPageMapContext(ctx context.Context, querySql string, args ...interface{}) (titles []string, result []map[string]interface{}, err error) {
 	return queryPageMap(db, ctx, querySql, args...)
 }
-func (db *DB) StmtWhereIn(paramStartIdx, paramsLen int) string {
-	return stmtWhereIn(db.DriverName(), paramStartIdx, paramsLen)
+
+// Return "?,?,?,?..." for default, or "@p1,@p2,@p3..." for mssql, or ":1,:2,:3..." for pgsql when paramStartIdx is 0.
+func (db *DB) StmtIn(paramStartIdx, paramsLen int) string {
+	return stmtIn(paramStartIdx, paramsLen, db.DriverName())
 }
 
 // A lazy function to commit the *sql.Tx

sqlbuilder.go

@@ -0,0 +1,35 @@
+package qsql
+
+type SqlBuilder struct {
+	table          []string
+	selectStr      []string
+	whereAndStr    []string
+	whereParams    []interface{}
+	whereOrStr     []string
+	whereOrParams  []interface{}
+	groupByStr     []string
+	havingStr      []string
+	havingParams   []interface{}
+	havingOrStr    []string
+	havingOrParams []interface{}
+	orderByStr     []string
+	offset         int
+	limit          int
+}
+
+func NewSqlBuilder(table string) *SqlBuilder {
+	return &SqlBuilder{
+		table: []string{table},
+	}
+}
+
+func (b *SqlBuilder) Joins(table string) *SqlBuilder {
+	b.table = append(b.table, table)
+	return b
+}
+
+func (b *SqlBuilder) Where(cond string, args ...interface{}) *SqlBuilder {
+	b.whereAndStr = append(b.whereAndStr, cond)
+	b.whereParams = append(b.whereParams, args...)
+	return b
+}

where.go

@@ -5,17 +5,11 @@ import (
 	"reflect"
 )
 
-// Extend the where in stmt
-//
-// Example for the first input:
-// fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args))
-// Or
-// fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args), qsql.DRV_NAME_MYSQL)
-//
-// Example for the second input:
-// fmt.Sprintf("select * from table_name where id=? in (%s)", qsql.StmtWhereIn(1,len(args))
-//
-func stmtWhereIn(driverName string, paramIdx, paramsLen int) string {
+func stmtIn(paramIdx, paramsLen int, driverNames ...string) string {
+	driverName := ""
+	if len(driverNames) > 0 {
+		driverName = driverNames[0]
+	}
 	drvName := getDrvName(nil, driverName)
 	switch drvName {
 	case DRV_NAME_ORACLE, _DRV_NAME_OCI8: