合并代码
This commit is contained in:
@@ -8,12 +8,17 @@ import com.gxwebsoft.common.core.web.PageResult;
|
||||
import com.gxwebsoft.common.system.entity.User;
|
||||
import com.gxwebsoft.tower.entity.TowerProject;
|
||||
import com.gxwebsoft.tower.entity.TowerProjectOrder;
|
||||
import com.gxwebsoft.tower.entity.TowerProjectPlace;
|
||||
import com.gxwebsoft.tower.entity.TowerProjectUser;
|
||||
import com.gxwebsoft.tower.param.TowerProjectParam;
|
||||
import com.gxwebsoft.tower.service.TowerProjectOrderService;
|
||||
import com.gxwebsoft.tower.service.TowerProjectPlaceService;
|
||||
import com.gxwebsoft.tower.service.TowerProjectService;
|
||||
import com.gxwebsoft.tower.service.TowerProjectUserService;
|
||||
import io.swagger.annotations.Api;
|
||||
import io.swagger.annotations.ApiOperation;
|
||||
import org.springframework.security.access.prepost.PreAuthorize;
|
||||
import org.springframework.transaction.annotation.Transactional;
|
||||
import org.springframework.web.bind.annotation.*;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
@@ -34,6 +39,10 @@ public class TowerProjectController extends BaseController {
|
||||
private TowerProjectService towerProjectService;
|
||||
@Resource
|
||||
private TowerProjectOrderService towerProjectOrderService;
|
||||
@Resource
|
||||
private TowerProjectUserService towerProjectUserService;
|
||||
@Resource
|
||||
private TowerProjectPlaceService towerProjectPlaceService;
|
||||
|
||||
@PreAuthorize("hasAuthority('tower:towerProject:list')")
|
||||
@ApiOperation("分页查询项目管理")
|
||||
@@ -60,6 +69,7 @@ public class TowerProjectController extends BaseController {
|
||||
}
|
||||
|
||||
@PreAuthorize("hasAuthority('tower:towerProject:save')")
|
||||
@Transactional(rollbackFor = {Exception.class})
|
||||
@ApiOperation("添加项目管理")
|
||||
@PostMapping()
|
||||
public ApiResult<?> save(@RequestBody TowerProject towerProject) {
|
||||
@@ -68,29 +78,59 @@ public class TowerProjectController extends BaseController {
|
||||
if (loginUser != null) {
|
||||
towerProject.setUserId(loginUser.getUserId());
|
||||
}
|
||||
System.out.println("towerProject = " + towerProject.getPlaces2());
|
||||
if (towerProjectService.save(towerProject)) {
|
||||
final Integer projectId = towerProject.getProjectId();
|
||||
// 批量添加项目成员
|
||||
if (!towerProject.getUsers().isEmpty()) {
|
||||
final Set<TowerProjectUser> users = towerProject.getUsers();
|
||||
users.forEach(d->{
|
||||
d.setProjectId(projectId);
|
||||
});
|
||||
towerProjectUserService.saveBatch(users);
|
||||
}
|
||||
// 批量添加项目订单
|
||||
if (!towerProject.getOrders().isEmpty()) {
|
||||
final Set<TowerProjectOrder> orders = towerProject.getOrders();
|
||||
orders.forEach(d->{
|
||||
d.setProjectId(towerProject.getProjectId());
|
||||
d.setProjectId(projectId);
|
||||
});
|
||||
towerProjectOrderService.saveBatch(orders);
|
||||
}
|
||||
// 批量添加检查部位配置
|
||||
if (!towerProject.getPlaces1().isEmpty()) {
|
||||
final Set<TowerProjectPlace> places1 = towerProject.getPlaces1();
|
||||
places1.forEach(d->{
|
||||
d.setProjectId(projectId);
|
||||
});
|
||||
towerProjectPlaceService.saveBatch(places1);
|
||||
}
|
||||
// 批量添加保养部位配置
|
||||
if (!towerProject.getPlaces2().isEmpty()) {
|
||||
final Set<TowerProjectPlace> places2 = towerProject.getPlaces2();
|
||||
places2.forEach(d->{
|
||||
d.setProjectId(projectId);
|
||||
});
|
||||
towerProjectPlaceService.saveBatch(places2);
|
||||
}
|
||||
return success("添加成功");
|
||||
}
|
||||
return fail("添加失败");
|
||||
}
|
||||
|
||||
@PreAuthorize("hasAuthority('tower:towerProject:update')")
|
||||
@Transactional(rollbackFor = {Exception.class})
|
||||
@ApiOperation("修改项目管理")
|
||||
@PutMapping()
|
||||
public ApiResult<?> update(@RequestBody TowerProject towerProject) {
|
||||
if (towerProjectService.updateById(towerProject)) {
|
||||
if (!towerProject.getOrders().isEmpty()) {
|
||||
final Set<TowerProjectOrder> orders = towerProject.getOrders();
|
||||
System.out.println("orders = " + orders);
|
||||
// 批量修改项目成员
|
||||
towerProjectUserService.saveOrUpdateBatch(towerProject.getUsers());
|
||||
// 批量修改项目订单
|
||||
towerProjectOrderService.saveOrUpdateBatch(towerProject.getOrders());
|
||||
}
|
||||
// 批量修改检查/保养部位配置
|
||||
towerProjectPlaceService.saveOrUpdateBatch(towerProject.getPlaces1());
|
||||
towerProjectPlaceService.saveOrUpdateBatch(towerProject.getPlaces2());
|
||||
return success("修改成功");
|
||||
}
|
||||
return fail("修改失败");
|
||||
|
||||
Reference in New Issue
Block a user