php/app/common/repositories/user/UserSignHongbaoRepository.php

<?php

namespace app\common\repositories\user;

use app\common\dao\user\UserSignHongbaoDao;
use app\common\enum\user\UserSignHongbaoEnum;
use app\common\repositories\BaseRepository;
use app\common\repositories\store\order\StoreOrderRepository;
use app\entity\data\user\UserEntity;
use app\entity\data\user\UserSignHongbaoEntity;
use think\db\exception\DbException;
use think\exception\ValidateException;

class UserSignHongbaoRepository extends BaseRepository
{
    public function __construct(UserSignHongbaoDao $dao)
    {
        $this->dao = $dao;
    }

    /**
     * @param int $id
     * @return UserSignHongbaoEntity
     */
    public function getUserSignHongbaoEntityById(int $id): UserSignHongbaoEntity
    {
        $entityList = $this->getUserSignHongbaoEntityListByIdList([$id]);
        $entity = array_shift($entityList);
        if ($entity instanceof UserSignHongbaoEntity) {
            return $entity;
        } else {
            /** @var UserSignHongbaoEntity */
            return UserSignHongbaoEntity::newInstance();
        }
    }

    /**
     * @param array $idList
     * @return UserSignHongbaoEntity[]
     */
    public function getUserSignHongbaoEntityListByIdList(array $idList): array
    {
        return $this->dao->getUserSignHongbaoEntityListByIdList($idList);
    }

    /**
     * @param string $orderType
     * @param array $orderIdList
     * @return UserSignHongbaoEntity[]
     */
    public function getUserSignHongbaoEntityListByOrderIdListAndPending(string $orderType, array $orderIdList): array
    {
        return $this->dao->getUserSignHongbaoEntityListByOrderIdListAndPending($orderType, $orderIdList);
    }

    public function create($data)
    {
        return $this->dao->create($data);
    }

    /**
     * 写入红包记录
     * @param UserSignHongbaoEntity $userSignHongbaoEntity
     * @return UserSignHongbaoEntity
     */
    public function createByEntity(UserSignHongbaoEntity $userSignHongbaoEntity): UserSignHongbaoEntity
    {
        $result = $this->dao->create($userSignHongbaoEntity->toUnderlineArray())->toArray();
        /** @var UserSignHongbaoEntity $entity */
        $entity = UserSignHongbaoEntity::newInstance($result);
        return $entity;
    }

    /**
     * 添加红包记录
     * @param int $uid
     * @param int $type
     * @param float $number
     * @param string $mark
     * @param string $orderType
     * @param string $orderId
     * @return UserSignHongbaoEntity
     */
    public function addUserSingHongbao(int $uid, int $type, float $number, string $mark = '', string $orderType = '', string $orderId = ''): UserSignHongbaoEntity
    {
        if (!empty($orderType)) {
            if (!empty($orderId)) {
                if ($orderType == UserSignHongbaoEnum::ORDER_TYPE['Platform']['code']) {
                    /** @var StoreOrderRepository $storeOrderRepository */
                    $storeOrderRepository = app()->make(StoreOrderRepository::class);
                    $storeOrderEntity = $storeOrderRepository->getStoreOrderEntityByOrderId((int)$orderId);
                    if (empty($storeOrderEntity->getOrderId())) {
                        throw new ValidateException('未查询到订单信息');
                    }
                }
            }
        }

        /** @var UserSignHongbaoEntity $userSignHongbaoEntity */
        $userSignHongbaoEntity = UserSignHongbaoEntity::newInstance();
        $userSignHongbaoEntity->setUid($uid)
            ->setType($type)
            ->setNumber($number)
            ->setMark($mark)
            ->setOrderType($orderType)
            ->setOrderId($orderId)
            ->setAddtime(time())
            ->setStatus(UserSignHongbaoEnum::STATUS['PendingConfirmation']['code']);

        return $this->createByEntity($userSignHongbaoEntity);
    }

