任务

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));
     }
 

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));
     }