swagger mysql

src/main/java/com/kiisoo/ic/activity/service/impl/PmnActivityInstanceServiceImpl.java

@@ -12,10 +12,17 @@ import com.kiisoo.ic.activity.mapper.PmnActivityInstanceDOMapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.kiisoo.ic.domain.service.PrivilageDomainService;
 import com.kiisoo.ic.generalize.entity.PrivilageOrganizational;
+import com.kiisoo.ic.generalize.entity.PrivilageOrganizationalMember;
+import com.kiisoo.ic.generalize.mapper.OrganizationalMemberMapper;
 import com.kiisoo.ic.generalize.mapper.RetailCompanyMapper;
 import com.kiisoo.ic.generalize.service.RetailCompanyService;
 import com.kiisoo.ic.store.entity.PoiStore;
 import com.kiisoo.ic.store.mapper.PoiStoreDOMapper;
+import com.kiisoo.ic.system.entity.PrivilageAccountDO;
+import com.kiisoo.ic.system.entity.PrivilageRoleDO;
+import com.kiisoo.ic.system.enums.RoleEnum;
+import com.kiisoo.ic.system.mapper.PrivilageAccountDOMapper;
+import com.kiisoo.ic.system.mapper.PrivilageRoleDOMapper;
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.ApplicationContext;
@@ -69,6 +76,12 @@ public class PmnActivityInstanceServiceImpl extends ServiceImpl<PmnActivityInsta
 
     @Autowired
     private PoiStoreDOMapper poiStoreDOMapper;
