|
@@ -4,13 +4,19 @@ package com.huimv.cattle.controller;
|
|
|
import cn.hutool.core.date.DateUtil;
|
|
|
import cn.hutool.core.util.ObjectUtil;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
+import com.huimv.cattle.pojo.FarmStockRank;
|
|
|
import com.huimv.cattle.pojo.VaccineDelivery;
|
|
|
+import com.huimv.cattle.pojo.vo.VaccineDeliveryVo;
|
|
|
import com.huimv.cattle.service.VaccineDeliveryService;
|
|
|
+import com.huimv.cattle.utils.FarmCodeUtils;
|
|
|
import com.huimv.common.utils.Result;
|
|
|
import com.huimv.common.utils.ResultCode;
|
|
|
+import org.apache.commons.lang3.StringUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.transaction.annotation.Transactional;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
+import javax.servlet.http.HttpServletRequest;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.Date;
|
|
|
import java.util.List;
|
|
@@ -32,14 +38,14 @@ public class VaccineDeliveryController {
|
|
|
private VaccineDeliveryService vaccineDeliveryService;
|
|
|
|
|
|
@PostMapping("/saveVaccineDelivery")
|
|
|
- public Result add(@RequestBody VaccineDelivery vaccineDelivery){
|
|
|
- Integer year = vaccineDelivery.getYear();
|
|
|
- VaccineDelivery year1 = vaccineDeliveryService.getOne(new QueryWrapper<VaccineDelivery>().eq("year", year));
|
|
|
- if (ObjectUtil.isNotEmpty(year1)){
|
|
|
- year1.setDeliveryCount(vaccineDelivery.getDeliveryCount());
|
|
|
- year1.setYear(vaccineDelivery.getYear());
|
|
|
- vaccineDeliveryService.updateById(year1);
|
|
|
- }else {
|
|
|
+ @Transactional
|
|
|
+ public Result add(@RequestBody VaccineDeliveryVo vaccineDeliveryVo, HttpServletRequest request){
|
|
|
+ String farmCode = FarmCodeUtils.getFarmCode(vaccineDeliveryVo.getFarmCode(), request);
|
|
|
+ List<VaccineDelivery> vaccineDeliveries = vaccineDeliveryVo.getVaccineDeliveryList();
|
|
|
+ vaccineDeliveryService.remove(new QueryWrapper<VaccineDelivery>().eq("farm_code",farmCode));
|
|
|
+ System.out.println(farmCode);
|
|
|
+ for (VaccineDelivery vaccineDelivery : vaccineDeliveries) {
|
|
|
+ vaccineDelivery.setFarmCode(farmCode);
|
|
|
vaccineDeliveryService.save(vaccineDelivery);
|
|
|
}
|
|
|
return new Result(10000,"添加成功",true);
|
|
@@ -68,10 +74,14 @@ public class VaccineDeliveryController {
|
|
|
}
|
|
|
|
|
|
@PostMapping("/getVaccineDelivery")
|
|
|
- public Result getVaccineDelivery(@RequestBody Map<String,Integer> paramMap){
|
|
|
- Integer yearNum = paramMap.get("yearNum");
|
|
|
+ public Result getVaccineDelivery(@RequestBody Map<String,String> paramMap, HttpServletRequest request){
|
|
|
+ String yearNum = paramMap.get("yearNum");
|
|
|
+ String farmCode = FarmCodeUtils.getFarmCode(paramMap.get("farmCode"), request);
|
|
|
+ if (StringUtils.isBlank(yearNum)){
|
|
|
+ yearNum ="0";
|
|
|
+ }
|
|
|
int year = DateUtil.year(new Date());
|
|
|
- List<VaccineDelivery> list = vaccineDeliveryService.list(new QueryWrapper<VaccineDelivery>().ge("year", (year - yearNum)));
|
|
|
+ List<VaccineDelivery> list = vaccineDeliveryService.list(new QueryWrapper<VaccineDelivery>().eq("farm_code",farmCode).ge("year", (year - Integer.parseInt(yearNum))));
|
|
|
return new Result(ResultCode.SUCCESS,list);
|
|
|
}
|
|
|
}
|