package dao

import (
	"context"
	"fmt"
	"go-common/app/service/live/wallet/model"
	"go-common/library/database/sql"
	"go-common/library/ecode"
	"go-common/library/log"
	"time"
)

const (
	_selWallet                  = "SELECT uid,gold,iap_gold,silver,cost_base,gold_recharge_cnt,gold_pay_cnt,silver_pay_cnt,snapshot_time,snapshot_gold,snapshot_iap_gold,snapshot_silver,reserved1,reserved2 FROM user_wallet_%d WHERE uid=? FOR UPDATE"
	_recharge                   = "UPDATE user_wallet_%d set %s = %s + %d,%s=%s+%d where uid = ?"
	_rechargeWihoutCnt          = "UPDATE user_wallet_%d set %s=%s + %d where uid = ?"
	_rechargeWithSnap           = "UPDATE user_wallet_%d set %s=%s + %d,%s = %s+ %d, snapshot_time = ? , snapshot_gold = ? , snapshot_iap_gold = ? , snapshot_silver = ? where uid = ?"
	_rechargeWithSnapWithoutCnt = "UPDATE user_wallet_%d set %s = %s + %d, snapshot_time = ? , snapshot_gold = ? , snapshot_iap_gold = ? , snapshot_silver = ? where uid = ?"
	_exchange                   = "UPDATE user_wallet_%d set %s = %s - %d , %s = %s + %d, %s = %s + %d, %s = %s + %d where uid = ?"
	_exhcangeWithSnap           = "UPDATE user_wallet_%d set %s = %s - %d , %s = %s + %d, %s = %s + %d, %s = %s + %d, snapshot_time = ? , snapshot_gold = ? , snapshot_iap_gold = ? , snapshot_silver = ? where uid = ? "
	_modifyCnt                  = "UPDATE user_wallet_%d set gold_pay_cnt = gold_pay_cnt + %d, gold_recharge_cnt = gold_recharge_cnt + %d, silver_pay_cnt = silver_pay_cnt + %d where uid = ?"
)

// 开启事务
func (d *Dao) BeginTx(c context.Context) (conn *sql.Tx, err error) {
	return d.db.Begin(c)
}

func (d *Dao) DoTx(c context.Context, doFunc func(conn *sql.Tx) (v interface{}, err error)) (v interface{}, err error) {
	conn, err := d.BeginTx(c)
	if err != nil {
		err = ecode.ServerErr
		return
	}
	v, err = doFunc(conn)
	var txErr error
	if err != nil {
		conn.Rollback()
		err = ecode.ServerErr
	} else {
		txErr = conn.Commit()
		if txErr != nil {
			err = ecode.ServerErr
			v = nil
		}
	}
	return v, err
}

// 为了后续的更新获取数据
func (d *Dao) WalletForUpdate(conn *sql.Tx, uid int64) (wallet *model.DetailWithSnapShot, err error) {
	row := conn.QueryRow(fmt.Sprintf(_selWallet, tableIndex(uid)), uid)
	wallet = &model.DetailWithSnapShot{}
	var snapShotTime time.Time
	if err = row.Scan(&wallet.Uid, &wallet.Gold, &wallet.IapGold, &wallet.Silver, &wallet.CostBase, &wallet.GoldRechargeCnt,
		&wallet.GoldPayCnt, &wallet.SilverPayCnt, &snapShotTime, &wallet.SnapShotGold, &wallet.SnapShotIapGold,
		&wallet.SnapShotSilver, &wallet.Reserved1, &wallet.Reserved2); err == sql.ErrNoRows {
		// 查询结果为空时，初始化数据
		_, err = d.InitWalletInTx(conn, uid, 0, 0, 0)
		wallet.SnapShotTime = snapShotTime.Format("2006-01-02 15:04:05")
		return
	}
	if err != nil {
		log.Error("[tx.wallet|Melonseed] row.Scan err: %s", err.Error())
		return
	}
	wallet.SnapShotTime = snapShotTime.Format("2006-01-02 15:04:05")
	return
}

// InitExp 初始化用户钱包,用于首次查询
func (d *Dao) InitWalletInTx(conn *sql.Tx, uid int64, gold int64, iap_gold int64, silver int64) (row int64, err error) {
	res, err := conn.Exec(fmt.Sprintf(_insWallet, tableIndex(uid)), uid, gold, iap_gold, silver)
	if err != nil {
		log.Error("[tx.wallet|InitWallet] Exec err: %v", err)
		return
	}
	return res.RowsAffected()
}

func (d *Dao) execSqlInTx(conn *sql.Tx, sql *string, params ...interface{}) (affect int64, err error) {
	res, err := conn.Exec(*sql, params...)
	if err != nil {
		log.Error("[tx.wallet|execSqlInTx] Exec err: %v sql:%s", err, *sql)
		return
	}
	return res.RowsAffected()
}

func (d *Dao) changeCoinInTx(conn *sql.Tx, uid int64, sysCoinTypeNo int32, num int64, originWallet *model.DetailWithSnapShot, cntField string) (affect int64, err error) {
	// 判断

	coinType := model.GetSysCoinTypeByNo(sysCoinTypeNo)
	var s string
	absNum := num
	if absNum < 0 {
		absNum = -absNum
	}
	if model.TodayNeedSnapShot(originWallet) {
		if cntField == "" {
			s = getRechargeWithoutCntWithSnapShotSQL(uid, coinType, num)
		} else {
			s = getRechargeWithSnapShotSQL(uid, coinType, num, cntField, absNum)
		}
		date := time.Now().Format("2006-01-02 15:04:05")
		return d.execSqlInTx(conn, &s, date, originWallet.Gold, originWallet.IapGold, originWallet.Silver, uid)
	} else {
		if cntField == "" {
			s = getRechargeWithoutCntSQL(uid, coinType, num)
		} else {
			s = getRechargeSQL(uid, coinType, num, cntField, absNum)
		}
		return d.execSqlInTx(conn, &s, uid)
	}
}

