package model

import (
    "context"
    "database/sql"
    "ycc-server/common/globalkey"
    "fmt"
    "time"

    "github.com/zeromicro/go-zero/core/stores/cache"
    "github.com/zeromicro/go-zero/core/stores/sqlx"
)

var _ QueryModel = (*customQueryModel)(nil)

type (
    // QueryModel is an interface to be customized, add more methods here,
    // and implement the added methods in customQueryModel.
    QueryModel interface {
        queryModel
        DeleteBefore(ctx context.Context, before time.Time) (int64, error)
        UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error
    }

	customQueryModel struct {
		*defaultQueryModel
	}
)

// NewQueryModel returns a model for the database table.
func NewQueryModel(conn sqlx.SqlConn, c cache.CacheConf) QueryModel {
    return &customQueryModel{
        defaultQueryModel: newQueryModel(conn, c),
    }
}

func (m *customQueryModel) DeleteBefore(ctx context.Context, before time.Time) (int64, error) {
	var affected int64 = 0

	// 使用事务处理批量删除
	err := m.defaultQueryModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error {
		query := fmt.Sprintf("DELETE FROM %s WHERE create_time < ? AND del_state = ?", m.defaultQueryModel.table)
		result, err := session.ExecCtx(ctx, query, before.Format("2006-01-02 15:04:05"), globalkey.DelStateNo)
		if err != nil {
			return err
		}

		rows, err := result.RowsAffected()
		if err != nil {
			return err
		}

		affected = rows
		return nil
	})

	if err != nil {
		return 0, err
	}

	return affected, nil
}

func (m *customQueryModel) UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error {
    builder := m.defaultQueryModel.SelectBuilder().Where("user_id = ?", sourceUserID)
    rows, err := m.defaultQueryModel.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", cacheYccQueryIdPrefix, r.Id))
        keys = append(keys, fmt.Sprintf("%s%v", cacheYccQueryOrderIdPrefix, r.OrderId))
    }

    _, 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.defaultQueryModel.tableName())
        if session != nil {
            return session.ExecCtx(ctx, query, targetUserID, sourceUserID)
        }
        return conn.ExecCtx(ctx, query, targetUserID, sourceUserID)
    }, keys...)
    return err
}