dongfeng-pay/service/models/transaction.go

/***************************************************
 ** @Desc : This file for ...
 ** @Time : 2019/10/19 14:17
 ** @Author : yuebin
 ** @File : transaction
 ** @Last Modified by : yuebin
 ** @Last Modified time: 2019/10/19 14:17
 ** @Software: GoLand
****************************************************/
package models

import (
	"github.com/astaxie/beego/logs"
	"github.com/astaxie/beego/orm"
	"juhe/service/common"
	"juhe/service/utils"
)

func OperatorAccount(accountUid, operatorType string, amount float64) (string, bool) {
	o := orm.NewOrm()
	o.Begin()

	defer func(interface{}) {
		if r := recover(); r != nil {
			o.Rollback()
			logs.Error("operator account fail")
		}
	}(o)

	msg := ""
	//处理事务
	accountInfo := new(AccountInfo)
	if err := o.Raw("select * from account_info where account_uid = ? for update", accountUid).QueryRow(accountInfo); err != nil || accountInfo.AccountUid == "" {
		logs.Error("operator account get account info for update fail: ", err)
		o.Rollback()
		return msg, false
	}

	accountInfo.UpdateTime = utils.GetBasicDateTime()
	flag := true

	switch operatorType {
	case common.PLUS_AMOUNT: //处理加款操作
		accountInfo.Balance = accountInfo.Balance + amount
		accountInfo.SettleAmount = accountInfo.SettleAmount + amount
	case common.SUB_AMOUNT: //处理减款
		if accountInfo.Balance >= amount && accountInfo.SettleAmount >= amount {
			accountInfo.Balance = accountInfo.Balance - amount
			accountInfo.SettleAmount = accountInfo.SettleAmount - amount
		} else {
			msg = "账户余额不够减"
			flag = false
		}
	case common.FREEZE_AMOUNT: //处理冻结款
		accountInfo.FreezeAmount = accountInfo.FreezeAmount + amount
	case common.UNFREEZE_AMOUNT: //处理解冻款
		if accountInfo.FreezeAmount >= amount {
			accountInfo.FreezeAmount = accountInfo.FreezeAmount - amount
		} else {
			msg = "账户冻结金额不够解冻款"
			flag = false
		}
	}
	if !flag {
		o.Rollback()
		return msg, false
	}

	if _, err := o.Update(accountInfo); err != nil {
		logs.Error("operator account update account fail: ", err)
		o.Rollback()
		return msg, false
	}
	//往account_history表中插入一条动账记录
	accountHistory := AccountHistoryInfo{AccountUid: accountUid, AccountName: accountInfo.AccountName, Type: operatorType,
		Amount: amount, Balance: accountInfo.Balance, CreateTime: utils.GetBasicDateTime(), UpdateTime: utils.GetBasicDateTime()}

	if _, err := o.Insert(&accountHistory); err != nil {
		logs.Error("operator account insert account history fail: ", err)
		o.Rollback()
		return msg, false
	}
	if err := o.Commit(); err != nil {
		logs.Error("operator account commit fail: ", err)
		return msg, false
	} else {
		logs.Info("操作账户成功")
		return "", true
	}
}