skyfffire пре 1 месец
родитељ
комит
538ddb2f5f

+ 1 - 1
src/main/java/common/jfinal/AppConfig.java

@@ -16,7 +16,7 @@ import modules.order.OrderController;
 import modules.upload.UploadController;
 import modules.user.UserController;
 import org.slf4j.Logger;
-import org.slf4j.LoggerFactory; // 导入这两个类
+import org.slf4j.LoggerFactory;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.ThreadPoolExecutor;

+ 17 - 0
src/main/java/modules/user/UserController.java

@@ -340,6 +340,23 @@ public class UserController extends MyController {
         }
     }
 
+    @Before(LoginInterceptor.class)
+    public void getBalance() {
+        User user = service.findUserByMobileNumber(this.<String>getSessionAttr("mobile_number"));
+
+        if (user == null) {
+            removeSessionAttr("dl-token");
+            removeSessionAttr("role");
+            removeSessionAttr("mobile_number");
+
+            renderJson(MyRet.fail("登录信息失效,请重新登录。").setCode(MyRet.CODE_NO_LOGIN));
+        } else {
+            setSessionAttr("role", user.getRole().toString());
+
+            renderJson(MyRet.ok("获取成功").setData(user.getBalance()));
+        }
+    }
+
     @Before(LoginInterceptor.class)
     public void getUserInfo() {
         User user = service.findUserByMobileNumber(this.<String>getSessionAttr("mobile_number"));

+ 5 - 0
src/test/rest/UserControllerTest.http

@@ -69,6 +69,11 @@ POST {{ baseUrl }}/user/getUserInfo
 Content-Type: application/json
 dl-token: {{dl_token_var}}
 
+### 获取账户火花余额
+POST {{ baseUrl }}/user/getBalance
+Content-Type: application/json
+dl-token: {{dl_token_var}}
+
 ### 【超级管理员】更新用户的方法,传什么更新什么,不传就不更新,id必须传
 POST {{ baseUrl }}/user/updateByAdmin
 Content-Type: application/json