自动化立体仓库 - WMS系统
lty
2 天以前 264a479aa1ad16818110e9f9d833bbf02e298f47
src/main/java/com/zy/asrs/controller/OpenController.java
@@ -1,23 +1,29 @@
package com.zy.asrs.controller;
import com.alibaba.fastjson.JSON;
import com.core.annotations.AppAuth;
import com.core.common.*;
import com.core.exception.CoolException;
import com.zy.asrs.entity.DocType;
import com.zy.asrs.entity.Mat;
import com.zy.asrs.entity.Order;
import com.zy.asrs.entity.OrderDetl;
import com.zy.asrs.entity.param.OpenOrderPakinParam;
import com.zy.asrs.service.*;
import com.zy.asrs.entity.ErpOrder;
import com.zy.asrs.entity.ErpOrderDetl;
import com.zy.asrs.entity.param.*;
import com.zy.asrs.entity.ErpMat;
import com.zy.asrs.service.ErpOrderDetlService;
import com.zy.asrs.service.ErpOrderService;
import com.zy.asrs.service.OpenService;
import com.zy.asrs.service.ErpMatService;
import com.zy.common.model.DetlDto;
import com.zy.common.web.BaseController;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.*;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
/**
 * Created by vincent on 2022/4/8
@@ -25,43 +31,187 @@
@Slf4j
@RestController
@RequestMapping("open/asrs")
public class OpenController {
public class OpenController extends BaseController {
    private static final boolean auth = true;
    public static final ArrayList<String> APP_KEY_LIST = new ArrayList<String>() {{
        add("ea1f0459efc02a79f046f982767939ae");
    }};
    @Autowired
    private OrderService orderService;
    private OpenService openService;
    @Autowired
    private OrderDetlService orderDetlService;
    private ErpMatService erpMatService;
    @Autowired
    private SnowflakeIdWorker snowflakeIdWorker;
    private ErpOrderService erpOrderService;
    @Autowired
    private DocTypeService docTypeService;
    @Autowired
    private MatService matService;
    @Autowired
    private LocDetlService locDetlService;
    private ErpOrderDetlService erpOrderDetlService;
    @Value("${open-asrs.auth.app-id:wms_8f7c3d126a944e3ab5c89d21f4a7c6b2}")
    private String openAppId;
    private void auth(String appkey, Object obj) {
        log.info("{}接口被访问!appkey:{},请求数据:{}", "open/sensorType/list/auth/v1", appkey, JSON.toJSONString(obj));
        if (Cools.isEmpty(appkey)) {
            throw new CoolException("认证失败,请确认appkey无误!");
    @Value("${open-asrs.auth.app-secret:D9f3A7xP1kLm8Q2tW5zH0sY6vR4bNcE7FjK3uM9pT2aL8qX6}")
    private String openAppSecret;
    @Value("${open-asrs.auth.token-valid-minutes:30}")
    private Integer tokenValidMinutes;
    private static final Map<String, TokenInfoParam> TOKEN_CACHE = new ConcurrentHashMap<>();
    @PostMapping("/getToken")
    public synchronized R getToken(@RequestBody(required = false) Map<String, String> param) {
        log.info("{}接口被访问;请求数据:{}", "/getToken", JSON.toJSONString(param));
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        if (!APP_KEY_LIST.contains(appkey)) {
            throw new CoolException("认证失败,请确认appkey无误!");
        String appId = param.get("appId");
        String appSecret = param.get("appSecret");
        if (Cools.isEmpty(appId)) {
            return R.error("参数[appId]不能为空");
        }
        if (Cools.isEmpty(appSecret)) {
            return R.error("参数[appSecret]不能为空");
        }
        if (Cools.isEmpty(openAppId) || Cools.isEmpty(openAppSecret)) {
            throw new CoolException("系统未配置open-asrs.auth.app-id或open-asrs.auth.app-secret");
        }
        if (!openAppId.equals(appId) || !openAppSecret.equals(appSecret)) {
            throw new CoolException("认证失败,请确认appId或appSecret无误!");
        }
        String token = UUID.randomUUID().toString().replace("-", "");
        long expireAt = System.currentTimeMillis() + tokenValidMinutes * 60L * 1000L;
        TOKEN_CACHE.put(token, new TokenInfoParam(appId, expireAt));
        Map<String, Object> result = new HashMap<>();
        result.put("token", token);
        result.put("validTime", tokenValidMinutes);
        return R.ok().add(result);
    }
    public static boolean validToken(String authorization) {
        if (Cools.isEmpty(authorization)) {
            return false;
        }
        TokenInfoParam tokenInfo = TOKEN_CACHE.get(authorization);
        if (tokenInfo == null) {
            return false;
        }
        if (System.currentTimeMillis() > tokenInfo.getExpireAt()) {
            TOKEN_CACHE.remove(authorization);
            return false;
        }
        return true;
    }
    @PostMapping("/order/matSync/default/v1")
    @AppAuth(memo = "商品信息同步接口")
    public synchronized R syncMatInfo(@RequestHeader(required = false) String appkey,
                                      @RequestBody(required = false) MatSyncParam param,
                                      HttpServletRequest request){
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        openService.syncMat(param);
        return R.ok();
    }
    @PostMapping("/erp/mat/sync")
    public synchronized R erpMatSync(@RequestBody(required = false) List<Map<String, Object>> body,
                                     HttpServletRequest request) {
        auth(null, body, request);
        if (Cools.isEmpty(body)) {
            return R.parse(BaseRes.PARAM);
        }
        openService.syncErpMat(body);
        Map<String, Object> res = new HashMap<>();
        res.put("result", "SUCCESS");
        return R.ok().add(res);
    }
    @PostMapping("/erp/order/addPrd")
    public synchronized R erpOrderAddPrd(@RequestBody Object body,
                                         HttpServletRequest request) {
        List<OrderDto> orders;
        if (body instanceof List) {
            orders = JSON.parseArray(JSON.toJSONString(body), OrderDto.class);
        } else {
            OrderDto dto = JSON.parseObject(JSON.toJSONString(body), OrderDto.class);
            orders = new ArrayList<>();
            orders.add(dto);
        }
        auth(null, orders, request);
        if (Cools.isEmpty(orders)) {
            return R.parse(BaseRes.PARAM);
        }
        return R.ok().add(openService.addErpOrder(orders));
    }
    @PostMapping("/erp/order/addQc")
    public synchronized R erpOrderAddQc(@RequestBody Object body,
                                         HttpServletRequest request) {
        List<OrderDto> orders;
        if (body instanceof List) {
            orders = JSON.parseArray(JSON.toJSONString(body), OrderDto.class);
        } else {
            OrderDto dto = JSON.parseObject(JSON.toJSONString(body), OrderDto.class);
            orders = new ArrayList<>();
            orders.add(dto);
        }
        auth(null, orders, request);
        if (Cools.isEmpty(orders)) {
            return R.parse(BaseRes.PARAM);
        }
        return R.ok().add(openService.addErpOrder(orders));
    }
    @PostMapping("/erp/order/addSrm")
    public synchronized R erpOrderAddSrm(@RequestBody Object body,
                                         HttpServletRequest request) {
        List<OrderDto> orders;
        if (body instanceof List) {
            orders = JSON.parseArray(JSON.toJSONString(body), OrderDto.class);
        } else {
            OrderDto dto = JSON.parseObject(JSON.toJSONString(body), OrderDto.class);
            orders = new ArrayList<>();
            orders.add(dto);
        }
        auth(null, orders, request);
        if (Cools.isEmpty(orders)) {
            return R.parse(BaseRes.PARAM);
        }
        return R.ok().add(openService.addErpOrder(orders));
    }
    @PostMapping("/MES/inventory/details")
    public synchronized R mesInventoryDetails(@RequestBody(required = false) Map<String, Object> param,
                                              HttpServletRequest request) throws IOException {
        auth(null, param, request);
        if (param == null) {
            param = new HashMap<>();
        }
        return R.ok().add(openService.mesInventoryDetails(param));
    }
    @PostMapping("/MES/inventory/summary")
    public synchronized R mesInventorySummary(@RequestBody(required = false) Map<String, Object> param,
                                              HttpServletRequest request) throws IOException {
        auth(null, param, request);
        if (param == null) {
            param = new HashMap<>();
        }
        return R.ok().add(openService.mesInventorySummary(param));
    }
    /**
     * 添加入库单
     */
    @PostMapping("/order/pakin/new/default/v1")
    @Transactional
    public synchronized R orderCreate(@RequestHeader(required = false) String appkey,
                                      @RequestBody OpenOrderPakinParam param) {
        auth(appkey, param);
    @PostMapping("/order/pakin/default/v1")
    public synchronized R pakinOrderCreate(@RequestHeader(required = false) String appkey,
                                           @RequestBody OpenOrderPakinParam param,
                                           HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
@@ -74,84 +224,7 @@
        if (Cools.isEmpty(param.getOrderDetails())) {
            return R.error("单据明细[orderDetails]不能为空");
        }
        Order order = orderService.selectByNo(param.getOrderNo());
        if (!Cools.isEmpty(order)) {
            return R.error(param.getOrderNo() + "单据已存在,请勿重复提交");
        }
        DocType docType = docTypeService.selectOrAdd(param.getOrderType());
        Date now = new Date();
        // 单据主档
        order = new Order(
                String.valueOf(snowflakeIdWorker.nextId()),    // 编号[非空]
                param.getOrderNo(),    // 订单编号
                DateUtils.convert(now),    // 单据日期
                docType.getDocId(),    // 单据类型
                null,    // 项目编号
                null,    //
                null,    // 调拨项目编号
                null,    // 初始票据号
                null,    // 票据号
                null,    // 客户编号
                null,    // 客户
                null,    // 联系方式
                null,    // 操作人员
                null,    // 合计金额
                null,    // 优惠率
                null,    // 优惠金额
                null,    // 销售或采购费用合计
                null,    // 实付金额
                null,    // 付款类型
                null,    // 业务员
                null,    // 结算天数
                null,    // 邮费支付类型
                null,    // 邮费
                null,    // 付款时间
                null,    // 发货时间
                null,    // 物流名称
                null,    // 物流单号
                1L,    // 订单状态
                1,    // 状态
                9527L,    // 添加人员
                now,    // 添加时间
                9527L,    // 修改人员
                now,    // 修改时间
                null    // 备注
        );
        if (!orderService.insert(order)) {
            throw new CoolException("生成单据主档失败,请联系管理员");
        }
        // 单据明细档
        List<DetlDto> list = new ArrayList<>();
        List<DetlDto> orderDetails = param.getOrderDetails();
        for (DetlDto detail : orderDetails) {
            DetlDto dto = new DetlDto(detail.getMatnr(), detail.getBatch(), detail.getAnfme());
            if (DetlDto.has(list, dto)) {
                DetlDto detlDto = DetlDto.find(list, dto.getMatnr(), dto.getBatch());
                assert detlDto != null;
                detlDto.setAnfme(detlDto.getAnfme() + detail.getAnfme());
            } else {
                list.add(dto);
            }
        }
        for (DetlDto detlDto : list) {
            Mat mat = matService.selectByMatnr(detlDto.getMatnr());
            if (Cools.isEmpty(mat)) {
                throw new CoolException(detlDto.getMatnr() + "编号商品检索失败,请先添加商品");
            }
            OrderDetl orderDetl = new OrderDetl();
            orderDetl.sync(mat);
            orderDetl.setOrderId(order.getId());
            orderDetl.setOrderNo(order.getOrderNo());
            orderDetl.setCreateBy(9527L);
            orderDetl.setCreateTime(now);
            orderDetl.setUpdateBy(9527L);
            orderDetl.setUpdateTime(now);
            orderDetl.setStatus(1);
            orderDetl.setQty(0.0D);
            if (!orderDetlService.insert(orderDetl)) {
                throw new CoolException("生成单据明细失败,请联系管理员");
            }
        }
        openService.pakinOrderCreate(param);
        return R.ok();
    }
@@ -159,19 +232,131 @@
     * 入库单回写
     */
    @PostMapping("/order/pakin/complete/default/v1")
    @Transactional
    public synchronized R orderComplete(@RequestHeader(required = false) String appkey) {
        auth(appkey, null);
    public synchronized R orderPakinComplete(@RequestHeader(required = false) String appkey,
                                             @RequestBody(required = false) OpenOrderCompleteParam param,
                                             HttpServletRequest request) {
        auth(appkey, param, request);
        return R.ok().add(openService.pakinOrderComplete(param));
    }
    /**
     * 添加出库单
     */
    @PostMapping("/order/pakout/default/v1")
    @AppAuth(memo = "添加订单出库")
    public synchronized R pakoutOrderCreate(@RequestHeader(required = false) String appkey,
                                            @RequestBody OpenOrderPakoutParam param,
                                            HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        if (Cools.isEmpty(param.getLgort())) {
            return R.error("单据编号[lgort]不能为空");
        }
        if (!param.getLgort().equals("5006")) {
            return R.ok();
        }
        if (Cools.isEmpty(param.getOrderNo())) {
            return R.error("单据编号[orderNo]不能为空");
        }
        if (Cools.isEmpty(param.getOrderType())) {
            return R.error("单据类型[orderType]不能为空");
        }
        if (Cools.isEmpty(param.getOrderDetails())) {
            return R.error("单据明细[orderDetails]不能为空");
        }
        openService.pakoutOrderCreate(param);
        return R.ok();
    }
    /**
     * 出库单回写
     */
    @PostMapping("/order/pakout/complete/default/v1")
    public synchronized R orderPakoutComplete(@RequestHeader(required = false) String appkey,
                                              @RequestBody(required = false) OpenOrderCompleteParam param,
                                              HttpServletRequest request) {
        auth(appkey, param, request);
        return R.ok().add(openService.pakoutOrderComplete(param));
    }
    /**
     * 库存统计
     */
    @RequestMapping("/stock/default/v1")
    public R queryStock(@RequestHeader(required = false) String appkey) {
        auth(appkey, null);
        return R.ok(locDetlService.queryStockTotal());
    public R queryStock(@RequestHeader(required = false) String appkey,
                        HttpServletRequest request) {
        auth(appkey, null, request);
        return R.ok().add(openService.queryStock());
    }
    private void auth(String appkey, Object obj, HttpServletRequest request) {
        log.info("{}接口被访问;token:{};请求数据:{}", "open/sensorType/list/auth/v1", request.getHeader("Authorization"), JSON.toJSONString(obj));
        request.setAttribute("cache", obj);
        if (!auth) {
            return;
        }
        String authorization = request.getHeader("Authorization");
        if (Cools.isEmpty(authorization)) {
            authorization = request.getHeader("authorization");
        }
        if (!Cools.isEmpty(authorization)) {
            String val = authorization.trim();
            if (val.toLowerCase().startsWith("bearer")) {
                int idx = val.indexOf(' ');
                authorization = idx > -1 ? val.substring(idx + 1).trim() : "";
            } else {
                authorization = val;
            }
        }
        if (!Cools.isEmpty(authorization)) {
            if (validToken(authorization)) {
                return;
            }
            throw new CoolException("认证失败,请确认Authorization无误!");
        }
        throw new CoolException("认证失败,请确认Authorization无误!");
    }
    public static void main(String[] args) {
        // 1
        System.out.println("======================================");
        OpenOrderPakinParam param = new OpenOrderPakinParam();
        param.setOrderNo(String.valueOf(new SnowflakeIdWorker().nextId()));
        param.setOrderType("打包上线单");
        param.setOrderTime(DateUtils.convert(new Date()));
        List<DetlDto> orderDetails = new ArrayList<>();
        param.setOrderDetails(orderDetails);
        for (int i = 0; i < 3; i++) {
            DetlDto detlDto = new DetlDto();
            switch (i) {
                case 0:
                    detlDto.setMatnr("MDH020030530");
                    detlDto.setBatch("500");
                    detlDto.setAnfme(18.0);
                    break;
                case 1:
                    detlDto.setMatnr("MDH020016416");
                    detlDto.setBatch("500");
                    detlDto.setAnfme(32.0);
                    break;
                case 2:
                    detlDto.setMatnr("LSH90152025");
                    detlDto.setAnfme(50.0);
                    break;
                default:
                    break;
            }
            orderDetails.add(detlDto);
        }
        System.out.println(JSON.toJSONString(param));
        // 2
        System.out.println("======================================");
        OpenOrderCompleteParam param1 = new OpenOrderCompleteParam();
        param1.setOrderNo("963001846497017856");
        System.out.println(JSON.toJSONString(param1));
    }
}