support scan assign slice cap

go-gorm · Aug 24, 2022 · 0e655ba · 0e655ba
1 parent 8c3018b
commit 0e655ba
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scan.go b/scan.go
@@ -246,7 +246,11 @@ func Scan(rows Rows, db *DB, mode ScanMode) {
 
 			if !update || reflectValue.Len() == 0 {
 				update = false
-				db.Statement.ReflectValue.Set(reflect.MakeSlice(reflectValue.Type(), 0, 20))
+				sliceCap := reflectValue.Cap()
+				if sliceCap <= 0 {
+					sliceCap = 20
+				}
+				db.Statement.ReflectValue.Set(reflect.MakeSlice(reflectValue.Type(), 0, sliceCap))
 			}
 
 			for initialized || rows.Next() {