From cecd7580b5e629173bdb5b258f2676d16d580f72 Mon Sep 17 00:00:00 2001
From: gongchunyi <deslre0381@gmail.com>
Date: 星期四, 30 七月 2026 15:49:55 +0800
Subject: [PATCH] fix: 安全培训考核签到修改

---
 src/main/java/com/ruoyi/safe/controller/SafeTrainingController.java |   14 ++++++++++++++
 1 files changed, 14 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/ruoyi/safe/controller/SafeTrainingController.java b/src/main/java/com/ruoyi/safe/controller/SafeTrainingController.java
index a1cd796..6251128 100644
--- a/src/main/java/com/ruoyi/safe/controller/SafeTrainingController.java
+++ b/src/main/java/com/ruoyi/safe/controller/SafeTrainingController.java
@@ -1,6 +1,8 @@
 package com.ruoyi.safe.controller;
 
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.framework.aspectj.lang.annotation.Log;
 import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
 import com.ruoyi.framework.web.domain.R;
@@ -37,6 +39,7 @@
     @GetMapping("/page")
     @Operation(summary = "鍒嗛〉鏌ヨ")
     public R page(Page page, SafeTrainingDto safeTrainingDto) {
+        safeTrainingDto.setCurrentUserId(SecurityUtils.getUserId());
         return R.ok(safeTrainingService.pageSafeTraining(page, safeTrainingDto));
     }
 
@@ -51,6 +54,17 @@
     @Log(title = "瀹夊叏鍩硅", businessType = BusinessType.INSERT)
     @PostMapping ("/sign")
     public R sign(@RequestBody SafeTrainingDetails safeTrainingDetails) {
+        if (safeTrainingDetails.getSafeTrainingId() == null || safeTrainingDetails.getUserId() == null) {
+            return R.fail("绛惧埌淇℃伅涓嶅畬鏁�");
+        }
+        long count = safeTrainingDetailsService.count(
+                Wrappers.<SafeTrainingDetails>lambdaQuery()
+                        .eq(SafeTrainingDetails::getSafeTrainingId, safeTrainingDetails.getSafeTrainingId())
+                        .eq(SafeTrainingDetails::getUserId, safeTrainingDetails.getUserId())
+        );
+        if (count > 0) {
+            return R.fail("宸茬鍒帮紝璇峰嬁閲嶅绛惧埌");
+        }
         return R.ok(safeTrainingDetailsService.save(safeTrainingDetails));
     }
 

--
Gitblit v1.9.3