Просмотр исходного кода

更新分配结果查询方法

hejq 7 лет назад
Родитель
Сommit
5ea76af67d

+ 9 - 1
src/main/java/com/uas/platform/b2b/dao/VendorDistributeDao.java

@@ -25,7 +25,15 @@ public interface VendorDistributeDao extends JpaSpecificationExecutor<VendorDist
 	 * @return List<Distribute>
 	 */
 	@Query("select d from VendorDistribute d where d.vendorId in (:vendorIds)")
-	List<VendorDistribute> findByVendorId(@Param("vendorIds") List<Long> vendorIds);
+	List<VendorDistribute> findByVendorIdList(@Param("vendorIds") List<Long> vendorIds);
+
+    /**
+     * 通过供应商表id查询客户分配
+     *
+     * @param veId 供应商关系表id
+     * @return 查询分配结果
+     */
+    List<VendorDistribute> findByVendorId(Long veId);
 
     /**
      * 通过用户UU和客户供应商关系查询分配是否存在

+ 2 - 3
src/main/java/com/uas/platform/b2b/service/impl/VendorsServiceImpl.java

@@ -40,7 +40,6 @@ import javax.persistence.criteria.CriteriaBuilder;
 import javax.persistence.criteria.CriteriaQuery;
 import javax.persistence.criteria.Root;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.Date;
 import java.util.HashSet;
 import java.util.Iterator;
@@ -477,7 +476,7 @@ public class VendorsServiceImpl implements VendorService {
      */
     @Override
     public void setSaleContacts(List<Vendor> vendorList, List<Long> idList) {
-        List<VendorDistribute> distributeList = vendorDistributeDao.findByVendorId(idList);
+        List<VendorDistribute> distributeList = vendorDistributeDao.findByVendorIdList(idList);
         List<VendorDistribute> distributeListCopy = new ArrayList<>();
         distributeListCopy.addAll(distributeList);
         if (!CollectionUtil.isEmpty(vendorList)) {
@@ -604,7 +603,7 @@ public class VendorsServiceImpl implements VendorService {
      */
     @Override
     public List<User> findUserInfoVendDistribute(Long veId) {
-        List<VendorDistribute> distributes = vendorDistributeDao.findByVendorId(Arrays.asList(veId));
+        List<VendorDistribute> distributes = vendorDistributeDao.findByVendorId(veId);
         List<User> users = userService.findUsersByEnUU(SystemSession.getUser().getEnterprise().getUu());
         users.forEach(user -> {
             distributes.forEach(distribute -> {