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 = "批量导入静态规则",