- 优化dao.GenQuestionMarks

2019-08-30 09:54:23 +08:00
parent 27adff6add
commit 7afd15cf0e
1 changed files with 11 additions and 5 deletions
--- a/business/model/dao/dao_utils.go
+++ b/business/model/dao/dao_utils.go
@@ -73,12 +73,18 @@ func WrapUpdateULEntity(item interface{}, lastOperator string) interface{} {
 	return item
 }

-func GenQuestionMarks(count int) string {
-	marks := ""
+func GenQuestionMarks(count int) (marks string) {
 	if count > 0 {
-		marks = "?"
-		for i := 1; i < count; i++ {
-			marks += ", ?"
+		if count == 1 {
+			marks = "?"
+		} else {
+			builder := &strings.Builder{}
+			builder.Grow(1 + (count-1)*3)
+			builder.WriteString("?")
+			for i := 1; i < count; i++ {
+				builder.WriteString(", ?")
+			}
+			marks = builder.String()
 		}
 	}
 	return marks