105 lines
4.2 KiB
Java
105 lines
4.2 KiB
Java
package com.wms.controller;
|
||
|
||
import com.alibaba.fastjson2.JSON;
|
||
import com.wms.constants.WmsConstants;
|
||
import com.wms.constants.enums.ResponseCode;
|
||
import com.wms.entity.app.ResponseEntity;
|
||
import com.wms.entity.app.display.LocationData;
|
||
import com.wms.entity.app.display.LocationInfo;
|
||
import com.wms.entity.table.Location;
|
||
import com.wms.entity.table.Stock;
|
||
import com.wms.service.LocationService;
|
||
import com.wms.service.StockService;
|
||
import com.wms.utils.HttpUtils;
|
||
import com.wms.utils.StringUtils;
|
||
import jakarta.servlet.http.HttpServletRequest;
|
||
import lombok.RequiredArgsConstructor;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.stereotype.Controller;
|
||
import org.springframework.web.bind.annotation.*;
|
||
|
||
import java.util.LinkedList;
|
||
import java.util.List;
|
||
|
||
@Controller
|
||
@CrossOrigin
|
||
@RequiredArgsConstructor(onConstructor = @__(@Autowired))
|
||
@RequestMapping(value = "/api/query")
|
||
public class DisplayController extends BaseController{
|
||
private final StockService stockService;
|
||
private final LocationService locationService;
|
||
private final HttpServletRequest servletRequest;
|
||
|
||
/**
|
||
* 大屏查询库位信息
|
||
* @param location 参数---库位
|
||
* @return 结果
|
||
*/
|
||
@GetMapping("/queryLocationInfo")
|
||
@ResponseBody
|
||
public String queryLocationInfo(@RequestParam String location) {
|
||
logger.info("接收到查询大屏库位信息请求，ip地址：{}，参数：{}", HttpUtils.getIpAddr(servletRequest), location);
|
||
ResponseEntity response = new ResponseEntity();
|
||
if (StringUtils.isEmpty(location)) {
|
||
logger.error("查询失败，参数中库位号为空");
|
||
response.setCode(ResponseCode.ERROR.getCode());
|
||
response.setMessage("查询失败，参数中库位号为空");
|
||
return JSON.toJSONString(response);
|
||
}
|
||
Location locationQuery = new Location();
|
||
locationQuery.setLocationId(location);
|
||
List<Location> locations = locationService.selLocations(locationQuery);
|
||
if (locations.size() == 0) {
|
||
logger.error("查询失败，该库位立体库中不存在");
|
||
response.setCode(ResponseCode.ERROR.getCode());
|
||
response.setMessage("查询失败，请检查库位号正确性");
|
||
return JSON.toJSONString(response);
|
||
}
|
||
Location currentLocation = locations.get(0);
|
||
String vehicleNo = currentLocation.getVehicleId();
|
||
if (StringUtils.isEmpty(vehicleNo)) {
|
||
// 设定
|
||
LocationInfo returnData = new LocationInfo();
|
||
returnData.setLocationId(location);
|
||
returnData.setVehicleNo(WmsConstants.EMPTY_STRING);
|
||
response.setCode(ResponseCode.OK.getCode());
|
||
response.setMessage("查询成功");
|
||
response.setReturnData(returnData);
|
||
return JSON.toJSONString(response);
|
||
}
|
||
// 查询库存
|
||
Stock stockQuery = new Stock();
|
||
stockQuery.setLocationId(location);
|
||
stockQuery.setVehicleId(vehicleNo);
|
||
List<Stock> stocks = stockService.selStocks(stockQuery);
|
||
List<LocationData> locationData = new LinkedList<>();
|
||
for (Stock tempStock : stocks) {
|
||
LocationData tempData = new LocationData();
|
||
tempData.setGoodsId(tempStock.getGoodsId());
|
||
tempData.setGoodsName(tempStock.getGoodsName());
|
||
tempData.setGoodsNum(tempStock.getRealNum());
|
||
locationData.add(tempData);
|
||
}
|
||
// 设定
|
||
LocationInfo returnData = new LocationInfo();
|
||
returnData.setLocationId(location);
|
||
returnData.setVehicleNo(WmsConstants.EMPTY_STRING);
|
||
returnData.setLocationData(locationData);
|
||
response.setCode(ResponseCode.OK.getCode());
|
||
response.setMessage("查询成功");
|
||
response.setReturnData(returnData);
|
||
return JSON.toJSONString(response);
|
||
}
|
||
|
||
/**
|
||
* 查询所有库位
|
||
* @return 结果
|
||
*/
|
||
@GetMapping("/getAllLocations")
|
||
@ResponseBody
|
||
public String queryLocationInfo() {
|
||
logger.info("接收到查询大屏所有库位信息请求，ip地址：{}", HttpUtils.getIpAddr(servletRequest));
|
||
return JSON.toJSONString(locationService.selLocations(new Location()));
|
||
}
|
||
}
|