From 7944fc1f0754c5b4441eb93b0fe4428f2b73055e Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期三, 26 二月 2025 09:22:39 +0800
Subject: [PATCH] 2.0新增

---
 DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java |   23 ++++++++++++++++++++---
 1 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java b/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java
index 6dc4398..b1a075b 100644
--- a/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java
+++ b/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();

--
Gitblit v1.7.1