初始化

dubin

2026-04-16 bab6283f5c0b5da44b7a342f9daef35473a13ab7

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

@@ -1,52 +1,195 @@
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.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.DocTypeService;
import com.zy.asrs.service.MatService;
import com.zy.asrs.service.OrderDetlService;
import com.zy.asrs.service.OrderService;
import com.zy.common.model.DetlDto;
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.service.wms.Result;
import com.zy.common.web.BaseController;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
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
 */
@Slf4j
@RestController
@RequestMapping("open/asrs/")
public class OpenController {
@RequestMapping("open/asrs")
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 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 SnowflakeIdWorker snowflakeIdWorker;
    @Autowired
    private DocTypeService docTypeService;
    @Autowired
    private MatService matService;
    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/default/v1")
    @Transactional
    public synchronized R orderCreate(@RequestBody OpenOrderPakinParam param) {
    @PostMapping("/order/pakin/default/v1")
    @AppAuth(memo = "入库单据下发")
    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);
        }
@@ -59,86 +202,292 @@
        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();
    }

    /**
     * 入库单回写
     */
    @PostMapping("/order/pakin/complete/default/v1")
    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,
                        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无误！");
            }
        }




    /**
     * 库存统计
     */
    @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;
        }
        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);
        }
    }
}