From c6c9b7d3baa822765768c110304d27623f946286 Mon Sep 17 00:00:00 2001
From: pjb <123456>
Date: 星期二, 15 七月 2025 14:29:09 +0800
Subject: [PATCH] `1`
---
zy-asrs-wms/src/main/java/com/zy/asrs/wms/system/controller/BaseController.java | 120 ++++++++++++++++++++++++++++++++++-------------------------
1 files changed, 69 insertions(+), 51 deletions(-)
diff --git a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/system/controller/BaseController.java b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/system/controller/BaseController.java
index 5a4516b..b3a784b 100644
--- a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/system/controller/BaseController.java
+++ b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/system/controller/BaseController.java
@@ -1,51 +1,69 @@
-package com.zy.asrs.wms.system.controller;
-
-import com.zy.asrs.wms.common.domain.BaseParam;
-import com.zy.asrs.wms.system.entity.User;
-import org.springframework.security.core.Authentication;
-import org.springframework.security.core.context.SecurityContextHolder;
-
-import java.util.Map;
-
-/**
- * Created by vincent on 1/30/2024
- */
-public class BaseController {
-
- public User getLoginUser() {
- try {
- Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
- if (authentication != null) {
- Object object = authentication.getPrincipal();
- if (object instanceof User) {
- return (User) object;
- }
- }
- } catch (Exception e) {
- System.out.println(e.getMessage());
- }
- return null;
- }
-
- public Long getLoginUserId() {
- User loginUser = getLoginUser();
- return loginUser == null ? null : loginUser.getId();
- }
-
- public Long getHostId() {
- User loginUser = getLoginUser();
- return loginUser == null ? null : loginUser.getHostId();
- }
-
- public <T extends BaseParam> T buildParam(Map<String, Object> map, Class<T> clz) {
- T t = null;
- try {
- t = clz.getDeclaredConstructor().newInstance();
- t.syncMap(map);
- } catch (Exception e) {
- e.printStackTrace();
- }
- return t;
- }
-
-}
+package com.zy.asrs.wms.system.controller;
+
+import com.zy.asrs.wms.common.domain.BaseParam;
+import com.zy.asrs.wms.system.entity.User;
+import com.zy.asrs.wms.system.entity.UserLogin;
+import com.zy.asrs.wms.system.service.UserLoginService;
+import com.zy.asrs.wms.system.service.UserRoleService;
+import com.zy.asrs.wms.system.service.UserService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import java.util.*;
+
+/**
+ * Created by vincent on 1/30/2024
+ */
+public class BaseController {
+
+ protected static final String RANGE_TIME_LINK = " - ";
+
+ @Autowired
+ private UserLoginService userLoginService;
+ @Autowired
+ private UserRoleService userRoleService;
+ @Autowired
+ private UserService userService;
+
+ public User getLoginUser() {
+ try {
+ Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+ if (authentication != null) {
+ Object object = authentication.getPrincipal();
+ if (object instanceof User) {
+ return (User) object;
+ }
+ if(object instanceof UserLogin) {
+ UserLogin userLogin = (UserLogin) object;
+ User user = userService.superGetById(userLogin.getUserId());
+ return user;
+ }
+ }
+ } catch (Exception e) {
+ System.out.println(e.getMessage());
+ }
+ return null;
+ }
+
+ public Long getLoginUserId() {
+ User loginUser = getLoginUser();
+ return loginUser == null ? 1 : loginUser.getId();
+ }
+
+ public Long getHostId() {
+ User loginUser = getLoginUser();
+ return loginUser == null ? 1 : loginUser.getHostId();
+ }
+
+ public <T extends BaseParam> T buildParam(Map<String, Object> map, Class<T> clz) {
+ T t = null;
+ try {
+ t = clz.getDeclaredConstructor().newInstance();
+ t.syncMap(map);
+ } catch (Exception e) {
+ e.printStackTrace();
+ }
+ return t;
+ }
+
+}
--
Gitblit v1.9.1