自动化立体仓库 - WMS系统
dubin
2026-04-16 bab6283f5c0b5da44b7a342f9daef35473a13ab7
src/main/java/com/zy/asrs/controller/OpenController.java
@@ -1,25 +1,29 @@
package com.zy.asrs.controller;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.core.annotations.AppAuth;
import com.core.common.*;
import com.core.exception.CoolException;
import com.zy.asrs.entity.WrkMast;
import com.zy.asrs.entity.param.*;
import com.zy.asrs.entity.vo.OpenInventoryVo;
import com.zy.asrs.entity.vo.TokenVo;
import com.zy.asrs.mapper.LocDetlMapper;
import com.zy.asrs.service.OpenService;
import com.zy.asrs.service.WrkMastService;
import com.zy.common.model.DetlDto;
import com.zy.common.service.wms.Result;
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.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;
/**
 * Created by vincent on 2022/4/8
@@ -38,6 +42,19 @@
    private OpenService openService;
    @Autowired
    private WrkMastService wrkMastService;
    @Value("${open-asrs.auth.app-id:wms_8f7c3d126a944e3ab5c89d21f4a7c6b2}")
    private String openAppId;
    @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, TokenVo> TOKEN_CACHE = new ConcurrentHashMap<>();
    @Autowired
    private LocDetlMapper locDetlMapper;
    @GetMapping("/order/delete/v1")
    @AppAuth(memo = "订单信息删除")
@@ -281,4 +298,196 @@
    }
    /*=============================================NEW===================================================*/
    private void authNew(String appkey, Object obj, HttpServletRequest request, String url) {
        log.info("{}接口被访问;appkey:{};请求数据:{}", url, appkey, 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 boolean validToken(String authorization) {
        if (Cools.isEmpty(authorization)) {
            return false;
        }
        TokenVo tokenInfo = TOKEN_CACHE.get(authorization);
        if (tokenInfo == null) {
            return false;
        }
        if (System.currentTimeMillis() > tokenInfo.getValidTime()) {
            TOKEN_CACHE.remove(authorization);
            return false;
        }
        return true;
    }
    /**
     * 获取Token
     * @param appkey
     * @param param
     * @param request
     * @return
     */
    @PostMapping("/getToken")
    public synchronized R getToken(@RequestHeader(required = false) String appkey,
                                   @RequestBody(required = false) TokenParam param,
                                   HttpServletRequest request){
//        authNew(appkey, param, request, "获取Token:/getToken");
        log.info("获取Token:/getToken接口被访问,appkey={},请求数据={}", appkey, JSON.toJSONString(param));
        if (Cools.isEmpty(param)){
            return R.parse(BaseRes.PARAM);
        }
        String appId = param.getAppId();
        String appSecret = param.getAppSecret();
        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().replaceAll("-", "");
        long expireAt = System.currentTimeMillis() + tokenValidMinutes * 60L *1000L;
        TOKEN_CACHE.put(token, new TokenVo(appId, expireAt));
        Map<String, Object> result = new HashMap<>();
        result.put("token", token);
        result.put("expireAt", tokenValidMinutes);
        return R.ok().add(result);
    }
    /**
     *物料基础信息同步
     * @param appKey
     * @param param
     * @param request
     * @return
     */
    @PostMapping("/erp/mat/sync")
    public synchronized R syncMatInfoV2(@RequestHeader(required = false) String appKey,
                                        @RequestBody(required = false) List<MatSyncParam.MatParam> param,
                                        HttpServletRequest request){
        authNew(appKey, param, request, "物料基础信息同步:/erp/mat/sync");
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        MatSyncParam matSyncParam = new MatSyncParam();
        List<MatSyncParam.MatParam> objects = new ArrayList<>();
        objects = param;
        matSyncParam.matDetails = objects;
        openService.syncMat(matSyncParam);
        return R.ok();
    }
    /**
     * 入/出库通知单下发
     * @param appKey
     * @param param
     * @param request
     * @return
     */
    @PostMapping("/erp/order/add")
    public synchronized R addOrder(@RequestHeader(required = false) String appKey,
                                   @RequestBody OpenOrderParam param,
                                   HttpServletRequest request){
        authNew(appKey, param, request, "入/出库通知单下发:/erp/order/add");
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        if (Cools.isEmpty(param.getOrderNo())){
            return R.error("单据编号[orderNo]不能为空");
        }
        if (Cools.isEmpty(param.getOrderType())){
            return R.error("单据类型[orderType]不能为空");
        }
        if (Cools.isEmpty(param.getWkType())){
            return R.error("业务类型[wkType]不能为空");
        }
        if (Cools.isEmpty(param.getOrderItems())){
            return R.error("订单明细[orderItems]不能为空");
        }
        openService.orderCreate(param);
        return R.ok();
    }
    /**
     * 库存明细查询
     * @param appKey
     * @param param
     * @param request
     * @return
     */
    @PostMapping("/erp/inventory/details")
    public synchronized Result inventory(@RequestHeader(required = false) String appKey,
                                         @RequestBody JSONObject param,
                                         HttpServletRequest request){
        authNew(appKey, param, request, "库存明细查询:/erp/inventory/details");
        try {
            String wareHouseId = param == null ? null : param.getString("wareHouseId");
            String locId = param == null ? null : param.getString("locId");
            String matNr = param == null ? null : param.getString("matNr");
            String orderNo = param == null ? null : param.getString("orderNo");
            String batch =  param == null ? null : param.getString("batch");
            return new Result(200, "操作成功", locDetlMapper.inventory(wareHouseId, locId, matNr, orderNo, batch));
        }catch (Exception e){
            return new Result(500, e.getMessage(), null);
        }
    }
    /**
     * 库存汇总查询
     * @param appKey
     * @param param
     * @param request
     * @return
     */
    @PostMapping("/erp/inventory/summary")
    public synchronized Result invSummary(@RequestHeader(required = false) String appKey,
                                     @RequestBody  JSONObject param,
                                     HttpServletRequest request){
        authNew(appKey, param, request, "库存汇总查询:/erp/inventory/summary");
        try {
            String wareHouseId = param == null ? null : param.getString("wareHouseId");
            String matNr = param == null ? null : param.getString("matNr");
            Collection<String> matNrs = null;
            if (!Cools.isEmpty(matNr)){
                matNrs = Arrays.stream(matNr.split(","))
                        .map(String::trim)
                        .filter(s -> !s.isEmpty())
                        .collect(Collectors.toList());
            }
            return new Result(200, "操作成功", locDetlMapper.invSummary(wareHouseId, matNrs));
        }catch (Exception e){
            return new Result(500, e.getMessage(), null);
        }
    }
}