add StmtWhereIn

Author: free1139

README.md

@@ -181,6 +181,16 @@ if err := qsql.QueryElem(mdb, &count, "SELECT count(*) FROM a WHERE id = ?", id)
     // ...
 }
 ```
+## Extend the where in stmt
+// Example for the first input:
+mdb := db.GetCache("master") 
+args:=[]int{1,2,3}
+mdb.Query(fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args))))
+// Or
+mdb.Query(fmt.Sprintf("select * from table_name where in (%s)", qsql.StmtWhereIn(0,len(args), qsql.DRV_NAME_MYSQL))
+
+// Example for the second input:
+mdb.Query(fmt.Sprintf("select * from table_name where id=? in (%s)", qsql.StmtWhereIn(1,len(args)))
 
 ## Mass query.
 ```text

api.go

@@ -6,6 +6,7 @@ package qsql
 import (
 	"context"
 	"database/sql"
+	"fmt"
 	"io"
 	"runtime/debug"
 
@@ -19,6 +20,9 @@ const (
 	DRV_NAME_POSTGRES  = "postgres"
 	DRV_NAME_SQLITE3   = "sqlite3"
 	DRV_NAME_SQLSERVER = "sqlserver" // or "mssql"
+
+	_DRV_NAME_OCI8  = "oci8"
+	_DRV_NAME_MSSQL = "mssql"
 )
 
 var (
@@ -31,6 +35,71 @@ var (
 	REFLECT_DRV_NAME = DRV_NAME_MYSQL
 )
 
+func getDrvName(exec Execer, driverName ...string) string {
+	drvName := REFLECT_DRV_NAME
+	db, ok := exec.(*DB)
+	if ok {
+		drvName = db.DriverName()
+	} else {
+		drvNamesLen := len(driverName)
+		if drvNamesLen > 0 {
+			if drvNamesLen != 1 {
+				panic(errors.New("'drvName' expect only one argument").As(driverName))
+			}
+			drvName = driverName[0]
+		}
+	}
+	return drvName
+}
+
+// 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(paramIdx, paramsLen int, driverName ...string) string {
+	drvName := getDrvName(nil, driverName...)
+	switch drvName {
+	case DRV_NAME_ORACLE, _DRV_NAME_OCI8:
+		// *outputInputs = append(*outputInputs, []byte(fmt.Sprintf(":%s,", f.Name))...)
+		panic("unknow how to implemented")
+	case DRV_NAME_POSTGRES:
+		result := []byte{}
+		for i := 0; i < paramsLen; i++ {
+			result = append(result, []byte(fmt.Sprintf(":%d,", paramIdx+i))...)
+		}
+		if len(result) > 0 {
+			return string(result[:len(result)-1]) // remove the last ','
+		}
+		return string(result)
+	case DRV_NAME_SQLSERVER, _DRV_NAME_MSSQL:
+		result := []byte{}
+		for i := 0; i < paramsLen; i++ {
+			result = append(result, []byte(fmt.Sprintf("@p%d,", paramIdx+i))...)
+		}
+		if len(result) > 0 {
+			return string(result[:len(result)-1]) // remove the last ','
+		}
+		return string(result)
+	default:
+		resultLen := paramsLen * 2
+		result := make([]byte, resultLen)
+		for i := 0; i < resultLen; i += 2 {
+			result[i] = '?'
+			result[i+1] = ','
+		}
+		if len(result) > 0 {
+			return string(result[:len(result)-1]) // remove the last ','
+		}
+		return string(result)
+	}
+}
+
 type Execer interface {
 	Exec(query string, args ...interface{}) (sql.Result, error)
 	ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error)
@@ -129,11 +198,11 @@ func ExecMultiTxContext(tx *sql.Tx, ctx context.Context, mTx []*MultiTx) error {
 
 // Reflect one db data to the struct. the struct tag format like `db:"field_title"`, reference to: http://github.com/jmoiron/sqlx
 // When you no set the REFLECT_DRV_NAME, you can point out with the drvName
-func InsertStruct(exec Execer, obj interface{}, tbName string, drvNames ...string) (sql.Result, error) {
-	return insertStruct(exec, context.TODO(), obj, tbName, drvNames...)
+func InsertStruct(exec Execer, obj interface{}, tbName string, drvName ...string) (sql.Result, error) {
+	return insertStruct(exec, context.TODO(), obj, tbName, drvName...)
 }
-func InsertStructContext(exec Execer, ctx context.Context, obj interface{}, tbName string, drvNames ...string) (sql.Result, error) {
-	return insertStruct(exec, ctx, obj, tbName, drvNames...)
+func InsertStructContext(exec Execer, ctx context.Context, obj interface{}, tbName string, drvName ...string) (sql.Result, error) {
+	return insertStruct(exec, ctx, obj, tbName, drvName...)
 }
 
 // A sql.Query implements

api_test.go

@@ -0,0 +1,26 @@
+package qsql
+
+import "testing"
+
+func TestStmtWhereIn(t *testing.T) {
+	sqlite3Output := StmtWhereIn(0, 3, DRV_NAME_SQLITE3)
+	if sqlite3Output != "?,?,?" {
+		t.Fatalf("expect '?,?,?', but: %s", sqlite3Output)
+	}
+	pgOutput := StmtWhereIn(0, 3, DRV_NAME_POSTGRES)
+	if pgOutput != ":0,:1,:2" {
+		t.Fatalf("expect ':0,:1,:2', but: %s", pgOutput)
+	}
+	pgOutput1 := StmtWhereIn(1, 3, DRV_NAME_POSTGRES)
+	if pgOutput1 != ":1,:2,:3" {
+		t.Fatalf("expect ':1,:2,:3', but: %s", pgOutput1)
+	}
+	msOutput := StmtWhereIn(0, 3, DRV_NAME_SQLSERVER)
+	if msOutput != "@p0,@p1,@p2" {
+		t.Fatalf("expect '@p0,@p1,@p2', but@ %s", msOutput)
+	}
+	msOutput1 := StmtWhereIn(1, 3, DRV_NAME_SQLSERVER)
+	if msOutput1 != "@p1,@p2,@p3" {
+		t.Fatalf("expect '@p1,@p2,@p3', but@ %s", msOutput1)
+	}
+}

qsql.go

@@ -25,20 +25,8 @@ const (
 
 // field flag like: `db:"name"`
 // more: github.com/jmoiron/sqlx
-func insertStruct(exec Execer, ctx context.Context, obj interface{}, tbName string, drvNames ...string) (sql.Result, error) {
-	drvName := REFLECT_DRV_NAME
-	db, ok := exec.(*DB)
-	if ok {
-		drvName = db.DriverName()
-	} else {
-		drvNamesLen := len(drvNames)
-		if drvNamesLen > 0 {
-			if drvNamesLen != 0 {
-				panic(errors.New("'drvNames' expect only one argument").As(drvNames))
-			}
-			drvName = drvNames[0]
-		}
-	}
+func insertStruct(exec Execer, ctx context.Context, obj interface{}, tbName string, driverName ...string) (sql.Result, error) {
+	drvName := getDrvName(exec, driverName...)
 
 	fields, err := reflectInsertStruct(obj, drvName)
 	if err != nil {

struct.go

@@ -206,20 +206,20 @@ func travelStructField(f *reflectx.FieldInfo, v *reflect.Value, order *int, drvN
 	}
 
 	*outputVals = append(*outputVals, v.Interface())
-	switch {
-	case strings.Index(*drvName, "oracle") > -1, strings.Index(*drvName, "oci8") > -1:
+	switch *drvName {
+	case DRV_NAME_ORACLE, _DRV_NAME_OCI8:
 		*order += 1
 		*outputNames = append(*outputNames, []byte("\""+f.Name+"\",")...)
 		*outputInputs = append(*outputInputs, []byte(fmt.Sprintf(":%s,", f.Name))...)
-	case strings.Index(*drvName, "postgres") > -1:
+	case DRV_NAME_POSTGRES:
 		*outputNames = append(*outputNames, []byte("\""+f.Name+"\",")...)
 		*outputInputs = append(*outputInputs, []byte(fmt.Sprintf(":%d,", *order))...)
 		*order += 1
-	case strings.Index(*drvName, "sqlserver") > -1, strings.Index(*drvName, "mssql") > -1:
+	case DRV_NAME_SQLSERVER, _DRV_NAME_MSSQL:
 		*outputNames = append(*outputNames, []byte("["+f.Name+"],")...)
 		*outputInputs = append(*outputInputs, []byte(fmt.Sprintf("@p%d,", *order))...)
 		*order += 1
-	case strings.Index(*drvName, "mysql") > -1:
+	case DRV_NAME_MYSQL:
 		*order += 1
 		*outputNames = append(*outputNames, []byte("`"+f.Name+"`,")...)
 		*outputInputs = append(*outputInputs, []byte("?,")...)