diff --git a/business-css/src/main/java/com/yfd/business/css/controller/CriticalDataController.java b/business-css/src/main/java/com/yfd/business/css/controller/CriticalDataController.java index 9f023b1..e11a5d2 100644 --- a/business-css/src/main/java/com/yfd/business/css/controller/CriticalDataController.java +++ b/business-css/src/main/java/com/yfd/business/css/controller/CriticalDataController.java @@ -4,10 +4,18 @@ import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.yfd.business.css.domain.CriticalData; import com.yfd.business.css.service.CriticalDataService; +import com.yfd.platform.system.service.IUserService; + + +import org.springframework.security.authentication.AnonymousAuthenticationToken; +import org.springframework.security.core.Authentication; +import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import jakarta.annotation.Resource; + +import java.time.LocalDateTime; import java.util.List; @RestController @@ -16,6 +24,8 @@ public class CriticalDataController { @Resource private CriticalDataService criticalDataService; + @Resource + private IUserService userService; /** * 1. 新增临界数据 @@ -26,9 +36,22 @@ public class CriticalDataController { */ @PostMapping public boolean create(@RequestBody CriticalData data) { + data.setModifier(currentUsername()); return criticalDataService.save(data); } + private String currentUsername() { + try { + Authentication auth = SecurityContextHolder.getContext().getAuthentication(); + if (auth == null || auth instanceof AnonymousAuthenticationToken) { + return "anonymous"; + } + return userService.getUsername(); + } catch (Exception e) { + return "anonymous"; + } + } + /** * 2. 修改临界数据 * 输入参数:请求体中的临界数据对象(需包含主键) @@ -38,6 +61,8 @@ public class CriticalDataController { */ @PutMapping public boolean update(@RequestBody CriticalData data) { + data.setModifier(currentUsername()); + data.setUpdatedAt(LocalDateTime.now()); return criticalDataService.updateById(data); }