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