From adca98ba944374e7e619beeebe97607f3f33572b Mon Sep 17 00:00:00 2001
From: tangzy <tangzy123456>
Date: Mon, 26 Jul 2021 16:43:14 +0800
Subject: [PATCH] 1.用户添加派遣字段,派遣记录

---
 src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java |   26 +++++++++++++++++++++++++-
 1 files changed, 25 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java b/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
index 9032235..96d44cb 100644
--- a/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
+++ b/src/main/java/org/springblade/modules/dispatcher/controller/DispatcherController.java
@@ -23,16 +23,20 @@
 import lombok.AllArgsConstructor;
 import javax.validation.Valid;
 
+import org.springblade.common.utils.arg;
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.mp.support.Query;
 import org.springblade.core.tool.api.R;
 import org.springblade.core.tool.utils.Func;
+import org.springblade.modules.dispatcher.service.IDispatcherUnitService;
 import org.springframework.web.bind.annotation.*;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import org.springblade.modules.dispatcher.entity.Dispatcher;
 import org.springblade.modules.dispatcher.vo.DispatcherVO;
 import org.springblade.modules.dispatcher.service.IDispatcherService;
 import org.springblade.core.boot.ctrl.BladeController;
+
+import java.util.Arrays;
 
 /**
  *  控制器
@@ -47,6 +51,8 @@
 public class DispatcherController extends BladeController {
 
 	private final IDispatcherService dispatcherService;
+
+	private final IDispatcherUnitService dispatcherUnitService;
 
 	/**
 	 * 详情
@@ -107,7 +113,25 @@
 	@PostMapping("/submit")
 	@ApiOperationSupport(order = 6)
 	@ApiOperation(value = "新增或修改", notes = "传入dispatcher")
-	public R submit(@Valid @RequestBody Dispatcher dispatcher) {
+	public R submit(@Valid @RequestBody Dispatcher dispatcher) throws Exception {
+		if (dispatcher.getId()==null){
+			if (null!=dispatcher.getUserIds() && dispatcher.getUserIds()!=""){
+				//计算派遣人数
+				dispatcher.setNum(Arrays.asList(dispatcher.getUserIds().split(",")).size());
+				//查询派遣单位的信息,设置行政区
+				dispatcher.setJurisdiction(dispatcherUnitService.getById(dispatcher.getDispatcherUnitId()).getJurisdiction().toString());
+			}
+			arg arg = new arg();
+			arg.test01(arg.url+"/dispatcher/save",dispatcher);
+		}
+		else {
+			if (null != dispatcher.getUserIds() && dispatcher.getUserIds() != "") {
+				//计算派遣人数
+				dispatcher.setNum(Arrays.asList(dispatcher.getUserIds().split(",")).size());
+				//查询派遣单位的信息,设置行政区
+				dispatcher.setJurisdiction(dispatcherUnitService.getById(dispatcher.getDispatcherUnitId()).getJurisdiction().toString());
+			}
+		}
 		return R.status(dispatcherService.saveOrUpdate(dispatcher));
 	}
 

--
Gitblit v1.9.3