NN2022081301/application/common/model/Users.php

<?php


namespace app\common\model;

use app\common\model\UsersBalanceRecord;

class Users extends BaseModel
{
    /**
     * @var string
     */
    protected $name = 'users';
    /**
     * @var array
     */
    protected $auto = [];
    /**
     * @var array
     */
    protected $insert = ['created_at','updated_at'];
    /**
     * @var array
     */
    protected $update = ['updated_at'];
    /**
     * @var array
     */
    protected $hidden = ['updated_at'];

    /**
     * 自关联推荐用户
     * @return \think\model\relation\BelongsTo
     */
    public function recommend()
    {
        return $this->belongsTo(self::class,'parent_id','id');
    }

    /**
     * 用户资金变动
     *
     * @author 许祖兴 < zuxing.xu@lettered.cn>
     * @date 2020/7/14 10:34
     *
     * @param int $userId 用户ID
     * @param float $amount 变动金额
     * @param string $remark 备注说明
     * @param bool $inc 是否增长
     * @throws \think\exception\PDOException
     */
    public function changeBalance($userId,$amount, $remark = "", $inc = false)
    {
        $action = $inc ? 'setInc' : 'setDec';
        $this->startTrans();
        try {
            // 查余额
            $user = $this->lock(true)->where(['id' => $userId])->find();
            // 资金变动
            $this->where(['id' => $userId])->{$action}('balance',$amount);
            // 记录写入
            $data = [
                'user_id' => $userId,
                'action' => $inc ? 1 : 0,
                'remark' => $remark
            ];
            // 加减
            if ($inc){
                $data['inc_amount'] = $amount;
                $data['aft_amount'] = sprintf("%.2f", round($user['balance'] + $amount, 2));
            }else{
                $data['dec_amount'] = $amount;
                $data['aft_amount'] = sprintf("%.2f", round($user['balance'] - $amount, 2));
            }
            UsersBalanceRecord::create($data,true);
            $this->commit();
        }catch (\Exception $e){
            $this->rollback();
        }
    }

    /**
     * 用户资产变动
     *
     * @author 许祖兴 < zuxing.xu@lettered.cn>
     * @date 2020/7/14 10:34
     *
     * @param int $userId 用户ID
     * @param float $amount 变动金额
     * @param string $remark 备注说明
     * @param bool $inc 是否增长
     * @throws \think\exception\PDOException
     */
    public function changeProperty($userId,$amount, $remark = "", $inc = false)
    {
        $action = $inc ? 'setInc' : 'setDec';
        $this->startTrans();
        try {
            // 查余额
            $user = $this->lock(true)->where(['id' => $userId])->find();
            // 资金变动
            $this->where(['id' => $userId])->{$action}('property',$amount);
            // 记录写入
            $data = [
                'user_id' => $userId,
                'action' => $inc ? 1 : 0,
                'remark' => $remark
            ];
            // 加减
            if ($inc){
                $data['inc_amount'] = $amount;
                $data['aft_amount'] = sprintf("%.3f", round($user['property'] + $amount, 3));
            }else{
                $data['dec_amount'] = $amount;
                $data['aft_amount'] = sprintf("%.3f", round($user['property'] - $amount, 3));
            }
            UsersPropertyRecord::create($data,true);
            $this->commit();
        }catch (\Exception $e){
            $this->rollback();
        }
    }
}