support multi-value insertion

outprog · Feb 2, 2019 · 45d8dc1 · 45d8dc1
1 parent 5810172
commit 45d8dc1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/kadb.go b/kadb.go
@@ -2,6 +2,7 @@ package kadb
 
 import (
 	"os"
+	"reflect"
 
 	"github.com/Shopify/sarama"
 	cluster "github.com/bsm/sarama-cluster"
@@ -48,7 +49,16 @@ func (k *Kadb) toDB(msg *sarama.ConsumerMessage) error {
 		return err
 	}
 	if dMsg != nil {
-		k.db.Create(dMsg)
+		switch reflect.TypeOf(dMsg).Kind() {
+		case reflect.Slice:
+			va := reflect.ValueOf(dMsg)
+			for i := 0; i < va.Len(); i++ {
+				value := va.Index(i)
+				k.db.Create(value.Interface())
+			}
+		case reflect.Ptr:
+			k.db.Create(dMsg)
+		}
 	}
 	return nil
 }