门店标签回写

src/main/java/com/gic/plug/web/controller/StoreController.java

@@ -10,6 +10,7 @@ import com.gic.auth.service.ResourceApiService;
 import com.gic.auth.service.UserResourceApiService;
 import com.gic.commons.util.EntityUtil;
 import com.gic.commons.webapi.reponse.RestResponse;
+import com.gic.enterprise.utils.UserDetail;
 import com.gic.enterprise.utils.UserDetailUtils;
 import com.gic.plug.web.qo.StoreWidgetCountQO;
 import com.gic.plug.web.qo.StoreWidgetQO;
@@ -34,6 +35,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -59,10 +61,45 @@ public class StoreController {
 
     @RequestMapping("/list-store-region")
     public RestResponse listStoreRegion(String search) {
-        Integer enterpriseId = UserDetailUtils.getUserDetail().getEnterpriseId();
+        UserDetail userDetail = UserDetailUtils.getUserDetail();
+        Integer enterpriseId = userDetail.getEnterpriseId();
+        String[] authRegionArr = null;
+        if(userDetail.getUserInfo().getSuperAdmin() != 1 && userDetail.getUserResourceInfo() != null){
+            Long storeResource = userDetail.getUserResourceInfo().getStoreResource();
+            if(storeResource != null){
+                ServiceResponse<StoreWidgetDTO> storeWidget = this.storeWidgetApiService.getStoreWidget(storeResource.intValue());
+                if(storeWidget.getResult() != null){
+                    String searchParam = storeWidget.getResult().getSearchParam();
+                    JSONObject json = JSON.parseObject(searchParam);
+                    JSONArray list = json.getJSONArray("list");
+                    for(int i=0; i< list.size(); i++){
+                        JSONObject o = list.getJSONObject(i);
+                        if(o.containsKey(StoreESFieldsEnum.REGIONID.getField())){
+                            authRegionArr = o.getString("value").split(" ");
+                        }
+                    }
+                }
+            }
+        }
         ServiceResponse<List<StoreRegionDTO>> serviceResponse = storeRegionApiService.listStoreRegion(enterpriseId, search);
         if (serviceResponse.isSuccess()) {
             List<StoreRegionDTO> result = serviceResponse.getResult();
+            List<StoreRegionVO> regionList = new ArrayList<>();
+            for(StoreRegionDTO storeRegionDTO : result){
+                if(authRegionArr != null){
+                    boolean exist = false;
+                    for(String s : authRegionArr){
+                        if(storeRegionDTO.getRegionId() == Integer.valueOf(s)){
+                            exist = true;
+                        }
+                        if(!exist){
+                            regionList.add(EntityUtil.changeEntityByJSON(StoreRegionVO.class, storeRegionDTO));
+                        }
+                    }
+                }else {
+                    regionList.add(EntityUtil.changeEntityByJSON(StoreRegionVO.class, storeRegionDTO));
+                }
+            }
             return RestResponse.success(EntityUtil.changeEntityListByOrika(StoreRegionVO.class, result));
         } else {
             return RestResponse.failure(serviceResponse.getCode(), serviceResponse.getMessage());