kknongchang_2506/app/Module/Fund/Services/FundService.php

<?php

namespace App\Module\Fund\Services;


use App\Module\Fund\Dto\CirculationDto;
use App\Module\Fund\Dto\FundAccountDto;
use App\Module\Fund\Dto\FundAdminDto;
use App\Module\Fund\Dto\TradeResultDto;
use App\Module\Fund\Dto\TransferResultDto;
use App\Module\Fund\Enums\FUND_TYPE;
use App\Module\Fund\Enums\LOG_TYPE;
use App\Module\Fund\Models\FundAdminModel;
use App\Module\Fund\Models\FundModel;
use App\Module\Fund\Logic\Circulation;
use App\Module\Fund\Logic\Transfer;
use App\Module\Fund\Logic\User;
use Illuminate\Support\Facades\DB;
use UCore\Db\Helper;


class FundService
{

    private int $userId;
    private int $fundId;
    private ?FundModel $fundModel;

    public function __construct(int $userId,  $fundId)
    {
        $this->userId = $userId;
        if(is_int($fundId)){
            $this->fundId = $fundId;
        }else{
            $this->fundId = $fundId->valueInt();
        }

        $this->fundModel = FundModel::query()
            ->where('user_id', $userId)
            ->where('fund_id', $fundId)
            ->first();
    }


    /**
     * 资金流转
     * (同用户，同币种，不同账户)
     *
     * @param  FUND_TYPE  $to_fund_id 目标资金类型
     * @param  int  $amount 金额
     * @param  int  $re_id 关联ID
     * @param  string  $re_type 关联类型
     * @param  string  $remark 备注
     * @return CirculationDto|string 成功返回DTO，失败返回错误信息
     */
    public function circulation(FUND_TYPE $to_fund_id, int $amount, int $re_id, string $re_type, string $remark)
    {
        # 确认货币种类一致
        if (!CurrencyService::check($this->fundId, $to_fund_id->valueInt())) {
            return '币种不一致,禁止划转';
        }

        # 实例化操作对象

        Helper::check_tr();
        # 先进行转账记录
        $circulation_id = Circulation::handle($this->userId, $this->fundId, $to_fund_id->valueInt(), $amount, $re_id, $re_type,
            $remark);
        if (is_string($circulation_id)) {
            return $circulation_id;
        }
        # 进行双方的资金修改
        # 先减少自己的
        $re46 = User::handle($this->userId, $this->fundId, -$amount, LOG_TYPE::CIRCULATION, $circulation_id, $remark);
        if (is_string($re46)) {
            return $re46;
        }
        # 再增加自己另一个账户
        $re51 = User::handle($this->userId, $to_fund_id->valueInt(), $amount, LOG_TYPE::CIRCULATION, $circulation_id, $remark);

        if (is_string($re51)) {
            return $re51;
        }

        // 获取流转记录
        $circulationModel = \App\Module\Fund\Models\FundCirculationModel::find($circulation_id);
        if (!$circulationModel) {
            return '流转记录不存在';
        }

        // 获取资金类型名称映射
        $fundNames = AccountService::getFundsDesc();

        // 将模型转换为DTO并返回
        return CirculationDto::fromModel($circulationModel, $fundNames);
    }

    /**
     * 获取账户信息
     *
     * @return FundAccountDto|null 账户DTO，如果账户不存在则返回null
     */
    public function getAccount(): ?FundAccountDto
    {
        // 如果账户不存在，返回null
        if (!$this->fundModel) {
            return null;
        }

        // 获取资金类型名称映射
        $fundNames = AccountService::getFundsDesc();

        // 将模型转换为DTO
        return FundAccountDto::fromModel($this->fundModel, $fundNames);
    }

    /**
     * 获取余额
     *
     * @return int 账户余额
     */
    public function balance(): int
    {
        return $this->fundModel->balance ?? 0;
    }

