修改镇数据源

huimv-cattle/src/main/java/com/huimv/cattle/service/impl/OverviewServiceImpl.java

@@ -131,12 +131,20 @@ public class OverviewServiceImpl extends ServiceImpl<OverviewMapper, Overview> i
         return new Result(ResultCode.SUCCESS,jsonObject);
     }
 
+    @Override
     public Result getListOverviewScreen(HttpServletRequest httpServletRequest, Map<String, String> paramsMap) {
         String farmCode =paramsMap.get("farmCode");
         if (farmCode==null) {
             farmCode = TokenSign.getFarmCode(httpServletRequest);
         }
-        Overview overview = overviewMapper.getListOverview();
+        Overview overview = new Overview();
+        if (farmCode == null || farmCode.equals(0)){
+            overview = overviewMapper.getListOverview();
+        }else {
+            QueryWrapper<Overview> queryWrapper = new QueryWrapper<>();
+            queryWrapper.eq("farm_code", farmCode);
+            overview = overviewMapper.selectOne(queryWrapper);
+        }
         JSONObject jsonObject = new JSONObject();
         if (ObjectUtil.isEmpty(overview)) {
             jsonObject.put("stockCount", 0);

huimv-cattle/src/main/java/com/huimv/cattle/service/impl/StockServiceImpl.java

@@ -156,7 +156,14 @@ public class StockServiceImpl extends ServiceImpl<StockMapper, Stock> implements
             farmCode = TokenSign.getFarmCode(httpServletRequest);
         }
         JSONObject jsonObject = new JSONObject();
-        Stock stock = stockMapper.getListStock();
+        Stock stock = new Stock();
+        if (farmCode == null || farmCode.equals(0)){
+            stock = stockMapper.getListStock();
+        }else {
+            QueryWrapper<Stock> queryWrapper = new QueryWrapper<>();
+            queryWrapper.eq("farm_code", farmCode);
+            stock = stockMapper.selectOne(queryWrapper);
+        }
         if (ObjectUtil.isEmpty(stock)) {
             jsonObject.put("fatStock", 0);
             jsonObject.put("cowStock", 0);