lyhzzz il y a 4 mois
Parent
commit
8360b7bef1

+ 4 - 0
src/main/java/com/fdkankan/ucenter/controller/ProductController.java

@@ -32,5 +32,9 @@ public class ProductController extends BaseController {
         return Result.success(productService.getByKey(typeKey));
     }
 
+    @GetMapping("/getFreeCount/{num}")
+    public Result getFreeCount(@PathVariable String num){
+        return Result.success(productService.getFreeCount(num));
+    }
 }
 

+ 3 - 0
src/main/java/com/fdkankan/ucenter/service/IProductService.java

@@ -2,6 +2,7 @@ package com.fdkankan.ucenter.service;
 
 import com.fdkankan.ucenter.entity.Product;
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.fdkankan.ucenter.entity.SceneCooperationCount;
 
 /**
  * <p>
@@ -14,4 +15,6 @@ import com.baomidou.mybatisplus.extension.service.IService;
 public interface IProductService extends IService<Product> {
 
     Product getByKey(String key);
+
+    Integer getFreeCount(String num);
 }

+ 2 - 0
src/main/java/com/fdkankan/ucenter/service/ISceneCooperationCountService.java

@@ -18,4 +18,6 @@ public interface ISceneCooperationCountService extends IService<SceneCooperation
     void saveCount(List<String> numList, Integer size);
 
      List<SceneCooperationCount> getByNumList(List<String> numList);
+
+    SceneCooperationCount getByNum(String num);
 }

+ 15 - 0
src/main/java/com/fdkankan/ucenter/service/impl/ProductServiceImpl.java

@@ -2,9 +2,13 @@ package com.fdkankan.ucenter.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.fdkankan.ucenter.entity.Product;
+import com.fdkankan.ucenter.entity.SceneCooperation;
+import com.fdkankan.ucenter.entity.SceneCooperationCount;
 import com.fdkankan.ucenter.mapper.IProductMapper;
 import com.fdkankan.ucenter.service.IProductService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.fdkankan.ucenter.service.ISceneCooperationCountService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
@@ -20,6 +24,8 @@ import java.util.List;
 @Service
 public class ProductServiceImpl extends ServiceImpl<IProductMapper, Product> implements IProductService {
 
+    @Autowired
+    ISceneCooperationCountService sceneCooperationCountService;
     @Override
     public Product getByKey(String key) {
         LambdaQueryWrapper<Product> wrapper = new LambdaQueryWrapper<>();
@@ -30,4 +36,13 @@ public class ProductServiceImpl extends ServiceImpl<IProductMapper, Product> imp
         }
         return list.get(0);
     }
+
+    @Override
+    public Integer getFreeCount(String num) {
+        SceneCooperationCount byNum = sceneCooperationCountService.getByNum(num);
+        if(byNum == null){
+            return 1;
+        }
+        return byNum.getCount();
+    }
 }

+ 11 - 0
src/main/java/com/fdkankan/ucenter/service/impl/SceneCooperationCountServiceImpl.java

@@ -71,4 +71,15 @@ public class SceneCooperationCountServiceImpl extends ServiceImpl<ISceneCooperat
         wrapper.in(SceneCooperationCount::getNum,numList);
         return this.list(wrapper);
     }
+
+    @Override
+    public SceneCooperationCount getByNum(String num) {
+        LambdaQueryWrapper<SceneCooperationCount> wrapper = new LambdaQueryWrapper<>();
+        wrapper.eq(SceneCooperationCount::getNum,num);
+        List<SceneCooperationCount> list = this.list(wrapper);
+        if(list == null || list.isEmpty()){
+            return null;
+        }
+        return list.get(0);
+    }
 }