vdi/nex-be/src/main/java/com/unisinsight/project/controller/DeviceImageMappingController.java

package com.unisinsight.project.controller;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.unisinsight.project.entity.dao.DeviceImageMapping;
import com.unisinsight.project.entity.req.DeviceImageMappingReq;
import com.unisinsight.project.entity.res.DeviceImageMappingRes;
import com.unisinsight.project.entity.res.ListReq;
import com.unisinsight.project.exception.BaseErrorCode;
import com.unisinsight.project.exception.Result;
import com.unisinsight.project.service.DeviceImageMappingService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @description:
 * @author: rdpnr_puzhi
 * @create: 2025/08/07
 */
@Slf4j
@RestController
@RequestMapping("/api/nex/v1/device/image/mapping")
@Api(tags = "终端镜像映射关系类")
public class DeviceImageMappingController {

    @Resource
    private DeviceImageMappingService deviceImageMappingService;

    @ApiOperation(value = "终端镜像映射新增")
    @PostMapping("/add")
    public Result<?> insert(@RequestBody ListReq<DeviceImageMappingReq> deviceImageMappingReq) {
        if (Objects.isNull(deviceImageMappingReq.getData())) {
            return Result.errorResult(BaseErrorCode.PARAMS_CHK_ERROR);
        }
        log.info("终端镜像映射新增请求参数为：{}", JSONUtil.toJsonStr(deviceImageMappingReq));

        List<DeviceImageMappingReq> reqData = deviceImageMappingReq.getData();
        List<DeviceImageMappingReq> addList = reqData.stream().distinct().filter(e -> Objects.isNull(e.getId())).collect(Collectors.toList());

        QueryWrapper<DeviceImageMapping> wrapper = new QueryWrapper<>();
        wrapper.lambda().eq(DeviceImageMapping::getDeviceId, reqData.get(0).getDeviceId());
        List<DeviceImageMapping> list = deviceImageMappingService.list(wrapper);

        if (CollectionUtil.isNotEmpty(list)) {
            Set<Long> requestIds = reqData.stream()
                    .map(DeviceImageMappingReq::getId)
                    .filter(Objects::nonNull)
                    .collect(Collectors.toSet());

            List<Long> idsToDelete = list.stream()
                    .map(DeviceImageMapping::getId)
                    .filter(Objects::nonNull)
                    .filter(id -> !requestIds.contains(id))
                    .collect(Collectors.toList());

            if (CollectionUtil.isNotEmpty(idsToDelete)) {
                deviceImageMappingService.removeByIds(idsToDelete);
                log.info("删除了 {} 条不匹配的旧数据，ID列表: {}", idsToDelete.size(), idsToDelete);
            }
        }

        if (CollectionUtil.isNotEmpty(addList)) {
            List<DeviceImageMapping> deviceImageMappings = BeanUtil.copyToList(addList, DeviceImageMapping.class);
            List<DeviceImageMapping> deviceImageMappingList = deviceImageMappings.stream().peek(e -> {
                e.setCreateUser("admin");
            }).collect(Collectors.toList());
            boolean insert = deviceImageMappingService.saveBatch(deviceImageMappingList);
            log.info("终端镜像映射新增insert:{}", insert);
            if (!insert) {
                return Result.errorResult(BaseErrorCode.HTTP_ERROR_CODE_500);
            }
        }

        return Result.successResult();
    }

    @ApiOperation(value = "终端镜像映射查询")
    @PostMapping("/select")
    public Result<List<DeviceImageMappingRes>> select(@RequestBody DeviceImageMappingReq deviceImageMappingReq) {
        if (Objects.isNull(deviceImageMappingReq)) {
            return Result.errorResult(BaseErrorCode.PARAMS_CHK_ERROR);
        }
        log.info("终端镜像映射查询请求参数为：{}", JSONUtil.toJsonStr(deviceImageMappingReq));
        return deviceImageMappingService.select(deviceImageMappingReq);
    }


}