From 8f8d6e2d973860120847d1569ed0e63951ce7ca3 Mon Sep 17 00:00:00 2001 From: PushM <584406942@qq.com> Date: Mon, 13 May 2024 20:10:01 +0800 Subject: [PATCH] =?UTF-8?q?1=E3=80=81=E9=9D=99=E6=80=81=E8=A7=84=E5=88=99?= =?UTF-8?q?=E6=96=B0=E5=BB=BA=E5=A2=9E=E5=8A=A0=E4=BB=8Esession=E8=8E=B7?= =?UTF-8?q?=E5=8F=96=E7=94=A8=E6=88=B7=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../rule/staticrule/StaticRuleController.java | 16 ++++++++++++++-- .../rule/staticrule/StaticRuleControllerApi.java | 3 ++- 2 files changed, 16 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleController.java b/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleController.java index 9c24221..834b34a 100644 --- a/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleController.java +++ b/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleController.java @@ -3,9 +3,12 @@ package com.realtime.protection.server.rule.staticrule; import com.alibaba.excel.EasyExcel; import com.realtime.protection.configuration.entity.rule.staticrule.StaticRuleObject; +import com.realtime.protection.configuration.entity.user.UserFull; import com.realtime.protection.configuration.response.ResponseResult; import com.realtime.protection.configuration.utils.enums.audit.AuditStatusEnum; +import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletResponse; +import jakarta.servlet.http.HttpSession; import jakarta.validation.Valid; import jakarta.validation.constraints.NotNull; import lombok.extern.slf4j.Slf4j; @@ -36,8 +39,17 @@ public class StaticRuleController implements StaticRuleControllerApi { */ @PostMapping("/new") @Override - public ResponseResult newStaticRuleObject(@RequestBody @Valid StaticRuleObject object) { - log.info("新增静态规则: {}", object); + public ResponseResult newStaticRuleObject(@RequestBody @Valid StaticRuleObject object, + HttpServletRequest request) { +// log.info("新增静态规则: {}", object); + //从http首部session字段获取用户信息 + HttpSession session = request.getSession(); + UserFull user = (UserFull) session.getAttribute("user"); + if (user != null) { + object.setStaticRuleCreateUsername(user.name); + object.setStaticRuleCreateUserId(Integer.valueOf(user.uid)); + object.setStaticRuleCreateDepart(user.getOrgName()); + } //调用service新增 staticRuleService.newStaticRuleObject(object); return ResponseResult.ok() diff --git a/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleControllerApi.java b/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleControllerApi.java index 53676d1..390111c 100644 --- a/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleControllerApi.java +++ b/src/main/java/com/realtime/protection/server/rule/staticrule/StaticRuleControllerApi.java @@ -9,6 +9,7 @@ import io.swagger.v3.oas.annotations.media.ExampleObject; import io.swagger.v3.oas.annotations.media.Schema; import io.swagger.v3.oas.annotations.responses.ApiResponse; import io.swagger.v3.oas.annotations.tags.Tag; +import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletResponse; import jakarta.validation.Valid; import jakarta.validation.constraints.NotNull; @@ -52,7 +53,7 @@ public interface StaticRuleControllerApi { requestBody = @io.swagger.v3.oas.annotations.parameters.RequestBody( description = "静态规则信息") ) - ResponseResult newStaticRuleObject(@RequestBody @Valid StaticRuleObject object); + ResponseResult newStaticRuleObject(@RequestBody @Valid StaticRuleObject object, HttpServletRequest request); @Operation( summary = "批量导入静态规则",