|
@@ -16,6 +16,9 @@
|
|
|
*/
|
|
|
package org.springblade.control.controller;
|
|
|
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
import io.swagger.annotations.Api;
|
|
|
import io.swagger.annotations.ApiOperation;
|
|
|
import io.swagger.annotations.ApiParam;
|
|
@@ -24,10 +27,16 @@ import lombok.AllArgsConstructor;
|
|
|
|
|
|
import javax.validation.Valid;
|
|
|
|
|
|
+import org.springblade.control.entity.DepartmentMonthPlan;
|
|
|
+import org.springblade.control.entity.UserFilesInfo;
|
|
|
+import org.springblade.control.service.IUserFilesInfoService;
|
|
|
import org.springblade.core.mp.support.Condition;
|
|
|
import org.springblade.core.mp.support.Query;
|
|
|
+import org.springblade.core.secure.BladeUser;
|
|
|
import org.springblade.core.tool.api.R;
|
|
|
import org.springblade.core.tool.utils.Func;
|
|
|
+import org.springblade.core.tool.utils.StringUtil;
|
|
|
+import org.springblade.system.user.entity.User;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
import org.springframework.web.bind.annotation.RequestParam;
|
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
@@ -50,6 +59,8 @@ public class UserContractInfoController extends BladeController {
|
|
|
|
|
|
private final IUserContractInfoService userContractInfoService;
|
|
|
|
|
|
+ private final IUserFilesInfoService userFilesInfoService;
|
|
|
+
|
|
|
/**
|
|
|
* 详情
|
|
|
*/
|
|
@@ -79,6 +90,17 @@ public class UserContractInfoController extends BladeController {
|
|
|
@ApiOperationSupport(order = 4)
|
|
|
@ApiOperation(value = "新增", notes = "传入userContractInfo")
|
|
|
public R save(@Valid @RequestBody UserContractInfo userContractInfo) {
|
|
|
+ if(userContractInfo.getType()==2){
|
|
|
+ Long userCount = userFilesInfoService.getBaseMapper().selectCount(
|
|
|
+ Wrappers.<UserFilesInfo>query().lambda()
|
|
|
+ .eq(UserFilesInfo::getUserId, userContractInfo.getUserId())
|
|
|
+ .eq(UserFilesInfo::getIsDeleted, 0));
|
|
|
+ if(userCount<=0){
|
|
|
+ UserFilesInfo userFilesInfo = new UserFilesInfo();
|
|
|
+ userFilesInfo.setUserId(userContractInfo.getUserId());
|
|
|
+ userFilesInfoService.saveOrUpdate(userFilesInfo);
|
|
|
+ }
|
|
|
+ }
|
|
|
return R.status(userContractInfoService.save(userContractInfo));
|
|
|
}
|
|
|
|
|
@@ -89,6 +111,17 @@ public class UserContractInfoController extends BladeController {
|
|
|
@ApiOperationSupport(order = 5)
|
|
|
@ApiOperation(value = "修改", notes = "传入userContractInfo")
|
|
|
public R update(@Valid @RequestBody UserContractInfo userContractInfo) {
|
|
|
+ if(userContractInfo.getType()==2){
|
|
|
+ Long userCount = userFilesInfoService.getBaseMapper().selectCount(
|
|
|
+ Wrappers.<UserFilesInfo>query().lambda()
|
|
|
+ .eq(UserFilesInfo::getUserId, userContractInfo.getUserId())
|
|
|
+ .eq(UserFilesInfo::getIsDeleted, 0));
|
|
|
+ if(userCount<=0){
|
|
|
+ UserFilesInfo userFilesInfo = new UserFilesInfo();
|
|
|
+ userFilesInfo.setUserId(userContractInfo.getUserId());
|
|
|
+ userFilesInfoService.saveOrUpdate(userFilesInfo);
|
|
|
+ }
|
|
|
+ }
|
|
|
return R.status(userContractInfoService.updateById(userContractInfo));
|
|
|
}
|
|
|
|
|
@@ -98,7 +131,28 @@ public class UserContractInfoController extends BladeController {
|
|
|
@PostMapping("/submit")
|
|
|
@ApiOperationSupport(order = 6)
|
|
|
@ApiOperation(value = "新增或修改", notes = "传入userContractInfo")
|
|
|
- public R submit(@Valid @RequestBody UserContractInfo userContractInfo) {
|
|
|
+ public R submit(@Valid @RequestBody UserContractInfo userContractInfo, BladeUser bladeUser) {
|
|
|
+
|
|
|
+ if(userContractInfo.getType()==2){
|
|
|
+ Long userCount = userFilesInfoService.getBaseMapper().selectCount(
|
|
|
+ Wrappers.<UserFilesInfo>query().lambda()
|
|
|
+ .eq(UserFilesInfo::getUserId, userContractInfo.getUserId())
|
|
|
+ .eq(UserFilesInfo::getIsDeleted, 0));
|
|
|
+ if(userCount<=0){
|
|
|
+ UserFilesInfo userFilesInfo = new UserFilesInfo();
|
|
|
+ userFilesInfo.setUserId(userContractInfo.getUserId());
|
|
|
+ userFilesInfoService.saveOrUpdate(userFilesInfo);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ UserContractInfo one = userContractInfoService.getOne( new LambdaQueryWrapper<UserContractInfo>()
|
|
|
+ .eq(UserContractInfo::getUserId, userContractInfo.getUserId())
|
|
|
+ .eq(UserContractInfo::getType, userContractInfo.getType())
|
|
|
+ .eq(UserContractInfo::getIsDeleted, 0));
|
|
|
+
|
|
|
+ if(one !=null && StringUtil.isNotBlank(one.getId()+"")) {
|
|
|
+ userContractInfo.setId(one.getId());
|
|
|
+ }
|
|
|
return R.status(userContractInfoService.saveOrUpdate(userContractInfo));
|
|
|
}
|
|
|
|