package com.vincent.rsf.server.manager.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.vincent.rsf.server.common.datasource.DataSourceNames;
|
import com.vincent.rsf.server.common.datasource.UseDataSource;
|
import com.vincent.rsf.server.manager.entity.CusItemSyncView;
|
import com.vincent.rsf.server.manager.mapper.CusItemSyncViewMapper;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.beans.factory.ObjectProvider;
|
import org.springframework.stereotype.Service;
|
|
import javax.sql.DataSource;
|
import java.util.ArrayList;
|
import java.util.Collection;
|
import java.util.Collections;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* cus_item_sync_view 查询；仅使用副库视图
|
*/
|
@UseDataSource(DataSourceNames.DJ_CLOUD_WMS)
|
@Service
|
@Slf4j
|
public class CusItemSyncViewQueryService {
|
|
@Autowired
|
private CusItemSyncViewMapper cusItemSyncViewMapper;
|
|
@Autowired
|
@Qualifier("cusItemSyncDataSource")
|
private ObjectProvider<DataSource> cusItemSyncDataSourceProvider;
|
|
/** 当前视图查询实际使用的数据源说明 */
|
public String effectiveDataSourceLabel() {
|
return cusItemSyncDataSourceProvider.getIfAvailable() != null ? DataSourceNames.DJ_CLOUD_WMS : "none";
|
}
|
|
/**
|
* 取视图前若干行
|
*/
|
public List<Map<String, Object>> probeSample(int limit) {
|
if (cusItemSyncDataSourceProvider.getIfAvailable() == null) {
|
return Collections.emptyList();
|
}
|
int n = Math.min(50, Math.max(1, limit));
|
IPage<CusItemSyncView> page = cusItemSyncViewMapper.selectPage(new Page<>(1, n), viewSelectWrapper());
|
return toViewMaps(page.getRecords());
|
}
|
|
public List<Map<String, Object>> listByItemNos(Collection<String> itemNos) {
|
if (itemNos == null || itemNos.isEmpty()) {
|
return Collections.emptyList();
|
}
|
List<String> codes = new ArrayList<>();
|
for (String code : itemNos) {
|
String c = StringUtils.trimToNull(code);
|
if (c != null) {
|
codes.add(c);
|
}
|
}
|
if (codes.isEmpty()) {
|
return Collections.emptyList();
|
}
|
if (cusItemSyncDataSourceProvider.getIfAvailable() == null) {
|
log.warn("dj-cloud-wms 数据源未配置，跳过视图查询");
|
return Collections.emptyList();
|
}
|
List<CusItemSyncView> rows = cusItemSyncViewMapper.selectList(
|
viewSelectWrapper().in(CusItemSyncView::getItemNo, codes));
|
return toViewMaps(rows);
|
}
|
|
private static LambdaQueryWrapper<CusItemSyncView> viewSelectWrapper() {
|
return Wrappers.<CusItemSyncView>lambdaQuery()
|
.select(CusItemSyncView::getItemNo, CusItemSyncView::getItemSpec, CusItemSyncView::getUnitNo);
|
}
|
|
private List<Map<String, Object>> toViewMaps(List<CusItemSyncView> rows) {
|
if (rows == null || rows.isEmpty()) {
|
return Collections.emptyList();
|
}
|
return rows.stream().map(this::toViewRowMap).collect(Collectors.toList());
|
}
|
|
private Map<String, Object> toViewRowMap(CusItemSyncView row) {
|
Map<String, Object> m = new LinkedHashMap<>(3);
|
m.put("item_no", row.getItemNo());
|
m.put("item_spec", row.getItemSpec());
|
m.put("unit_no", row.getUnitNo());
|
return m;
|
}
|
}
|
