店铺二维码

src/main/java/com/kiisoo/ic/webappmy/controller/WebAppMyController.java

@@ -4,7 +4,9 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.kiisoo.ic.common.BaseController;
 import com.kiisoo.ic.store.constant.Constants;
 import com.kiisoo.ic.store.entity.PoiStoreStaff;
+import com.kiisoo.ic.store.entity.StoreUserDTO;
 import com.kiisoo.ic.store.mapper.PoiStoreStaffDOMapper;
+import com.kiisoo.ic.store.mapper.PrivilageCpUserStoreDOMapper;
 import com.kiisoo.ic.webappmy.vo.StafferInfoVO;
 import com.kiisoo.ic.webappmy.vo.UserParam;
 import lombok.extern.slf4j.Slf4j;
@@ -12,6 +14,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 
@@ -27,6 +30,8 @@ public class WebAppMyController extends BaseController {
     @Autowired
     private PoiStoreStaffDOMapper poiStoreStaffDOMapper;
 
+    @Autowired
+    private PrivilageCpUserStoreDOMapper privilageCpUserStoreDOMapper;
     /**
      * 获取用户信息
      */
@@ -51,14 +56,12 @@ public class WebAppMyController extends BaseController {
     @GetMapping("/get/shop/qr")
     public Map<String,Object> shopIcQr(long shopId){
         try {
-            PoiStoreStaff poiStoreStaff = new PoiStoreStaff();
-            poiStoreStaff.setStoreId(shopId);
-            poiStoreStaff.setType(Constants.TYPE_MANAGER);
-            poiStoreStaff.setStatus(Constants.STATUS_USEFUL);
             String img = null;
-            List<PoiStoreStaff> poiStoreStaffs = poiStoreStaffDOMapper.selectList(new QueryWrapper<>(poiStoreStaff));
-            if(poiStoreStaffs.size() > 0){
-                img = poiStoreStaffs.get(0).getEpWechatQrCode();
+            List<Long> shopIds = new ArrayList<>();
+            shopIds.add(shopId);
+            List<StoreUserDTO> cpUserStoreDO = privilageCpUserStoreDOMapper.selectCpUserByShops(shopIds);
+            if(cpUserStoreDO.size() > 0){
+                img = cpUserStoreDO.get(0).getQrCodeAction();
                 return data(img);
             }else{
                 throw new Exception();