v1.1

app/main/model/orderModel.go

@@ -1,6 +1,10 @@
 package model
 
 import (
+	"context"
+	"database/sql"
+	"fmt"
+
 	"github.com/zeromicro/go-zero/core/stores/cache"
 	"github.com/zeromicro/go-zero/core/stores/sqlx"
 )
@@ -12,6 +16,7 @@ type (
 	// and implement the added methods in customOrderModel.
 	OrderModel interface {
 		orderModel
+		UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error
 	}
 
 	customOrderModel struct {
@@ -25,3 +30,26 @@ func NewOrderModel(conn sqlx.SqlConn, c cache.CacheConf) OrderModel {
 		defaultOrderModel: newOrderModel(conn, c),
 	}
 }
+
+func (m *customOrderModel) UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error {
+	builder := m.defaultOrderModel.SelectBuilder().Where("user_id = ?", sourceUserID)
+	rows, err := m.defaultOrderModel.FindAll(ctx, builder, "")
+	if err != nil {
+		return err
+	}
+
+	keys := make([]string, 0, len(rows)*2)
+	for _, r := range rows {
+		keys = append(keys, fmt.Sprintf("%s%v", cacheYccOrderIdPrefix, r.Id))
+		keys = append(keys, fmt.Sprintf("%s%v", cacheYccOrderOrderNoPrefix, r.OrderNo))
+	}
+
+	_, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
+		query := fmt.Sprintf("UPDATE %s SET user_id = ? WHERE user_id = ?", m.defaultOrderModel.tableName())
+		if session != nil {
+			return session.ExecCtx(ctx, query, targetUserID, sourceUserID)
+		}
+		return conn.ExecCtx(ctx, query, targetUserID, sourceUserID)
+	}, keys...)
+	return err
+}