修改存栏状态

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

@@ -15,4 +15,5 @@ import org.springframework.stereotype.Repository;
 @Repository
 public interface StockStatusMapper extends BaseMapper<StockStatus> {
 
+    StockStatus getStockStatus();
 }

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

@@ -15,4 +15,8 @@
         id, breed_stock, calf_stock, raise_stock
     </sql>
 
+    <select id="getStockStatus" resultType="com.huimv.cattle.pojo.StockStatus">
+        select * from stock_status
+    </select>
+
 </mapper>

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

@@ -30,9 +30,7 @@ public class StockStatusServiceImpl extends ServiceImpl<StockStatusMapper, Stock
 
     @Override
     public Result getStockStatus() {
-        QueryWrapper<StockStatus> queryWrapper = new QueryWrapper<>();
-        queryWrapper.orderByDesc("id").last("limit 1");
-        StockStatus stockStatus = stockStatusMapper.selectOne(queryWrapper);
+        StockStatus stockStatus = stockStatusMapper.getStockStatus();
         if (ObjectUtil.isEmpty(stockStatus)){
             stockStatus = new StockStatus();
             stockStatus.setBreedStock(0);
@@ -45,7 +43,15 @@ public class StockStatusServiceImpl extends ServiceImpl<StockStatusMapper, Stock
 
     @Override
     public Result saveStockStatus(StockStatus stockStatus) {
-        stockStatusMapper.insert(stockStatus);
+        if (ObjectUtil.isEmpty(stockStatusMapper.selectById(stockStatus))){
+            StockStatus stockStatus1 = new StockStatus();
+            stockStatus1.setBreedStock(stockStatus.getBreedStock());
+            stockStatus1.setRaiseStock(stockStatus.getRaiseStock());
+            stockStatus1.setCalfStock(stockStatus.getCalfStock());
+            stockStatusMapper.insert(stockStatus1);
+        }else {
+            stockStatusMapper.updateById(stockStatus);
+        }
         return new Result(ResultCode.SUCCESS);
     }
 }