原矿报表山上山下一起查

sckw-modules/sckw-transport/src/main/java/com/sckw/transport/repository/KwtLogisticsOrderUnitRepository.java

@@ -104,4 +104,11 @@ public class KwtLogisticsOrderUnitRepository extends ServiceImpl<KwtLogisticsOrd
                 .eq(KwtLogisticsOrderUnit::getEntId, entId)
                 .eq(KwtLogisticsOrderUnit::getUnitType,unitType));
     }
+
+    public List<KwtLogisticsOrderUnit> queryByEntIdsAndUnitType(List<Long> entIds, int unitType) {
+        return list(Wrappers.<KwtLogisticsOrderUnit>lambdaQuery()
+                .eq(KwtLogisticsOrderUnit::getDelFlag,0)
+                .in(KwtLogisticsOrderUnit::getEntId, entIds)
+                .eq(KwtLogisticsOrderUnit::getUnitType,unitType));
+    }
 }

sckw-modules/sckw-transport/src/main/java/com/sckw/transport/service/app/WaybillOrderService.java

@@ -2330,7 +2330,7 @@ public class WaybillOrderService {
         }
 
         // 1. 物流企业
-        List<KwtLogisticsOrderUnit> logOrderUnits = logisticsOrderUnitRepository.queryByEntIdAndUnitType(mockEntIds.get(0), Global.NUMERICAL_ONE);
+        List<KwtLogisticsOrderUnit> logOrderUnits = logisticsOrderUnitRepository.queryByEntIdsAndUnitType(mockEntIds, Global.NUMERICAL_ONE);
         if (CollectionUtils.isEmpty(logOrderUnits)) {
             log.info("当前企业无物流订单企业信息！，entId:{}", param.getEntId());
             return Collections.emptyList();