初始化

dubin

2026-04-16 bab6283f5c0b5da44b7a342f9daef35473a13ab7

 src/main/java/com/zy/asrs/controller/OpenController.java

@@ -1,19 +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.param.OpenOrderCompleteParam;
import com.zy.asrs.entity.param.OpenOrderPakinParam;
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.common.model.DetlDto;
import com.zy.asrs.service.WrkMastService;
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 java.util.ArrayList;
import java.util.Date;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

/**
 * Created by vincent on 2022/4/8
@@ -21,7 +31,7 @@
@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>() {{
@@ -30,14 +40,156 @@

    @Autowired
    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 = "订单信息删除")
    public synchronized R orderDelete(@RequestHeader(required = false) String appkey,
                                  @RequestParam String orderNo,
                                  HttpServletRequest request) {
        auth(appkey, orderNo, request);
        if (Cools.isEmpty(orderNo)) {
            return R.parse(BaseRes.PARAM);
        }
        openService.orderDelete(orderNo);
        return R.ok();

    }


    @PostMapping("/tagSync/v1")
    @AppAuth(memo = "物料分类同步")
    public synchronized R tagSync(@RequestHeader(required = false) String appkey,
                                        @RequestBody(required = false) List<Map<String,String>> param,
                                        HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        ArrayList<TagParam> params = new ArrayList<>();
        for (Map<String,String> map : param) {
            TagParam tagParam  = new TagParam(map);
            params.add(tagParam);
        }

        openService.syncTag(params);
        return R.ok();

    }


    @PostMapping("/wrkMast/finish/v1")
    @AppAuth(memo = "四向车任务完成")
    public synchronized R wrkMastFinish(@RequestHeader(required = false) String appkey,
                                        @RequestBody(required = false) ShuttleFinishParam param,
                                        HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        try {
            WrkMast wrkMast = wrkMastService.selectOne(new EntityWrapper<WrkMast>().eq("wrk_no", param.getTaskNo()).eq("crn_no",7));
            if (Cools.isEmpty(wrkMast)) {
                throw new CoolException("四向库任务未找到");
            }
            if (wrkMast.getIoType() == 1 || wrkMast.getIoType() == 10|| wrkMast.getIoType() == 11 || wrkMast.getIoType() == 53 || wrkMast.getIoType() == 54 || wrkMast.getIoType() == 57) {
                wrkMast.setWrkSts(4L);
            }else {
                wrkMast.setWrkSts(14L);
            }
            if (!wrkMastService.updateById(wrkMast)){
                return R.error();
            }
        }catch (Exception e) {
            throw new CoolException(e);
        }


        return R.ok();
    }



    @PostMapping("/locDetl/select/time/v1")
    public synchronized R locDetlSelect(@RequestHeader(required = false) String appkey,
                                        @RequestBody(required = false) TimeSelectParam param,
                                        HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
        List<LocDetlByTimeDTO> locDetlByTimeDTOS = openService.selectTimeLocDetl(param);
        return R.ok(locDetlByTimeDTOS);
    }

    @PostMapping("/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("/orderSync/default/v1")
    @AppAuth(memo = "订单信息同步接口")
    public synchronized R syncOrderInfo(@RequestHeader(required = false) String appkey,
                                      @RequestBody(required = false) GTOrderParam param,
                                      HttpServletRequest request){
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }

        return R.ok();
    }



    /****************************************************************************/
    /********************************* 打包上线 **********************************/
    /****************************************************************************/

//    @PostMapping("/order/pakin/new/package/v1")
//    @AppAuth(memo = "打包上线接口")
//    public synchronized R pakinOrderPackage(@RequestHeader(required = false) String appkey,
//                                            @RequestBody(required = false) PackParam param,
//                                            HttpServletRequest request) {
//        auth(appkey, param, request);
//        if (Cools.isEmpty(param)) {
//            return R.parse(BaseRes.PARAM);
//        }
//        openService.packageUp(param);
//        return R.ok();
//    }

    /**
     * 添加入库单
     */
    @PostMapping("/order/pakin/new/default/v1")
    @PostMapping("/order/pakin/default/v1")
    @AppAuth(memo = "入库单据下发")
    public synchronized R pakinOrderCreate(@RequestHeader(required = false) String appkey,
                                           @RequestBody OpenOrderPakinParam param) {
        auth(appkey, param);
                                           @RequestBody OpenOrderPakinParam param,
                                           HttpServletRequest request) {
        auth(appkey, param, request);
        if (Cools.isEmpty(param)) {
            return R.parse(BaseRes.PARAM);
        }
@@ -58,70 +210,284 @@
     * 入库单回写
     */
    @PostMapping("/order/pakin/complete/default/v1")
    public synchronized R orderComplete(@RequestHeader(required = false) String appkey,
                                        @RequestBody OpenOrderCompleteParam param) {
        auth(appkey, param);
    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);
    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("{}接口被访问；appkey:{}；请求数据：{}", request.getRequestURI(), appkey, JSON.toJSONString(obj));
            request.setAttribute("cache", obj);
            if (!auth) {
                return;
            }
            if (Cools.isEmpty(appkey)) {
                throw new CoolException("认证失败，请确认appkey无误！");
            }
            if (!APP_KEY_LIST.contains(appkey)) {
                throw new CoolException("认证失败，请确认appkey无误！");
            }
        }

    private void auth(String appkey, Object obj) {
        log.info("{}接口被访问！appkey:{},请求数据：{}", "open/sensorType/list/auth/v1", appkey, JSON.toJSONString(obj));



    /**
     * 库存统计
     */
    @RequestMapping("/agv/haikang/complete/v1")
    public R agvHaiKang(@RequestHeader(required = false) String wrkNo,
                        HttpServletRequest request) {
//        auth(wrkNo, null, request);
        return R.ok().add(openService.agvHaiKang(wrkNo));
    }


    /*=============================================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;
        }
        if (Cools.isEmpty(appkey)) {
            throw new CoolException("认证失败，请确认appkey无误！");
        String authorization = request.getHeader("Authorization");
        if (Cools.isEmpty(authorization)) {
            authorization = request.getHeader("authorization");
        }
        if (!APP_KEY_LIST.contains(appkey)) {
            throw new CoolException("认证失败，请确认appkey无误！");
        }
    }

    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.setAnfme(18.0);
                    break;
                case 1:
                    detlDto.setMatnr("MDH020016416");
                    detlDto.setAnfme(32.0);
                    break;
                case 2:
                    detlDto.setMatnr("LSH90152025");
                    detlDto.setAnfme(50.0);
                    break;
                default:
                    break;
        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;
            }
            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));
        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);
        }
    }
}