2.0新增

无关风月

2025-02-27 19c35c58f8b767d7d9c57d0cf910fd091dc7669f

 DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java

@@ -23,7 +23,9 @@
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.*;

import javax.annotation.Resource;
@@ -73,10 +75,87 @@

    @Autowired
    private IOrderLogisticsService orderLogisticsService;
    @Autowired
    private IInviteService inviteService;

    /**
     * 获取用户邀请二维码
     * @param request
     * @return
     */
    @ResponseBody
    @PostMapping("/api/user/getCode")
    @ApiOperation(value = "获取用户邀请二维码", tags = {"司机端-2.0新增"}, notes = "")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResultUtil<Map<String,String>> getCode(HttpServletRequest request){
        try {
            Integer uid = driverService.getUserIdFormRedis(request);
            if(null == uid){
                return ResultUtil.tokenErr();
            }
            Driver userInfo = driverService.selectById(uid);
            if (userInfo.getCode()==null){
                userInfo = driverService.generateCode(userInfo);
            }
            driverService.generateCode(userInfo);
            Map<String, String> res = new HashMap<>();
            res.put("code",userInfo.getCode());
            int i = inviteService.selectCount(new EntityWrapper<Invite>()
                    .eq("inviteUserId", uid)
                    .eq("userType", 2));

            res.put("inviteNumber",i+"");
            return ResultUtil.success(res);
        }catch (Exception e){
            e.printStackTrace();
            return ResultUtil.runErr();
        }
    }

    /**
     * 获取用户邀请二维码
     * @param request
     * @return
     */
    @ResponseBody
    @PostMapping("/api/user/inviteList")
    @ApiOperation(value = "获取司机邀请记录", tags = {"司机端-2.0新增"}, notes = "")
    @ApiImplicitParams({
            @ApiImplicitParam(value = "开始时间 yyyy-MM-dd", name = "startTime", required = false, dataType = "String"),
            @ApiImplicitParam(value = "结束时间 yyyy-MM-dd", name = "endTime", required = false, dataType = "String"),
            @ApiImplicitParam(value = "页码（首页1）", name = "pageNum", required = true, dataType = "int"),
            @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<List<Invite>> inviteList(String startTime,String endTime,Integer pageNum,Integer size,HttpServletRequest request){
        try {
            Integer uid = driverService.getUserIdFormRedis(request);
            if(null == uid){
                return ResultUtil.tokenErr();
            }
            if (StringUtils.hasLength(startTime)){
                startTime = startTime + " 00:00:00";
                endTime = endTime + " 23:59:59";
            }
            pageNum = (pageNum - 1) * size;
            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();
            return ResultUtil.runErr();
        }
    }
    /**
     * 获取短信验证码
     * @param phone
     * @return