diff --git a/file-web/src/main/java/com/qiwenshare/file/controller/FiletransferController.java b/file-web/src/main/java/com/qiwenshare/file/controller/FiletransferController.java index f1e6040..8b0dc29 100644 --- a/file-web/src/main/java/com/qiwenshare/file/controller/FiletransferController.java +++ b/file-web/src/main/java/com/qiwenshare/file/controller/FiletransferController.java @@ -142,9 +142,18 @@ public class FiletransferController { */ @RequestMapping(value = "/getstorage", method = RequestMethod.GET) @ResponseBody - public RestResult getStorage() { + public RestResult getStorage(@RequestHeader("token") String token) { RestResult restResult = new RestResult(); - UserBean sessionUserBean = (UserBean) SecurityUtils.getSubject().getPrincipal(); + //UserBean sessionUserBean = (UserBean) SecurityUtils.getSubject().getPrincipal(); + UserBean sessionUserBean = new UserBean(); + boolean isRemoteLogin = qiwenFileConfig.isRemoteLogin(); + if (isRemoteLogin) { + + RestResult restUserBean = remoteUserService.checkUserLoginInfo(token); + sessionUserBean = restUserBean.getData(); + } else { + sessionUserBean = (UserBean) SecurityUtils.getSubject().getPrincipal(); + } StorageBean storageBean = new StorageBean(); if (qiwenFileConfig.isShareMode()){ storageBean.setUserId(2L);