订单结算-共富体系用户升级

app/common/dao/shared/SharedProsperityUserDao.php

@@ -70,6 +70,31 @@ class SharedProsperityUserDao extends BaseDao
     }
 
     /**
+     * @param int $parentId
+     * @return SharedProsperityUserEntity[]
+     */
+    public function getSharedProsperityUserEntityListByParentId(int $parentId): array
+    {
+        $entityList = [];
+        try {
+            /** @var SharedProsperityUser $model */
+            $model = $this->getModel();
+            $dataRes = $model::getDB()
+                ->where('parent_id', '=', $parentId)
+                ->select()
+                ->toArray();
+            if (!empty($dataRes)) {
+                $entityList = array_map(function ($data) {
+                    return SharedProsperityUserEntity::newInstance($data);
+                }, $dataRes);
+            }
+        } catch (\Exception $e) {
+
+        }
+        return $entityList;
+    }
+
+    /**
      * 获取所有直推用户（不分页）
      * @param int $userId
      * @return array

app/common/enum/shared/SharedProsperityUserEnum.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace app\common\enum\shared;
+
+use app\common\enum\CommonEnum;
+
+class SharedProsperityUserEnum extends CommonEnum
+{
+    // 统计团队 数据 递归层数
+    const TEAM_DEPTH = 5;
+}

app/common/repositories/shared/SharedProsperityUserRepository.php

@@ -5,6 +5,7 @@ namespace app\common\repositories\shared;
 use app\common\dao\shared\SharedProsperityUserDao;
 use app\common\dao\user\UserDao;
 use app\common\enum\CommonEnum;
+use app\common\enum\shared\SharedProsperityUserEnum;
 use app\common\repositories\BaseRepository;
 use app\entity\data\shared\SharedProsperityRecommendConfigEntity;
 use app\entity\data\shared\SharedProsperityUserEntity;
@@ -75,6 +76,16 @@ class SharedProsperityUserRepository extends BaseRepository
     }
 
     /**
+     * 获取直接下级用户ID列表
+     * @param int $parentId 用户ID
+     * @return SharedProsperityUserEntity[] 直接下级用户ID列表
+     */
+    public function getSharedProsperityUserEntityListByParentId(int $parentId): array
+    {
+        return $this->dao->getSharedProsperityUserEntityListByParentId($parentId);
+    }
+
+    /**
      * @param array $userIdList
      * @return SharedProsperityUserEntity[]
      */
@@ -117,7 +128,7 @@ class SharedProsperityUserRepository extends BaseRepository
         }
     }
 
