From 9ae0190c4112587eea11b2f0c1a0f8fcf31a146e Mon Sep 17 00:00:00 2001
From: zzl <961867786@qq.com>
Date: Fri, 5 Sep 2025 09:05:41 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=BB=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../web/controller/business/TTaskExportController.java     | 7 +++++++
 .../web/controller/business/TTaskImportController.java     | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskExportController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskExportController.java
index 84d4ffe..044d999 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskExportController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskExportController.java
@@ -5,6 +5,7 @@ import com.ruoyi.business.service.ITTaskExportService;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.SysDept;
 import com.ruoyi.common.core.domain.pdf.PageSet;
 import com.ruoyi.common.core.domain.pdf.PdfProperty;
 import com.ruoyi.common.core.domain.pdf.ShoulderItem;
@@ -13,6 +14,7 @@ import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.common.utils.pdf.PdfUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.common.utils.translation.TranslateUtils;
+import com.ruoyi.system.service.ISysDeptService;
 import org.apache.commons.compress.utils.Lists;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
@@ -35,6 +37,9 @@ public class TTaskExportController extends BaseController
     @Autowired
     private ITTaskExportService tTaskExportService;
 
+    @Autowired
+    private ISysDeptService deptService;
+
     /**
      * 查询导出任务列表
      */
@@ -110,6 +115,8 @@ public class TTaskExportController extends BaseController
     @PostMapping("/add")
     public AjaxResult add(@RequestBody TTaskExport tTaskExport)
     {
+        SysDept dept = deptService.selectDeptById(tTaskExport.getDeptId());
+        tTaskExport.setOrgCode(dept.getOrgCode());
         return toAjax(tTaskExportService.insertTTaskExport(tTaskExport));
     }
 
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskImportController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskImportController.java
index 4a14adf..d65c51e 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskImportController.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/business/TTaskImportController.java
@@ -5,6 +5,7 @@ import com.ruoyi.business.service.ITTaskImportService;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.SysDept;
 import com.ruoyi.common.core.domain.pdf.PageSet;
 import com.ruoyi.common.core.domain.pdf.PdfProperty;
 import com.ruoyi.common.core.domain.pdf.ShoulderItem;
@@ -13,6 +14,7 @@ import com.ruoyi.common.enums.BusinessType;
 import com.ruoyi.common.utils.pdf.PdfUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.common.utils.translation.TranslateUtils;
+import com.ruoyi.system.service.ISysDeptService;
 import org.apache.commons.compress.utils.Lists;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.access.prepost.PreAuthorize;
@@ -35,6 +37,9 @@ public class TTaskImportController extends BaseController
     @Autowired
     private ITTaskImportService tTaskImportService;
 
+    @Autowired
+    private ISysDeptService deptService;
+
     /**
      * 查询导入任务列表
      */
@@ -113,6 +118,8 @@ public class TTaskImportController extends BaseController
     @PostMapping("/add")
     public AjaxResult add(@RequestBody TTaskImport tTaskImport)
     {
+        SysDept dept = deptService.selectDeptById(tTaskImport.getDeptId());
+        tTaskImport.setOrgCode(dept.getOrgCode());
         return toAjax(tTaskImportService.insertTTaskImport(tTaskImport));
     }