diff --git a/src/main/java/com/kiisoo/ic/webappmy/controller/WebAppMyController.java b/src/main/java/com/kiisoo/ic/webappmy/controller/WebAppMyController.java index 9f7434d..dec825e 100644 --- a/src/main/java/com/kiisoo/ic/webappmy/controller/WebAppMyController.java +++ b/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 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 poiStoreStaffs = poiStoreStaffDOMapper.selectList(new QueryWrapper<>(poiStoreStaff)); - if(poiStoreStaffs.size() > 0){ - img = poiStoreStaffs.get(0).getEpWechatQrCode(); + List shopIds = new ArrayList<>(); + shopIds.add(shopId); + List cpUserStoreDO = privilageCpUserStoreDOMapper.selectCpUserByShops(shopIds); + if(cpUserStoreDO.size() > 0){ + img = cpUserStoreDO.get(0).getQrCodeAction(); return data(img); }else{ throw new Exception();