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

---
 UserOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/UserInfoController.java |   21 ++++++++++++++++++---
 1 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/UserOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/UserInfoController.java b/UserOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/UserInfoController.java
index c8c1ed2..36dda04 100644
--- a/UserOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/UserInfoController.java
+++ b/UserOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/UserInfoController.java
@@ -68,7 +68,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 = userInfoService.getUserIdFormRedis(request);
             if(null == uid){
@@ -78,7 +78,14 @@
             if (userInfo.getCode()==null){
                 userInfo = userInfoService.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", 1));
+
+            res.put("inviteNumber",i+"");
+            return ResultUtil.success(res);
         }catch (Exception e){
             e.printStackTrace();
             return ResultUtil.runErr();
@@ -99,7 +106,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 = userInfoService.getUserIdFormRedis(request);
             if(null == uid){
@@ -110,6 +117,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