support scan assign slice cap

go-gorm · Sep 5, 2022 · 0a96898 · 0a96898
1 parent b3eb1c8
commit 0a96898
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scan.go b/scan.go
@@ -248,7 +248,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() {