package com.ruoyi.production.controller;
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.ruoyi.framework.web.domain.R;
|
import com.ruoyi.production.dto.ProcessRouteDto;
|
import com.ruoyi.production.pojo.ProcessRoute;
|
import com.ruoyi.production.service.ProcessRouteService;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.util.StringUtils;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.Arrays;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
@RestController
|
@RequestMapping("/processRoute")
|
@Api(tags = "工艺路线")
|
public class ProcessRouteController {
|
|
@Autowired
|
private ProcessRouteService processRouteService;
|
|
@GetMapping("/page")
|
@ApiOperation("分页查询")
|
public R page(Page<ProcessRouteDto> page, ProcessRouteDto processRouteDto) {
|
return R.ok(processRouteService.pageProcessRouteDto(page, processRouteDto));
|
}
|
|
@ApiOperation("新增工艺路线")
|
@PostMapping
|
public R add(@RequestBody ProcessRoute processRoute) {
|
normalizeProductBinding(processRoute);
|
return R.ok(processRouteService.saveProcessRoute(processRoute));
|
}
|
|
@ApiOperation("修改工艺路线")
|
@PutMapping
|
public R update(@RequestBody ProcessRoute processRoute) {
|
normalizeProductBinding(processRoute);
|
return R.ok(processRouteService.updateById(processRoute));
|
}
|
|
@ApiOperation("删除工艺路线")
|
@DeleteMapping("/{ids}")
|
public R delete(@PathVariable("ids") Long[] ids) {
|
return R.ok(processRouteService.batchDelete(Arrays.asList(ids)));
|
}
|
|
private void normalizeProductBinding(ProcessRoute processRoute) {
|
if (processRoute == null) {
|
return;
|
}
|
String ids = null;
|
List<Long> idList = processRoute.getProductModelIdList();
|
if (idList != null && !idList.isEmpty()) {
|
ids = idList.stream()
|
.filter(item -> item != null && item > 0)
|
.distinct()
|
.map(String::valueOf)
|
.collect(Collectors.joining(","));
|
} else if (StringUtils.hasText(processRoute.getProductModelIds())) {
|
ids = Arrays.stream(processRoute.getProductModelIds().split(","))
|
.map(String::trim)
|
.filter(StringUtils::hasText)
|
.distinct()
|
.collect(Collectors.joining(","));
|
} else if (processRoute.getProductModelId() != null) {
|
ids = String.valueOf(processRoute.getProductModelId());
|
}
|
processRoute.setProductModelIds(ids);
|
if (StringUtils.hasText(ids)) {
|
processRoute.setProductModelId(Long.valueOf(ids.split(",")[0]));
|
}
|
}
|
}
|
