无关风月
2025-02-26 7944fc1f0754c5b4441eb93b0fe4428f2b73055e
DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java
@@ -23,6 +23,7 @@
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import io.swagger.models.auth.In;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.*;
@@ -88,7 +89,7 @@
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResultUtil<String> getCode(HttpServletRequest request){
    public ResultUtil<Map<String,String>> getCode(HttpServletRequest request){
        try {
            Integer uid = driverService.getUserIdFormRedis(request);
            if(null == uid){
@@ -98,12 +99,20 @@
            if (userInfo.getCode()==null){
                userInfo = driverService.generateCode(userInfo);
            }
            return ResultUtil.success(userInfo.getCode());
            Map<String, String> res = new HashMap<>();
            res.put("code",userInfo.getCode());
            int i = inviteService.selectCount(new EntityWrapper<Invite>()
                    .eq("inviteUserId", uid)
                    .eq("useType", 2));
            res.put("inviteNumber",i+"");
            return ResultUtil.success(res);
        }catch (Exception e){
            e.printStackTrace();
            return ResultUtil.runErr();
        }
    }
    /**
     * 获取用户邀请二维码
     * @param request
@@ -119,7 +128,7 @@
            @ApiImplicitParam(value = "页条数", name = "size", required = true, dataType = "int"),
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResultUtil inviteList(String startTime,String endTime,Integer pageNum,Integer size,HttpServletRequest request){
    public ResultUtil<List<Invite>> inviteList(String startTime,String endTime,Integer pageNum,Integer size,HttpServletRequest request){
        try {
            Integer uid = driverService.getUserIdFormRedis(request);
            if(null == uid){
@@ -130,6 +139,14 @@
                endTime = endTime + " 23:59:59";
            }
            List<Invite> invites = inviteService.inviteList(uid,startTime,endTime,pageNum,size);
            for (Invite invite : invites) {
                // 将手机号phone中间四位替换为*
                String phone = invite.getPhone();
                if (phone != null && phone.length() > 4) {
                    phone = phone.substring(0, 3) + "****" + phone.substring(7);
                    invite.setPhone(phone);
                }
            }
            return ResultUtil.success(invites);
        }catch (Exception e){
            e.printStackTrace();