skyfffire 1 mese fa
parent
commit
47bd2d49a3

+ 21 - 1
src/main/java/modules/user/UserController.java

@@ -129,8 +129,28 @@ public class UserController extends MyController {
             user.set("mobile_number", mobileNumber);
         }
 
+        // 推荐人
+        String referrerIdStr = requestBodyJson.getString("referrer_id");
+        if (StrKit.notBlank(referrerIdStr)) {
+            int referrerIdInt;
+            try {
+                referrerIdInt = Integer.parseInt(referrerIdStr);
+
+                if (referrerIdInt <= 0) {
+                    renderJson(MyRet.fail("推荐人(referrer_id)期待是正整数,你传的是: " + referrerIdStr));
+                    return;
+                }
+                
+                // 推荐人
+                user.setReferrerId(referrerIdInt);
+            } catch (Exception e) {
+                renderJson(MyRet.fail("推荐人(referrer_id)格式不正确: " + e.getMessage()));
+                return;
+            }
+        }
+
         // 参数补充
-        user.set("role", 2); // 例如:2表示普通用户
+        user.set("role", 2); // 2表示普通用户
         // 密码加密并存储
         user.set("pwd_md5_md5", HashKit.md5(pwd));
         user.set("integral", 0);

+ 3 - 2
src/test/rest/UserControllerTest.http

@@ -46,7 +46,7 @@ POST {{ baseUrl }}/user/sendVerifyCode
 Content-Type: application/json
 
 {
-  "mobile_number": "17781855864"
+  "mobile_number": "17760515340"
 }
 
 ### 注册
@@ -57,7 +57,8 @@ Content-Type: application/json
   "mobile_number": "17760515340",
   "pwd_md5": "e10adc3949ba59abbe56e057f20f883e",
   "repeat_pwd_md5": "e10adc3949ba59abbe56e057f20f883e",
-  "verify_code": "6962"
+  "verify_code": "7810",
+  "referrer_id": ""
 }
 
 ### 获取充值方式