From f6b489d950dc687529ea4280b9e3a8dee0a90b60 Mon Sep 17 00:00:00 2001 From: xupei Date: Wed, 8 Jan 2025 09:50:13 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=AE=A1=E7=90=86=E5=91=98?= =?UTF-8?q?=E9=A1=B5=E9=9D=A2=E8=AE=BF=E9=97=AE=E8=80=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ConvenientInquiryController.java | 30 ++++++++++++++++--- 1 file changed, 26 insertions(+), 4 deletions(-) diff --git a/src/main/java/com/ai/da/controller/ConvenientInquiryController.java b/src/main/java/com/ai/da/controller/ConvenientInquiryController.java index c1158fe6..3d74c777 100644 --- a/src/main/java/com/ai/da/controller/ConvenientInquiryController.java +++ b/src/main/java/com/ai/da/controller/ConvenientInquiryController.java @@ -62,7 +62,11 @@ public class ConvenientInquiryController { public Response> getDesignStatistic(@RequestParam String startTime, @RequestParam String endTime, @RequestParam(required = false) List ids, @RequestParam(required = false) String email) { Long accountId = UserContext.getUserHolder().getId(); - if (accountId.equals(31L) || accountId.equals(87L) || accountId.equals(83L) || accountId.equals(6L) || accountId.equals(4L) || accountId.equals(73L)) { + String userEmail = accountService.getById(accountId).getUserEmail(); + if (accountId.equals(31L) || accountId.equals(87L) || accountId.equals(83L) + || accountId.equals(6L) || accountId.equals(4L) || accountId.equals(73L) + || userEmail.equals("joho8228@hotmail.com") || userEmail.equals("wanninghua160@gmail.com") + ) { if (StringUtil.isNullOrEmpty(startTime)) startTime = "2024-02-01 00:00:00"; if (StringUtil.isNullOrEmpty(endTime)) { SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss"); @@ -145,7 +149,14 @@ public class ConvenientInquiryController { @ApiOperation("添加用户") @PostMapping("/addUser") public Response addUser(@Valid @RequestBody AccountAddDTO accountAddDTO) { - return Response.success(convenientInquiryService.addUser(accountAddDTO)); + Long userAccountId = UserContext.getUserHolder().getId(); + if (userAccountId.equals(31L) || userAccountId.equals(87L) + || userAccountId.equals(6L) || userAccountId.equals(4L) || userAccountId.equals(73L) + ) { + return Response.success(convenientInquiryService.addUser(accountAddDTO)); + } else { + return Response.fail("Sorry, you don't have permission"); + } } @ApiOperation("修改用户信息") @@ -154,14 +165,25 @@ public class ConvenientInquiryController { @ApiParam(value = "有效期截止时间的毫秒级unix格式") @RequestParam @Nullable Long validEndTime, @ApiParam(value = "用户类型 1/2/3/0 -> yearly/monthly/trial/visitor") @RequestParam @Nullable Integer systemUser, @ApiParam("积分") @RequestParam @Nullable Long credits) { - return Response.success(convenientInquiryService.modifyUser(accountId, validEndTime, systemUser, credits)); + Long userAccountId = UserContext.getUserHolder().getId(); + if (userAccountId.equals(31L) || userAccountId.equals(87L) + || userAccountId.equals(6L) || userAccountId.equals(4L) || userAccountId.equals(73L) + ) { + return Response.success(convenientInquiryService.modifyUser(accountId, validEndTime, systemUser, credits)); + } else { + return Response.fail("Sorry, you don't have permission"); + } } @ApiOperation("获取用户信息") @PostMapping("/getUserInfo") public Response> getUserInfo(@Valid @RequestBody QueryUserConditionsVO queryUserConditionsVO) { Long accountId = UserContext.getUserHolder().getId(); - if (accountId.equals(31L) || accountId.equals(87L) || accountId.equals(83L) || accountId.equals(6L) || accountId.equals(4L) || accountId.equals(73L)) { + String userEmail = accountService.getById(accountId).getUserEmail(); + if (accountId.equals(31L) || accountId.equals(87L) || accountId.equals(83L) + || accountId.equals(6L) || accountId.equals(4L) || accountId.equals(73L) + || userEmail.equals("joho8228@hotmail.com") || userEmail.equals("wanninghua160@gmail.com") + ) { return Response.success(convenientInquiryService.getUserInfo(queryUserConditionsVO)); } else { return Response.fail("Sorry, you don't have permission");