存栏状态

huimv-cattle/src/main/java/com/huimv/cattle/controller/StockStatusController.java

@@ -1,10 +1,18 @@
 package com.huimv.cattle.controller;
 
 
+import com.huimv.cattle.pojo.StockStatus;
+import com.huimv.cattle.service.StockStatusService;
+import com.huimv.common.utils.Result;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 
 import org.springframework.web.bind.annotation.RestController;
 
+import java.util.Map;
+
 /**
  * <p>
  *  前端控制器
@@ -16,6 +24,18 @@ import org.springframework.web.bind.annotation.RestController;
 @RestController
 @RequestMapping("/stock-status")
 public class StockStatusController {
+    @Autowired
+    private StockStatusService stockStatusService;
+
+    @PostMapping("/getStockStatus")
+    public Result getStockStatus(){
+        return stockStatusService.getStockStatus();
+    }
 
+    @PostMapping("/saveStockStatus")
+    public Result saveStockStatus(@RequestBody StockStatus stockStatus){
+        return stockStatusService.saveStockStatus(stockStatus);
+    }
+    
 }
 

huimv-cattle/src/main/java/com/huimv/cattle/mapper/StockStatusMapper.java

@@ -2,6 +2,7 @@ package com.huimv.cattle.mapper;
 
 import com.huimv.cattle.pojo.StockStatus;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.springframework.stereotype.Repository;
 
 /**
  * <p>
@@ -11,6 +12,7 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper;
  * @author zn
  * @since 2022-12-14
  */
+@Repository
 public interface StockStatusMapper extends BaseMapper<StockStatus> {
 
 }

huimv-cattle/src/main/java/com/huimv/cattle/pojo/StockStatus.java

@@ -21,13 +21,16 @@ public class StockStatus implements Serializable {
 
     private static final long serialVersionUID=1L;
 
-      @TableId(value = "id", type = IdType.AUTO)
+    @TableId(value = "id", type = IdType.AUTO)
     private Integer id;
 
+    //种牛存栏
     private Integer breedStock;
 
+    //犊牛存栏
     private Integer calfStock;
 
+    //育成牛存栏
     private Integer raiseStock;
 
 

huimv-cattle/src/main/java/com/huimv/cattle/service/StockStatusService.java

@@ -2,6 +2,9 @@ package com.huimv.cattle.service;
 
 import com.huimv.cattle.pojo.StockStatus;
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.huimv.common.utils.Result;
+
+import java.util.Map;
 
 /**
  * <p>
@@ -13,4 +16,7 @@ import com.baomidou.mybatisplus.extension.service.IService;
  */
 public interface StockStatusService extends IService<StockStatus> {
 
+    Result getStockStatus();
+
+    Result saveStockStatus(StockStatus stockStatus);
 }

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

@@ -1,11 +1,19 @@
 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.StockStatus;
 import com.huimv.cattle.mapper.StockStatusMapper;
 import com.huimv.cattle.service.StockStatusService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.huimv.common.utils.Result;
+import com.huimv.common.utils.ResultCode;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.Map;
+
 /**
  * <p>
  *  服务实现类
@@ -17,4 +25,27 @@ import org.springframework.stereotype.Service;
 @Service
 public class StockStatusServiceImpl extends ServiceImpl<StockStatusMapper, StockStatus> implements StockStatusService {
 
+    @Autowired
+    private StockStatusMapper stockStatusMapper;
+
+    @Override
+    public Result getStockStatus() {
+        QueryWrapper<StockStatus> queryWrapper = new QueryWrapper<>();
+        queryWrapper.orderByDesc("id").last("limit 1");
+        StockStatus stockStatus = stockStatusMapper.selectOne(queryWrapper);
+        if (ObjectUtil.isEmpty(stockStatus)){
+            stockStatus = new StockStatus();
+            stockStatus.setBreedStock(0);
+            stockStatus.setCalfStock(0);
+            stockStatus.setRaiseStock(0);
+            stockStatusMapper.insert(stockStatus);
+        }
+        return new Result(ResultCode.SUCCESS,stockStatus);
+    }
+
+    @Override
+    public Result saveStockStatus(StockStatus stockStatus) {
+        stockStatusMapper.insert(stockStatus);
+        return new Result(ResultCode.SUCCESS);
+    }
 }