-    public function updateByUserId($userId,$data)
+    public function updateByUserId($userId, $data): int
     {
         try {
             return $this->dao->updateByUserId($userId, $data);
@@ -289,7 +300,9 @@ class SharedProsperityUserRepository extends BaseRepository
      */
     public function identityUpgrade(array $userIdList)
     {
-        $sharedProsperityUserEntityList = $this->getSharedProsperityUserEntityListByUserIdList($userIdList);
+        if (empty($userIdList)) {
+            return;
+        }
 
         /** @var SharedProsperityRecommendConfigRepository $sharedProsperityRecommendConfigRepository */
         $sharedProsperityRecommendConfigRepository = app()->make(SharedProsperityRecommendConfigRepository::class);
@@ -297,23 +310,214 @@ class SharedProsperityUserRepository extends BaseRepository
         usort($sharedProsperityRecommendConfigEntityList, function (SharedProsperityRecommendConfigEntity $a, SharedProsperityRecommendConfigEntity $b) {
             return $a->getPv() <=> $b->getPv();  // 从小到大排序
         });
+
+        $sharedProsperityUserEntityList = $this->getSharedProsperityUserEntityListByUserIdList($userIdList);
+
+        // 批量更新用户等级
+        $this->batchUpgradeUsers($sharedProsperityUserEntityList, $sharedProsperityRecommendConfigEntityList);
+
+        // foreach ($sharedProsperityUserEntityList as $sharedProsperityUserEntity) {
+        //     $partnerLevel = 0;
+        //     foreach ($sharedProsperityRecommendConfigEntityList as $sharedProsperityRecommendConfigEntity) {
+        //         if ($sharedProsperityUserEntity->getPv() >= $sharedProsperityRecommendConfigEntity->getPv()) {
+        //             $partnerLevel = $sharedProsperityRecommendConfigEntity->getLevel();
+        //         } else {
+        //             // 如果本次验证的等级结果大于 之前的等级，说明用户升级了 需要更新等级
+        //             if ($partnerLevel > $sharedProsperityUserEntity->getPartnerLevel()) {
+        //                 $this->updateByEntity(
+        //                     SharedProsperityUserEntity::newInstance()
+        //                         ->setId($sharedProsperityUserEntity->getId())
+        //                         ->setPartnerLevel($partnerLevel)
+        //                 );
+        //             }
+        //             break;
+        //         }
+        //     }
+        // }
+    }
+
+    /**
+     * 批量更新用户等级
+     * @param SharedProsperityUserEntity[] $sharedProsperityUserEntityList 用户实体列表
+     * @param SharedProsperityRecommendConfigEntity[] $sharedProsperityRecommendConfigEntityList 等级配置列表
+     * @return void
+     */
+    private function batchUpgradeUsers(array $sharedProsperityUserEntityList, array $sharedProsperityRecommendConfigEntityList)
+    {
         foreach ($sharedProsperityUserEntityList as $sharedProsperityUserEntity) {
-            $partnerLevel = 0;
-            foreach ($sharedProsperityRecommendConfigEntityList as $sharedProsperityRecommendConfigEntity) {
-                if ($sharedProsperityUserEntity->getPv() >= $sharedProsperityRecommendConfigEntity->getPv()) {
-                    $partnerLevel = $sharedProsperityRecommendConfigEntity->getLevel();
-                } else {
-                    // 如果本次验证的等级结果大于 之前的等级，说明用户升级了 需要更新等级
-                    if ($partnerLevel > $sharedProsperityUserEntity->getPartnerLevel()) {
-                        $this->updateByEntity(
-                            SharedProsperityUserEntity::newInstance()
-                                ->setId($sharedProsperityUserEntity->getId())
-                                ->setPartnerLevel($partnerLevel)
-                        );
-                    }
-                    break;
+
+            // // 计算用户的团队业绩（向下5级）
+            // $teamPv = $this->calculateTeamPv($sharedProsperityUserEntity->getId());
+
+            // 根据团队业绩确定等级
+            // $newLevel = $this->calculateLevelByTeamPv($teamPv, $sharedProsperityRecommendConfigEntityList);
+            //
+            // // 如果新等级大于当前等级，则更新
+            // if ($newLevel > $sharedProsperityUserEntity->getPartnerLevel()) {
+            //     $this->updateByEntity(
+            //         SharedProsperityUserEntity::newInstance()
+            //             ->setId($sharedProsperityUserEntity->getId())
+            //             ->setPartnerLevel($newLevel)
+            //     );
+            // }
+
+            $this->upgradeUpperLevels($sharedProsperityUserEntity->getUserId(), $sharedProsperityRecommendConfigEntityList);
+        }
+    }
+
+    /**
+     * 向上追溯更新上级身份等级（最多向上5层）
+     * @param int $userId 当前用户ID
+     * @param SharedProsperityRecommendConfigEntity[] $sharedProsperityRecommendConfigEntityList 等级配置列表
+     * @return void
+     */
+    private function upgradeUpperLevels(int $userId, array $sharedProsperityRecommendConfigEntityList)
+    {
+        $maxLevels = SharedProsperityUserEnum::TEAM_DEPTH;
+        $currentUserId = $userId;
+
+        for ($i = 0; $i < $maxLevels; $i++) {
+            // 获取上级用户ID
+            $sharedProsperityUserEntity = $this->getSharedProsperityUserEntityByUserId($currentUserId);
+            $parentId = $sharedProsperityUserEntity->getParentId();
+
+            if (empty($parentId)) {
+                break; // 没有上级，停止追溯
+            }
+
+            // 重新计算上级的团队业绩和等级
+            $this->upgradeSingleUser($parentId, $sharedProsperityRecommendConfigEntityList);
+
+            // 继续向上追溯
+            $currentUserId = $parentId;
+        }
+    }
+
+    /**
+     * 升级单个用户身份
+     * @param int $userId 用户ID
+     * @param SharedProsperityRecommendConfigEntity[] $sharedProsperityRecommendConfigEntityList 等级配置列表
+     * @return void
+     */
+    private function upgradeSingleUser(int $userId, array $sharedProsperityRecommendConfigEntityList)
+    {
+        // 获取用户实体
+        $sharedProsperityUserEntity = $this->getSharedProsperityUserEntityByUserId($userId);
+
+        if (empty($sharedProsperityUserEntity->getId())) {
+            return;
+        }
+
+        // 计算团队业绩
+        $teamPvMap = $this->calculateTeamPv($userId);
+
+        // 当前只计算团队业绩的 小团队业绩
+        $maxKey = array_keys($teamPvMap, max($teamPvMap))[0];
+        unset($teamPvMap[$maxKey]);
+        $teamPvBySmallSum = array_sum($teamPvMap);
+
+        // 计算新等级
+        $newLevel = $this->calculateLevelByTeamPv($teamPvBySmallSum, $sharedProsperityRecommendConfigEntityList);
+
+        // 如果新等级大于当前等级，则更新
+        if ($newLevel > $sharedProsperityUserEntity->getPartnerLevel()) {
+            $this->updateByEntity(
+                SharedProsperityUserEntity::newInstance()
+                    ->setId($sharedProsperityUserEntity->getId())
+                    ->setPartnerLevel($newLevel)
+            );
+        }
+    }
+
+    /**
+     * 计算用户的团队业绩（向下5级）
+     * @param int $userId 用户ID
+     * @return array 团队业绩
+     */
+    private function calculateTeamPv(int $userId): array
+    {
+        $teamPvMap = [];
+
+        // 获取所有下级用户ID（向下5级）
+        // 第一层级在这里查询 因为要统计大小区
+        $sharedProsperityUserEntityListByParentId = $this->getSharedProsperityUserEntityListByParentId($userId);
+
+        // 循环 直属下级
+        foreach ($sharedProsperityUserEntityListByParentId as $sharedProsperityUserEntityByUnderling) {
+            $departmentTeamUserIdList = $this->getUserIdListByParentIdForDepth($sharedProsperityUserEntityByUnderling->getUserId(), SharedProsperityUserEnum::TEAM_DEPTH - 1);
+            $departmentTeamUserIdList[] = $sharedProsperityUserEntityByUnderling->getUserId();
+
+            // 直属团队的 业绩总和
+            $teamPv = '0.00';
+            if (!empty($departmentTeamUserIdList)) {
+                // 批量获取下级用户的个人业绩并求和
+                // 假设有方法批量获取用户的PV
+                $sharedProsperityUserEntityListByTeam = $this->getSharedProsperityUserEntityListByUserIdList($departmentTeamUserIdList);
+                foreach ($sharedProsperityUserEntityListByTeam as $sharedProsperityUserEntityByTeam) {
+                    $teamPv = bcadd($teamPv, (string)$sharedProsperityUserEntityByTeam->getPv(), 2);
                 }
             }
+            $teamPvMap[$sharedProsperityUserEntityByUnderling->getUserId()] = $teamPv;
         }
+
+        return $teamPvMap;
+    }
+
+
+    /**
+     * 根据团队业绩计算等级
+     * @param float $teamPv 团队业绩
+     * @param array $configEntityList 等级配置列表
+     * @return int 等级
+     */
+    private function calculateLevelByTeamPv(float $teamPv, array $configEntityList): int
+    {
+        $level = 0;
+
+        foreach ($configEntityList as $configEntity) {
+            if ($teamPv >= $configEntity->getPv()) {
+                $level = $configEntity->getLevel();
+            } else {
+                break;
+            }
+        }
+
+        return $level;
+    }
+
+    /**
+     * 获取下级用户ID列表（向下指定层级）
+     * @param int $parentId 用户ID
+     * @param int $maxLevel 最大层级数
+     * @param int $currentLevel 当前层级（内部递归使用）
+     * @return array 下级用户ID列表
+     */
+    private function getUserIdListByParentIdForDepth(int $parentId, int $maxLevel, int $currentLevel = 1): array
+    {
+        if ($currentLevel > $maxLevel) {
+            return [];
+        }
+
+        $subordinateIds = [];
+
+        // 获取直接下级用户ID列表（假设有方法获取直接下级）
+        $sharedProsperityUserEntityListByParentId = $this->getSharedProsperityUserEntityListByParentId($parentId);
+
+        if (empty($sharedProsperityUserEntityListByParentId)) {
+            return [];
+        }
+
+        // 添加直接下级
+        $subordinateIds = array_merge($subordinateIds, array_column($sharedProsperityUserEntityListByParentId, 'userId'));
+
+        // 递归获取下级的下级
+        foreach ($sharedProsperityUserEntityListByParentId as $sharedProsperityUserEntity) {
+            $nestedSubordinateIds = $this->getUserIdListByParentIdForDepth($sharedProsperityUserEntity->getUserId(), $maxLevel, $currentLevel + 1);
+            if (!empty($nestedSubordinateIds)) {
+                $subordinateIds = array_merge($subordinateIds, $nestedSubordinateIds);
+            }
+        }
+
+        return array_unique($subordinateIds);
     }
 }