创建订单-修复下单问题

app/common/dao/shared/SharedProsperityUserDao.php

@@ -19,6 +19,31 @@ class SharedProsperityUserDao extends BaseDao
     }
 
     /**
+     * @param array $idList
+     * @return SharedProsperityUserEntity[]
+     */
+    public function getSharedProsperityUserEntityListByIdList(array $idList): array
+    {
+        $entityList = [];
+        try {
+            /** @var SharedProsperityUser $model */
+            $model = $this->getModel();
+            $dataRes = $model::getDB()
+                ->whereIn('id', $idList)
+                ->select()
+                ->toArray();
+            if (!empty($dataRes)) {
+                $entityList = array_map(function ($data) {
+                    return SharedProsperityUserEntity::newInstance($data);
+                }, $dataRes);
+            }
+        } catch (\Exception $e) {
+
+        }
+        return $entityList;
+    }
+
+    /**
      * @param array $userIdList
      * @return SharedProsperityUserEntity[]
      */
@@ -269,7 +294,7 @@ class SharedProsperityUserDao extends BaseDao
                 ->field('partner_level, user_id')
                 ->select()
                 ->toArray();
-            
+
             $grouped = [];
             foreach ($dataRes as $row) {
                 $level = $row['partner_level'];

app/common/repositories/shared/SharedProsperityUserRepository.php

@@ -24,6 +24,31 @@ class SharedProsperityUserRepository extends BaseRepository
     }
 
     /**
+     * @param int $id
+     * @return SharedProsperityUserEntity
+     */
+    public function getSharedProsperityUserEntityById(int $id): SharedProsperityUserEntity
+    {
+        $entityList = $this->getSharedProsperityUserEntityListByIdList([$id]);
+        $entity = array_shift($entityList);
+        if ($entity instanceof SharedProsperityUserEntity) {
+            return $entity;
+        } else {
+            /** @var SharedProsperityUserEntity */
+            return SharedProsperityUserEntity::newInstance();
+        }
+    }
+
+    /**
+     * @param array $idList
+     * @return SharedProsperityUserEntity[]
+     */
+    public function getSharedProsperityUserEntityListByIdList(array $idList): array
+    {
+        return $this->dao->getSharedProsperityUserEntityListByIdList($idList);
+    }
+
+    /**
      * @param int $userId
      * @return SharedProsperityUserEntity
      */
@@ -111,7 +136,7 @@ class SharedProsperityUserRepository extends BaseRepository
         }
         $result = $this->update($sharedProsperityUserEntity->getId(), $sharedProsperityUserEntity->toUnderlineArray());
         if ($result > 0) {
-            return $this->getSharedProsperityUserEntityByUserId($sharedProsperityUserEntity->getUserId());
+            return $this->getSharedProsperityUserEntityById($sharedProsperityUserEntity->getId());
         }
         return SharedProsperityUserEntity::newInstance();
     }