张延森 4 年之前
父節點
當前提交
e34baff877
共有 1 個檔案被更改,包括 14 行新增10 行删除
  1. 14
    10
      src/main/java/com/shigongli/controller/TaHouseController.java

+ 14
- 10
src/main/java/com/shigongli/controller/TaHouseController.java 查看文件

126
         }
126
         }
127
 
127
 
128
         if (!StringUtils.isEmpty(taHouse.getShopId())) {
128
         if (!StringUtils.isEmpty(taHouse.getShopId())) {
129
-            boolean found = false;
130
-            for (TaShopKeeper keeper: shopKeeperList) {
131
-                if (taHouse.getShopId().equals(keeper.getShopId())) {
132
-                    found = true;
133
-                }
134
-            }
135
-
136
-            if (!found) {
129
+            if (!isContaints(shopKeeperList, taHouse.getShopId())) {
137
                 return ResponseBean.error("您无权新增房源, 或房源所属民宿出错", ResponseBean.ERROR_UNAVAILABLE);
130
                 return ResponseBean.error("您无权新增房源, 或房源所属民宿出错", ResponseBean.ERROR_UNAVAILABLE);
138
             }
131
             }
139
         } else {
132
         } else {
182
             return ResponseBean.error("店主数据异常", ResponseBean.ERROR_UNAVAILABLE);
175
             return ResponseBean.error("店主数据异常", ResponseBean.ERROR_UNAVAILABLE);
183
         }
176
         }
184
 
177
 
185
-        TaShopKeeper taShopKeeper = shopKeeperList.get(0);
186
-        if (!taShopKeeper.getShopId().equals(taHouse.getShopId())) {
178
+        if (!isContaints(shopKeeperList, taHouse.getShopId())) {
187
             return ResponseBean.error("您无权进行当前操作", ResponseBean.ERROR_UNAVAILABLE);
179
             return ResponseBean.error("您无权进行当前操作", ResponseBean.ERROR_UNAVAILABLE);
188
         }
180
         }
189
 
181
 
275
         return city;
267
         return city;
276
     }
268
     }
277
 
269
 
270
+    private boolean isContaints(List<TaShopKeeper> shopKeepers, String shopId) {
271
+        if (null == shopKeepers) return false;
272
+
273
+        for (TaShopKeeper shopKeeper : shopKeepers) {
274
+            if (shopKeeper.getShopId().equals(shopId)) {
275
+                return true;
276
+            }
277
+        }
278
+
279
+        return false;
280
+    }
281
+
278
 }
282
 }