    /**
     * 转账
     * （不同人，同账户/同币种）
     *
     * @param int $toUserId 接收用户ID
     * @param int $amount 金额
     * @param string $remark 备注
     * @return TransferResultDto|string 成功返回DTO，失败返回错误信息
     */
    public function transfer(int $toUserId, int $amount, string $remark)
    {
        # 实例化操作对象

        # 开启事务
        DB::beginTransaction();
        # 先进行转账记录
        $transfer_id = Transfer::to_user(
            $this->userId,
            $this->fundId,
            $toUserId,
            $amount,
            $remark);
        if (is_string($transfer_id)) {
            DB::rollBack();
            return $transfer_id;
        }
        # 进行双方的资金修改
        # 先减少自己的
        $re46 = User::handle($this->userId, $this->fundId, -$amount, LOG_TYPE::TRANSFER, $transfer_id, $remark);
        if (is_string($re46)) {
            DB::rollBack();
            return $re46;
        }
        # 再增加别人的
        $re51 = User::handle($toUserId, $this->fundId, $amount, LOG_TYPE::TRANSFER, $transfer_id, $remark);
        if (is_string($re51)) {
            DB::rollBack();
            return $re51;
        }
        DB::commit();

        // 获取转账记录
        $transferModel = \App\Module\Fund\Models\FundTransferModel::find($transfer_id);
        if (!$transferModel) {
            return '转账记录不存在';
        }

        // 获取资金类型名称映射
        $fundNames = AccountService::getFundsDesc();

        // 获取用户名称映射
        $userNames = [];
        $userIds = [$this->userId, $toUserId];
        $users = \App\Module\User\Models\User::whereIn('id', $userIds)->get();
        foreach ($users as $user) {
            $userNames[$user->id] = $user->username;
        }

        // 将模型转换为DTO并返回
        return TransferResultDto::fromModel($transferModel, $fundNames, $userNames);
    }


    /**
     * 贸易(业务关联)
     * 同账户/同币种，不同用户
     *
     * @param  int  $toUserId 接收用户ID
     * @param  int  $amount 金额
     * @param  string  $transfer_type 关联类型
     * @param  string  $transfer_id 关联ID
     * @param  string  $remark 备注
     * @return TradeResultDto|string 成功返回DTO，失败返回错误信息
     */
    public function trade(int $toUserId, int $amount, $transfer_type, $transfer_id, string $remark)
    {
        # 检查事务开启
        Helper::check_tr();
        $full_transfer_id = $transfer_type.'-'.$transfer_id;
        # 先减少自己的
        $re46 = User::handle($this->userId, $this->fundId, -$amount, LOG_TYPE::TRADE, $full_transfer_id, $remark);

        if (is_string($re46)) {
            \UCore\Trace::addData('error', $re46);
            return $re46;
        }
        # 再增加别人的
        $re51 = User::handle($toUserId, $this->fundId, $amount, LOG_TYPE::TRANSFER, $full_transfer_id, $remark);
        if (is_string($re51)) {
            \UCore\Trace::addData('error', $re51);
            return $re51;
        }

        // 获取资金类型名称映射
        $fundNames = AccountService::getFundsDesc();

        // 获取用户名称映射
        $userNames = [];
        $userIds = [$this->userId, $toUserId];
        $users = \App\Module\User\Models\User::whereIn('id', $userIds)->get();
        foreach ($users as $user) {
            $userNames[$user->id] = $user->username;
        }

        // 创建DTO并返回
        return TradeResultDto::create(
            $this->userId,
            $toUserId,
            $this->fundId,
            $amount,
            $transfer_type,
            $transfer_id,
            $remark,
            $fundNames,
            $userNames
        );
    }


    /**
     * Admin 资金操作
     *
     * @param  int  $admin_id
     * @param  FUND_TYPE  $fund_id
     * @param  int  $fund_fee
     * @param $remark
     * @return FundAdminDto|string 成功返回DTO，失败返回错误信息
     */
    public function admin_operate(int $admin_id, FUND_TYPE $fund_id, int $fund_fee, $remark)
    {

        $data = [
            'total_fee' => $fund_fee,
            'status' => 1,
            'user_id' => $this->userId,
            'fund_id' => $fund_id->valueInt(),
            'admin_id' => $admin_id,
            'create_time' => time(),
            'remark' => $remark
        ];

        # 启动事务
        DB::beginTransaction();
        # 写日志
        $fund_admin = new FundAdminModel();
        $fund_admin->setData($data);
        if ($fund_admin->save() === false) {
            DB::rollBack();

            return '_Model-error';
        }
        # 进行资金操作
        $re = \App\Module\Fund\Logic\User::handle($this->userId, $fund_id->value, $fund_fee,
                                                  \App\Module\Fund\Enums\LOG_TYPE::ADMIN, $fund_admin->id, $remark);
        if (is_string($re)) {
            DB::rollBack();

            return $re;
        }
        DB::commit();

        // 获取资金类型名称映射
        $fundNames = AccountService::getFundsDesc();

        // 将模型转换为DTO并返回
        return FundAdminDto::fromModel($fund_admin, $fundNames);
    }

}