diff --git a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestDemandController.java b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestDemandController.java index dc787e2bd..6906bc52c 100755 --- a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestDemandController.java +++ b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestDemandController.java @@ -6,6 +6,7 @@ import com.iformall.annotation.SystemServiceLog; import com.iformall.common.ErrorCode; import com.iformall.common.InvestResultData; import com.iformall.common.ResultData; +import com.iformall.controller.base.BaseController; import com.iformall.domain.po.InvestDemandEntity; import com.iformall.domain.vo.invest.InvestDemandVo; import com.iformall.domain.vo.invest.InvestPageQuery; @@ -30,7 +31,7 @@ import io.swagger.annotations.ApiOperation; @RestController @RequestMapping("invest/demand") @Api(tags = "需求管理") -public class InvestDemandController { +public class InvestDemandController extends BaseController { @Autowired private InvestDemandService investDemandService; diff --git a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestFollowRecordController.java b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestFollowRecordController.java index 2c707ed07..957e4f503 100755 --- a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestFollowRecordController.java +++ b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestFollowRecordController.java @@ -5,6 +5,7 @@ import java.util.Arrays; import com.iformall.common.ErrorCode; import com.iformall.common.InvestResultData; import com.iformall.common.ResultData; +import com.iformall.controller.base.BaseController; import com.iformall.domain.po.InvestFollowRecordEntity; import com.iformall.domain.vo.invest.InvestPageQuery; import com.iformall.domain.vo.invest.InvestPageResult; @@ -29,7 +30,7 @@ import io.swagger.annotations.ApiOperation; @RestController @RequestMapping("invest/followrecord") @Api(tags = "跟踪管理") -public class InvestFollowRecordController { +public class InvestFollowRecordController extends BaseController { @Autowired private InvestFollowRecordService investFollowRecordService; diff --git a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestOperateRecordController.java b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestOperateRecordController.java index 959348c57..3e16b5d9c 100755 --- a/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestOperateRecordController.java +++ b/mallinkAdmin/src/main/java/com/iformall/controller/invest/InvestOperateRecordController.java @@ -6,6 +6,7 @@ import java.util.List; import com.iformall.common.ErrorCode; import com.iformall.common.InvestResultData; import com.iformall.common.ResultData; +import com.iformall.controller.base.BaseController; import com.iformall.domain.po.InvestOperateRecordEntity; import com.iformall.domain.vo.invest.InvestDemandVo; import com.iformall.domain.vo.invest.InvestPageQuery; @@ -31,7 +32,7 @@ import io.swagger.annotations.ApiOperation; @RestController @RequestMapping("invest/operaterecord") @Api(tags = "操作记录") -public class InvestOperateRecordController { +public class InvestOperateRecordController extends BaseController { @Autowired private InvestOperateRecordService investOperateRecordService; diff --git a/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestUserContext.java b/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestUserContext.java new file mode 100644 index 000000000..efb86e6a4 --- /dev/null +++ b/mallinkService/src/main/java/com/iformall/domain/vo/invest/InvestUserContext.java @@ -0,0 +1,30 @@ +package com.iformall.domain.vo.invest; + +import com.iformall.domain.po.MallUserInfo; + +import java.util.Objects; + +public class InvestUserContext { + + public static final ThreadLocal threadLocal = ThreadLocal.withInitial(() -> new MallUserInfo()); + + + public static Long getUserId() { + if (Objects.isNull(getUser())) return null; + return getUser().getId(); + } + + public static MallUserInfo getUser() { + return threadLocal.get(); + } + + public static void setUser(MallUserInfo user) { + if (user != null) { + threadLocal.set(user); + } + } + + public static void remove() { + threadLocal.remove(); + } +}