+    @Autowired
+    private PrivilageRoleDOMapper privilageRoleDOMapper;
+    @Autowired
+    private PrivilageAccountDOMapper privilageAccountDOMapper;
+    @Autowired
+    private OrganizationalMemberMapper organizationalMemberMapper;
 
     @Transactional
     public boolean saveVO(ActivityInstanceVO activityInstanceVO, boolean isTerminate) {
@@ -163,7 +176,16 @@ public class PmnActivityInstanceServiceImpl extends ServiceImpl<PmnActivityInsta
     public Map<String, Object> pageVO(Long userId, Date startDate, Date endDate, Long organizationId, Long storeId, Integer status, Integer pageNum, Integer pageSize, Date beginTime, Date endTime) {
         Map<String, Object> results = new HashMap<>();
         List<Long> shopIds = new ArrayList<>();
-
+        if(userId != null) {
+            PrivilageAccountDO privilageAccountDO = privilageAccountDOMapper.selectOne(Wrappers.<PrivilageAccountDO>lambdaQuery().eq(PrivilageAccountDO::getUserId, userId));
+            PrivilageRoleDO roleDO = privilageRoleDOMapper.selectRoleName(privilageAccountDO.getUserId());
+            System.out.println(roleDO.getCode());
+            System.out.println(RoleEnum.getCodeByName("运营人员").getRoleCode());
+            if(roleDO.getCode().equals(RoleEnum.getCodeByName("运营人员").getRoleCode())) {
+                PrivilageOrganizationalMember organizationalMember = organizationalMemberMapper.selectOne(Wrappers.<PrivilageOrganizationalMember>lambdaQuery().eq(PrivilageOrganizationalMember::getUserId, userId));
+                organizationId = organizationalMember.getOrgId();
+            }
+        }
 //        if(userId != null) {
 //            if (storeId != null) {
 //                shopIds.add(storeId);

src/main/java/com/kiisoo/ic/store/controller/PoiStoreController.java

@@ -29,9 +29,9 @@ public class PoiStoreController extends BaseController {
     private PoiStoreService poiStoreService;
 
 
-
     /**
      * 根据用户id，区域id获取店铺
+     *
      * @return
      */
     @RequestMapping(value = "/user/shop", method = RequestMethod.POST)
@@ -61,11 +61,12 @@ public class PoiStoreController extends BaseController {
 
     /**
      * 根据用户id，区域id获取店铺
+     *
      * @return
      */
     @RequestMapping(value = "/user/shopDTO", method = RequestMethod.POST)
     @ResponseBody
-    public Map<String,Object> getUserShopDTO(@RequestParam("userId")long userId, @RequestParam(value = "customerIds",required =false)String customerIds){
+    public Map<String, Object> getUserShopDTO(@RequestParam("userId") long userId, @RequestParam(value = "customerIds", required = false) String customerIds, @RequestParam("scheduleId") long scheduleId) {
         try {
             List<Long> customers = null;
             if (StringUtils.isNotBlank(customerIds)) {
@@ -80,13 +81,14 @@ public class PoiStoreController extends BaseController {
                     }
                 }
             }
-            return data(poiStoreService.getRegionShopDTO(userId,customers));
+            return data(poiStoreService.getRegionShopDTO(userId, customers, scheduleId));
         } catch (Exception e) {
             log.error("获取用户店铺失败", e);
             return fail();
         }
 
     }
+
     /**
      * 查询所有店铺
      *

src/main/java/com/kiisoo/ic/store/service/PoiStoreService.java

@@ -2,6 +2,12 @@ package com.kiisoo.ic.store.service;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.kiisoo.ic.activity.entity.PmnActivityInstance;
+import com.kiisoo.ic.activity.entity.PmnActivityStoreRelation;
+import com.kiisoo.ic.activity.entity.PmnPromotionSrc;
+import com.kiisoo.ic.activity.service.IPmnActivityInstanceService;
+import com.kiisoo.ic.activity.service.IPmnActivityStoreRelationService;
+import com.kiisoo.ic.activity.service.IPmnPromotionSrcService;
 import com.kiisoo.ic.customer.service.CustomerViewService;
 import com.kiisoo.ic.domain.entity.PrivilageDomainEntityDO;
 import com.kiisoo.ic.domain.mapper.PrivilageDomainEntityDOMapper;
@@ -53,6 +59,10 @@ public class PoiStoreService {
     private PrivilageOrganizationalDomainMapper privilageOrganizationalDomainMapper;
     @Autowired
     private RetailCompanyMapper retailCompanyMapper;
+    @Autowired
+    private IPmnActivityInstanceService pmnActivityInstanceService;
+    @Autowired
+    private IPmnActivityStoreRelationService storeRelationService;
     /**
      * 根据区域id找到店铺集合
      * @param regionId
@@ -96,17 +106,27 @@ public class PoiStoreService {
      * 根据区域id找到店铺集合，若没有，则查用户所有店铺
      * @param regionId
      */
-    public List<PoiStoreDTO> getRegionShopDTO(long userId,List<Long> customerIds){
+    public List<PoiStoreDTO> getRegionShopDTO(long userId,List<Long> customerIds, Long scheduleId){
         List<PoiStoreDTO> poiStores = new ArrayList<>();
         //找到店铺集合
         if(null != customerIds && customerIds.size()>0){
             Set<Long> cs = new HashSet<>();
             cs.addAll(customerIds);
+            Set<Long> joined = new HashSet<>();
+            if(scheduleId != null) {
+                List<PmnActivityInstance> instances = pmnActivityInstanceService.list(Wrappers.<PmnActivityInstance>lambdaQuery().eq(PmnActivityInstance::getScheduleId, scheduleId).select(PmnActivityInstance::getId));
+                Set<Long> instanceIds = instances.stream().map(PmnActivityInstance::getId).collect(Collectors.toSet());
+                List<PmnActivityStoreRelation> relations = storeRelationService.list(Wrappers.<PmnActivityStoreRelation>lambdaQuery().in(PmnActivityStoreRelation::getInstanceId, instanceIds).select(PmnActivityStoreRelation::getStoreId));
+                joined = relations.stream().map(PmnActivityStoreRelation::getStoreId).collect(Collectors.toSet());
+            }
             Set<Long> childs = retailCompanyMapper.selectChildList(cs);
             List<PrivilageOrganizationalDomain> domains = privilageOrganizationalDomainMapper.selectList(Wrappers.<PrivilageOrganizationalDomain>lambdaQuery().in(PrivilageOrganizationalDomain::getOrgId, childs).select(PrivilageOrganizationalDomain::getDomainId));
             Set<Long> domainIds = domains.stream().map(PrivilageOrganizationalDomain::getDomainId).collect(Collectors.toSet());
             List<PrivilageDomainEntityDO> entityDOS = privilageDomainEntityDOMapper.selectList(Wrappers.<PrivilageDomainEntityDO>lambdaQuery().in(PrivilageDomainEntityDO::getDomainId, domainIds).eq(PrivilageDomainEntityDO::getType, 3).select(PrivilageDomainEntityDO::getEntityId));
             Set<Long> shopIds = entityDOS.stream().map(PrivilageDomainEntityDO::getEntityId).collect(Collectors.toSet());
+            if(joined.size() > 0) {
+                shopIds.removeAll(joined);
+            }
             if(shopIds.size() > 0){
                 poiStores = poiStoreDOMapper.selectDTOBatchIds(shopIds);
             }