DssCarInfoController.java 2.51 KB
package com.ruoyi.controller.dss;

import com.ruoyi.common.TipEnum;
import com.ruoyi.common.core.domain.ResponseResult;
import com.ruoyi.domain.caiinfo.CarInfo;
import com.ruoyi.domain.dss.car.info.dto.GetPlatesDTO;
import com.ruoyi.framework.security.filter.JwtAuthenticationTokenFilter;
import com.ruoyi.service.carinfo.CarInfoService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.validation.Valid;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author liujun
 * @date 2024年07月16日 15:13
 */
@Slf4j
@RestController
@RequestMapping("/dss")
@Api(tags = "【蓝斯一期】车辆信息")
public class DssCarInfoController {
    @Autowired
    private CarInfoService carInfoService;

    @GetMapping(value = "/device/getPlates")
    @ApiOperation("模糊查询车辆车牌列表")
    public ResponseResult<Set<String>> getPlates(@Valid @RequestBody GetPlatesDTO dto, BindingResult bindingResult) {
        if (bindingResult.hasErrors()) {
            return ResponseResult.error(bindingResult.getFieldError().getDefaultMessage());
        }

        String msg = JwtAuthenticationTokenFilter.validateDevice(dto.getDevice());
        if(StringUtils.isNotEmpty(msg)){
           log.info(msg);
           return ResponseResult.error(TipEnum.TIP_401.getCode(),TipEnum.TIP_401.getMsg());
        }

        CarInfo carInfo = convertCarInfo(dto);
        if (StringUtils.isEmpty(carInfo.getPlateNum())) {
            return ResponseResult.success();
        }

        List<CarInfo> carInfos = carInfoService.likePlateNumPlateNumTop30(carInfo);
        if (CollectionUtils.isEmpty(carInfos)) {
            return ResponseResult.success();
        }
        Set<String> plateNums = carInfos.stream().map(CarInfo::getPlateNum).collect(Collectors.toSet());
        return ResponseResult.success(plateNums);
    }

    private CarInfo convertCarInfo(GetPlatesDTO dto) {
        CarInfo carInfo = new CarInfo();
        carInfo.setPlateNum(dto.getPlate());

        return carInfo;
    }
}