u

code/base-project/base-go-v2/internal/db/update.go

@@ -0,0 +1,103 @@
+package db
+
+import (
+	"fmt"
+	"strings"
+)
+
+// ---------------- 内部辅助函数 ----------------
+
+// buildUpdateSQL 构建 UPDATE SQL 和参数
+// idColumn: 主键列名
+func buildUpdateSQL(table string, data map[string]interface{}, idColumn string) (string, []interface{}) {
+	var setParts []string
+	var values []interface{}
+	i := 1
+	for k, v := range data {
+		setParts = append(setParts, fmt.Sprintf("%s=$%d", k, i))
+		values = append(values, v)
+		i++
+	}
+	// WHERE id = $n
+	sql := fmt.Sprintf("UPDATE %s SET %s WHERE %s=$%d",
+		table,
+		strings.Join(setParts, ", "),
+		idColumn,
+		i,
+	)
+	return sql, values
+}
+
+// ---------------- 公共方法 ----------------
+
+// UpdateOne 更新单条记录，按主键 idColumn
+func UpdateOne(table string, idColumn string, id interface{}, data map[string]interface{}) (int64, error) {
+	tx, err := DB.Beginx()
+	if err != nil {
+		return 0, err
+	}
+
+	sql, values := buildUpdateSQL(table, data, idColumn)
+	values = append(values, id) // 最后一个参数是 id
+
+	res, err := tx.Exec(sql, values...)
+	if err != nil {
+		err := tx.Rollback()
+		if err != nil {
+			return 0, err
+		}
+		return 0, err
+	}
+
+	if err := tx.Commit(); err != nil {
+		return 0, err
+	}
+
+	return res.RowsAffected()
+}
+
+// UpdateBatch 批量更新，dataList 中每个 map 必须包含主键 idColumn
+func UpdateBatch(table string, idColumn string, dataList []map[string]interface{}) (int64, error) {
+	if len(dataList) == 0 {
+		return 0, nil
+	}
+
+	tx, err := DB.Beginx()
+	if err != nil {
+		return 0, err
+	}
+
+	total := int64(0)
+	for _, data := range dataList {
+		idValue, ok := data[idColumn]
+		if !ok {
+			tx.Rollback()
+			return 0, fmt.Errorf("缺少主键列 %s", idColumn)
+		}
+
+		// 移除主键列，否则会重复在 SET 中出现
+		delete(data, idColumn)
+
+		sql, values := buildUpdateSQL(table, data, idColumn)
+		values = append(values, idValue)
+
+		res, err := tx.Exec(sql, values...)
+		if err != nil {
+			tx.Rollback()
+
+			return 0, err
+		}
+		rows, err := res.RowsAffected()
+		if err != nil {
+			tx.Rollback()
+			return 0, err
+		}
+		total += rows
+	}
+
+	if err := tx.Commit(); err != nil {
+		return 0, err
+	}
+
+	return total, nil
+}