    /**
     * 将 贡献值 直推贡献值 记录设置为生效状态 并对用户的 贡献值 直推贡献值 字段进行增减
     * @param array $idList
     * @return array
     */
    public function executeByIdList(array $idList): array
    {
        $userSignHongbaoEntityList = $this->getUserSignHongbaoEntityListByIdList($idList);
        $userIdList = [];

        // 忽略 已经生效的 记录
        $userSignHongbaoEntityList = array_filter($userSignHongbaoEntityList, function ($entity) use (&$userIdList) {
            // 积分数据不为空 用户不为空 状态属于待结算 并且结算字段为空
            if ((!empty($entity->getId()) && !empty($entity->getUid()) && ($entity->getStatus() == UserSignHongbaoEnum::STATUS['PendingConfirmation']['code']) && empty($entity->getSettlementTime()))) {
                $userIdList[] = $entity->getUid();
                return true;
            }
            return false;
        });

        if (!empty($userSignHongbaoEntityList)) {
            // 获取用户 映射信息
            /** @var UserRepository $userRepository */
            $userRepository = app()->make(UserRepository::class);
            /** @var UserEntity[] $userEntityMapByUid */
            $userEntityMapByUid = [];
            if (!empty($userIdList)) {
                $userEntityMapByUid = $userRepository->getUserEntityMapByUidList($userIdList);
            }

            /** @var UserEntity[] $updateUserEntityList */
            $updateUserEntityList = [];
            /** @var UserSignHongbaoEntity[] $updateUserSignHongbaoEntityList */
            $updateUserSignHongbaoEntityList = [];
            foreach ($userSignHongbaoEntityList as $userSignHongbaoEntity) {

                $userEntity = $userEntityMapByUid[$userSignHongbaoEntity->getUid()];
                if (empty($userEntity) || empty($userEntity->getUid())) {
                    throw new ValidateException("未查询到用户信息(HongbaoId:{$userSignHongbaoEntity->getId()})");
                }

                // 组建 更新 用户信息
                if (isset($updateUserEntityList[$userEntity->getUid()])) {
                    $updateUserEntity = $updateUserEntityList[$userEntity->getUid()];
                } else {
                    /** @var UserEntity $updateUserEntity */
                    $updateUserEntity = UserEntity::newInstance();
                    $updateUserEntity
                        ->setUid($userEntity->getUid())
                        ->setHongbao($userEntity->getHongbao());
                }
                // 组建 账单 记录
                if (isset($updateUserSignHongbaoEntityList[$userSignHongbaoEntity->getId()])) {
                    $updateUserSignHongbaoEntity = $updateUserSignHongbaoEntityList[$userSignHongbaoEntity->getId()];
                } else {
                    /** @var UserSignHongbaoEntity $updateUserSignHongbaoEntity */
                    $updateUserSignHongbaoEntity = UserSignHongbaoEntity::newInstance();
                    // 更新 账单记录 状态
                    $updateUserSignHongbaoEntity
                        ->setId($userSignHongbaoEntity->getId())
                        ->setStatus(UserSignHongbaoEnum::STATUS['Valid']['code'])
                        ->setSettlementTime(date('Y-m-d H:i:s'));
                }

                if ($userSignHongbaoEntity->getType() == UserSignHongbaoEnum::TYPE['Consumption']['code']) {
                    // 消耗
                    $updateUserEntity->setHongbao(bcsub($updateUserEntity->getHongbao(), $userSignHongbaoEntity->getNumber(), 2));
                    if ($updateUserEntity->getHongbao() < 0.00) {
                        throw new ValidateException('用户红包不足');
                    }
                } else {
                    $updateUserEntity->setHongbao(bcadd($updateUserEntity->getHongbao(), $userSignHongbaoEntity->getNumber(), 2));
                }
                $updateUserSignHongbaoEntity->setSurplus($updateUserEntity->getHongbao());

                $updateUserEntityList[$userSignHongbaoEntity->getUid()] = $updateUserEntity;
                $updateUserSignHongbaoEntityList[$userSignHongbaoEntity->getId()] = $updateUserSignHongbaoEntity;
            }
            if (!empty($updateUserEntityList)) {
                $userRepository->batchUpdateUserDataByDataList(
                    array_map(function (UserEntity $updateUserEntity) {
                        return $updateUserEntity->toUnderlineArray();
                    }, $updateUserEntityList)
                );
            }

            if (!empty($updateUserBillEntityList)) {
                $this->batchUpdateUserSignHongbaoDataByDataList(
                    array_map(function (UserSignHongbaoEntity $updateUserSignHongbaoEntity) {
                        return $updateUserSignHongbaoEntity->toUnderlineArray();
                    }, $updateUserSignHongbaoEntityList)
                );
            }
        }
        return $idList;
    }

    /**
     * @param $dataList
     * @return array
     */
    public function batchUpdateUserSignHongbaoDataByDataList($dataList): array
    {
        return $this->dao->saveAll($dataList);
    }
}