|
@@ -3,8 +3,10 @@ package com.huimv.cattle.service.impl;
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.huimv.cattle.mapper.StockMapper;
|
|
|
+import com.huimv.cattle.pojo.DataSource;
|
|
|
import com.huimv.cattle.pojo.StockStatus;
|
|
|
import com.huimv.cattle.mapper.StockStatusMapper;
|
|
|
+import com.huimv.cattle.service.DataSourceService;
|
|
|
import com.huimv.cattle.service.StockStatusService;
|
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
|
import com.huimv.cattle.token.TokenSign;
|
|
@@ -14,6 +16,7 @@ import org.apache.commons.lang3.StringUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
+import javax.annotation.Resource;
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
import java.util.Map;
|
|
|
|
|
@@ -30,6 +33,8 @@ public class StockStatusServiceImpl extends ServiceImpl<StockStatusMapper, Stock
|
|
|
|
|
|
@Autowired
|
|
|
private StockStatusMapper stockStatusMapper;
|
|
|
+ @Resource
|
|
|
+ private DataSourceService dataSourceService;
|
|
|
|
|
|
@Override
|
|
|
public Result getStockStatus(HttpServletRequest request, Map<String, String> paramsMap) {
|
|
@@ -76,4 +81,28 @@ public class StockStatusServiceImpl extends ServiceImpl<StockStatusMapper, Stock
|
|
|
}
|
|
|
return new Result(ResultCode.SUCCESS);
|
|
|
}
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public Result getStockStatusScreen(HttpServletRequest request, Map<String, String> paramsMap) {
|
|
|
+ String farmCode = paramsMap.get("farmCode");
|
|
|
+ if (StringUtils.isBlank(farmCode)){
|
|
|
+ farmCode = TokenSign.getFarmCode(request);
|
|
|
+ }
|
|
|
+ DataSource dataSource = dataSourceService.getDataSourceStatus();
|
|
|
+ StockStatus stockStatus = new StockStatus();
|
|
|
+ if (dataSource.getDsStatus() == 1 && dataSource.getViewType() ==2) {
|
|
|
+ stockStatus = stockStatusMapper.getStockStatusScreen();
|
|
|
+ }else{
|
|
|
+ stockStatus = stockStatusMapper.getStockStatus(farmCode);
|
|
|
+ if (ObjectUtil.isEmpty(stockStatus)){
|
|
|
+ stockStatus = new StockStatus();
|
|
|
+ stockStatus.setBreedStock(0);
|
|
|
+ stockStatus.setCalfStock(0);
|
|
|
+ stockStatus.setRaiseStock(0);
|
|
|
+ stockStatus.setFarmCode(farmCode);
|
|
|
+ stockStatusMapper.insert(stockStatus);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return new Result(ResultCode.SUCCESS,stockStatus);
|
|
|
+ }
|
|
|
}
|