Add BatchWriter

Author: minhduc140583

batch_inserter.go

@@ -70,7 +70,7 @@ func (w *BatchInserter) Write(ctx context.Context, models interface{}) ([]int, [
 		models2 = models
 	}
 	s := reflect.ValueOf(models2)
-	_, er2 := InsertBatchWithSchema(ctx, w.db, w.tableName, models2, w.ToArray, w.BuildParam)
+	_, er2 := InsertBatchWithSchema(ctx, w.db, w.tableName, models2, w.ToArray, w.BuildParam, w.Schema)
 
 	if er2 == nil {
 		// Return full success

batch_writer.go

@@ -0,0 +1,67 @@
+package sql
+
+import (
+	"context"
+	"database/sql"
+	"database/sql/driver"
+	"reflect"
+)
+
+type BatchWriter struct {
+	db           *sql.DB
+	tableName    string
+	Map          func(ctx context.Context, model interface{}) (interface{}, error)
+	Schema       *Schema
+	ToArray      func(interface{}) interface {
+		driver.Valuer
+		sql.Scanner
+	}
+}
+func NewBatchWriter(db *sql.DB, tableName string, modelType reflect.Type, options ...func(context.Context, interface{}) (interface{}, error)) *BatchWriter {
+	var mp func(context.Context, interface{}) (interface{}, error)
+	if len(options) > 0 && options[0] != nil {
+		mp = options[0]
+	}
+	return NewBatchWriterWithArray(db, tableName, modelType, nil, mp)
+}
+func NewBatchWriterWithArray(db *sql.DB, tableName string, modelType reflect.Type, toArray func(interface{}) interface {
+	driver.Valuer
+	sql.Scanner
+}, options ...func(context.Context, interface{}) (interface{}, error)) *BatchWriter {
+	var mp func(context.Context, interface{}) (interface{}, error)
+	if len(options) > 0 && options[0] != nil {
+		mp = options[0]
+	}
+	schema := CreateSchema(modelType)
+	return &BatchWriter{db: db, tableName: tableName, Schema: schema, Map: mp, ToArray: toArray}
+}
+
+func (w *BatchWriter) Write(ctx context.Context, models interface{}) ([]int, []int, error) {
+	successIndices := make([]int, 0)
+	failIndices := make([]int, 0)
+	var m interface{}
+	var er0 error
+	if w.Map != nil {
+		m, er0 = MapModels(ctx, models, w.Map)
+		if er0 != nil {
+			s0 := reflect.ValueOf(m)
+			_, er0b := InterfaceSlice(m)
+			failIndices = ToArrayIndex(s0, failIndices)
+			return successIndices, failIndices, er0b
+		}
+	} else {
+		m = models
+	}
+	s := reflect.ValueOf(m)
+	_, er2 := SaveBatchWithArray(ctx, w.db, w.tableName, m, w.ToArray, w.Schema)
+
+	if er2 == nil {
+		// Return full success
+		successIndices = ToArrayIndex(s, successIndices)
+		return successIndices, failIndices, er2
+	} else {
+		// Return full fail
+		failIndices = ToArrayIndex(s, failIndices)
+	}
+	return successIndices, failIndices, er2
+}

writer.go

@@ -33,6 +33,13 @@ func Commit(tx *sql.Tx, err error, options...bool) error {
 	}
 	return tx.Commit()
 }
+func Rollback(tx *sql.Tx, err error, options...int64) (int64, error) {
+	tx.Rollback()
+	if len(options) > 0 {
+		return options[0], err
+	}
+	return -1, err
+}
 func End(tx *sql.Tx, res int64, err error, options...bool) (int64, error) {
 	er := Commit(tx, err, options...)
 	return res, er