// RechargeGold 充值IOS金瓜子 记入充值总值
func (d *Dao) RechargeCoinInTx(conn *sql.Tx, uid int64, sysCoinTypeNo int32, num int64, originWallet *model.DetailWithSnapShot) (affect int64, err error) {
	rechargerCntField := model.GetRechargeCnt(sysCoinTypeNo)
	return d.changeCoinInTx(conn, uid, sysCoinTypeNo, num, originWallet, rechargerCntField)
}

func (d *Dao) PayCoinInTx(conn *sql.Tx, uid int64, sysCoinTypeNo int32, num int64, originWallet *model.DetailWithSnapShot) (affect int64, err error) {
	cntField := model.GetPayCnt(sysCoinTypeNo)
	return d.changeCoinInTx(conn, uid, sysCoinTypeNo, -num, originWallet, cntField)
}

func (d *Dao) ModifyCoinInTx(conn *sql.Tx, uid int64, sysCoinTypeNo int32, num int64, originWallet *model.DetailWithSnapShot) (affect int64, err error) {
	return d.changeCoinInTx(conn, uid, sysCoinTypeNo, num, originWallet, "")
}

func (d *Dao) ExchangeCoinInTx(conn *sql.Tx, uid int64, srcSysCoinTypeNo int32, srcNum int64, destSysCoinTypeNo int32, destNum int64, originWallet *model.DetailWithSnapShot) (affect int64, err error) {
	rechargeCntNum := destNum
	var rechargerCntField string
	if destSysCoinTypeNo == model.SysCoinTypeSilver { // 如果为银瓜子则不存在充值统计总数字段 使用 消费统计字段代替　适配sql
		rechargerCntField = "silver_pay_cnt"
		rechargeCntNum = 0
	} else {
		rechargerCntField = model.GetRechargeCnt(destSysCoinTypeNo)
	}
	payCntField := model.GetPayCnt(srcSysCoinTypeNo)
	srcCoinType := model.GetSysCoinTypeByNo(srcSysCoinTypeNo)
	destCoinType := model.GetSysCoinTypeByNo(destSysCoinTypeNo)

	var s string
	if model.TodayNeedSnapShot(originWallet) {
		s = fmt.Sprintf(_exhcangeWithSnap, tableIndex(uid), srcCoinType, srcCoinType, srcNum, destCoinType, destCoinType, destNum, rechargerCntField, rechargerCntField, rechargeCntNum, payCntField, payCntField, srcNum)
		date := time.Now().Format("2006-01-02 15:04:05")
		return d.execSqlInTx(conn, &s, date, originWallet.Gold, originWallet.IapGold, originWallet.Silver, uid)
	} else {
		s = fmt.Sprintf(_exchange, tableIndex(uid), srcCoinType, srcCoinType, srcNum, destCoinType, destCoinType, destNum, rechargerCntField, rechargerCntField, rechargeCntNum, payCntField, payCntField, srcNum)
		return d.execSqlInTx(conn, &s, uid)
	}
}

func getRechargeWithSnapShotSQL(uid int64, coinType string, num int64, cntField string, cntNum int64) string {
	return fmt.Sprintf(_rechargeWithSnap, tableIndex(uid), coinType, coinType, num, cntField, cntField, cntNum)
}

func getRechargeWithoutCntWithSnapShotSQL(uid int64, coinType string, num int64) string {
	return fmt.Sprintf(_rechargeWithSnapWithoutCnt, tableIndex(uid), coinType, coinType, num)
}

func getRechargeSQL(uid int64, coinType string, num int64, cntField string, cntNum int64) string {
	return fmt.Sprintf(_recharge, tableIndex(uid), coinType, coinType, num, cntField, cntField, cntNum)
}

func getRechargeWithoutCntSQL(uid int64, coinType string, num int64) string {
	return fmt.Sprintf(_rechargeWihoutCnt, tableIndex(uid), coinType, coinType, num)
}

func (d *Dao) NewCoinStreamRecordInTx(conn *sql.Tx, record *model.CoinStreamRecord) (int64, error) {
	s := fmt.Sprintf(_newCoinStremaRecord, getCoinStreamTable(record.TransactionId))
	date := model.GetWalletFormatTime(record.OpTime)
	return d.execSqlInTx(conn, &s, record.Uid, record.TransactionId, record.ExtendTid, record.CoinType,
		record.DeltaCoinNum, record.OrgCoinNum, record.OpResult, record.OpReason, record.OpType, date,
		record.BizCode, record.Area, record.Source, record.MetaData, record.BizSource,
		record.Platform, record.Reserved1, record.Version)
}

func (d *Dao) NewCoinExchangeRecordInTx(conn *sql.Tx, record *model.CoinExchangeRecord) (int64, error) {
	s := fmt.Sprintf(_newCoinExchange, getCoinExchangeTable(record.Uid))
	date := model.GetWalletFormatTime(record.ExchangeTime)
	return d.execSqlInTx(conn, &s, record.Uid, record.TransactionId, record.SrcType, record.SrcNum, record.DestType, record.DestNum, record.Status, date)
}

func (d *Dao) ModifyCntInTx(conn *sql.Tx, uid int64, goldPay int, goldRecharge int, silverPay int) (int64, error) {
	s := fmt.Sprintf(_modifyCnt, tableIndex(uid), goldPay, goldRecharge, silverPay)
	return d.execSqlInTx(conn, &s, uid)
}