代码优化
This commit is contained in:
spllzh
360
src/main/java/com/rj/common/DataEnrichmentUtil.java
Normal file
360
src/main/java/com/rj/common/DataEnrichmentUtil.java
Normal file
@@ -0,0 +1,360 @@
|
||||
package com.rj.common;
|
||||
|
||||
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
||||
import com.baomidou.mybatisplus.extension.service.IService;
|
||||
import com.rj.entity.Dealership;
|
||||
import com.rj.entity.ProjectManagement;
|
||||
import com.rj.entity.SalesManagement;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import java.util.*;
|
||||
import java.util.function.Function;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* 数据丰富工具类
|
||||
* 用于批量获取关联数据并填充到主数据中，减少代码冗余
|
||||
*
|
||||
* @author 系统生成
|
||||
* @since 2025-08-07
|
||||
*/
|
||||
@Component
|
||||
public class DataEnrichmentUtil {
|
||||
|
||||
@Autowired
|
||||
private ServiceManager serviceManager;
|
||||
|
||||
/**
|
||||
* 批量获取关联数据并创建映射
|
||||
*
|
||||
* @param ids 主键ID列表
|
||||
* @param service 对应的Service
|
||||
* @param idGetter ID获取函数
|
||||
* @param nameGetter 名称获取函数
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID到名称的映射
|
||||
*/
|
||||
public static <T, ID> Map<ID, String> batchGetNameMap(
|
||||
Collection<ID> ids,
|
||||
IService<T> service,
|
||||
Function<T, ID> idGetter,
|
||||
Function<T, String> nameGetter) {
|
||||
|
||||
if (ids == null || ids.isEmpty()) {
|
||||
return new HashMap<>();
|
||||
}
|
||||
|
||||
LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
|
||||
// 使用lambda表达式来避免类型问题
|
||||
queryWrapper.in(entity -> idGetter.apply(entity), ids);
|
||||
List<T> entities = service.list(queryWrapper);
|
||||
|
||||
return entities.stream()
|
||||
.collect(Collectors.toMap(
|
||||
idGetter,
|
||||
nameGetter,
|
||||
(existing, replacement) -> existing // 如果有重复key，保留第一个
|
||||
));
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取关联数据并创建映射（支持自定义查询条件）
|
||||
*
|
||||
* @param ids 主键ID列表
|
||||
* @param service 对应的Service
|
||||
* @param idGetter ID获取函数
|
||||
* @param nameGetter 名称获取函数
|
||||
* @param additionalCondition 额外的查询条件
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID到名称的映射
|
||||
*/
|
||||
public static <T, ID> Map<ID, String> batchGetNameMap(
|
||||
Collection<ID> ids,
|
||||
IService<T> service,
|
||||
Function<T, ID> idGetter,
|
||||
Function<T, String> nameGetter,
|
||||
Function<LambdaQueryWrapper<T>, LambdaQueryWrapper<T>> additionalCondition) {
|
||||
|
||||
if (ids == null || ids.isEmpty()) {
|
||||
return new HashMap<>();
|
||||
}
|
||||
|
||||
LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
|
||||
// 使用lambda表达式来避免类型问题
|
||||
queryWrapper.in(ids.size() > 0, entity -> idGetter.apply(entity), ids);
|
||||
|
||||
if (additionalCondition != null) {
|
||||
queryWrapper = additionalCondition.apply(queryWrapper);
|
||||
}
|
||||
|
||||
List<T> entities = service.list(queryWrapper);
|
||||
|
||||
return entities.stream()
|
||||
.collect(Collectors.toMap(
|
||||
idGetter,
|
||||
nameGetter,
|
||||
(existing, replacement) -> existing
|
||||
));
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取关联数据并创建映射（支持多个字段）
|
||||
*
|
||||
* @param ids 主键ID列表
|
||||
* @param service 对应的Service
|
||||
* @param idGetter ID获取函数
|
||||
* @param fieldGetters 字段获取函数映射
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID到字段值的映射
|
||||
*/
|
||||
public static <T, ID> Map<ID, Map<String, Object>> batchGetFieldMap(
|
||||
Collection<ID> ids,
|
||||
IService<T> service,
|
||||
Function<T, ID> idGetter,
|
||||
Map<String, Function<T, Object>> fieldGetters) {
|
||||
|
||||
if (ids == null || ids.isEmpty()) {
|
||||
return new HashMap<>();
|
||||
}
|
||||
|
||||
LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
|
||||
// 使用lambda表达式来避免类型问题
|
||||
queryWrapper.in(ids.size() > 0, entity -> idGetter.apply(entity), ids);
|
||||
List<T> entities = service.list(queryWrapper);
|
||||
|
||||
return entities.stream()
|
||||
.collect(Collectors.toMap(
|
||||
idGetter,
|
||||
entity -> {
|
||||
Map<String, Object> fieldMap = new HashMap<>();
|
||||
fieldGetters.forEach((fieldName, getter) ->
|
||||
fieldMap.put(fieldName, getter.apply(entity)));
|
||||
return fieldMap;
|
||||
},
|
||||
(existing, replacement) -> existing
|
||||
));
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取关联数据并创建映射（支持自定义查询条件和多个字段）
|
||||
*
|
||||
* @param ids 主键ID列表
|
||||
* @param service 对应的Service
|
||||
* @param idGetter ID获取函数
|
||||
* @param fieldGetters 字段获取函数映射
|
||||
* @param additionalCondition 额外的查询条件
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID到字段值的映射
|
||||
*/
|
||||
public static <T, ID> Map<ID, Map<String, Object>> batchGetFieldMap(
|
||||
Collection<ID> ids,
|
||||
IService<T> service,
|
||||
Function<T, ID> idGetter,
|
||||
Map<String, Function<T, Object>> fieldGetters,
|
||||
Function<LambdaQueryWrapper<T>, LambdaQueryWrapper<T>> additionalCondition) {
|
||||
|
||||
if (ids == null || ids.isEmpty()) {
|
||||
return new HashMap<>();
|
||||
}
|
||||
|
||||
LambdaQueryWrapper<T> queryWrapper = new LambdaQueryWrapper<>();
|
||||
// 使用lambda表达式来避免类型问题
|
||||
queryWrapper.in(ids.size() > 0, entity -> idGetter.apply(entity), ids);
|
||||
|
||||
if (additionalCondition != null) {
|
||||
queryWrapper = additionalCondition.apply(queryWrapper);
|
||||
}
|
||||
|
||||
List<T> entities = service.list(queryWrapper);
|
||||
|
||||
return entities.stream()
|
||||
.collect(Collectors.toMap(
|
||||
idGetter,
|
||||
entity -> {
|
||||
Map<String, Object> fieldMap = new HashMap<>();
|
||||
fieldGetters.forEach((fieldName, getter) ->
|
||||
fieldMap.put(fieldName, getter.apply(entity)));
|
||||
return fieldMap;
|
||||
},
|
||||
(existing, replacement) -> existing
|
||||
));
|
||||
}
|
||||
|
||||
/**
|
||||
* 从实体列表中提取指定字段的ID列表
|
||||
*
|
||||
* @param entities 实体列表
|
||||
* @param idGetter ID获取函数
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID列表
|
||||
*/
|
||||
public static <T, ID> List<ID> extractIds(
|
||||
Collection<T> entities,
|
||||
Function<T, ID> idGetter) {
|
||||
|
||||
if (entities == null || entities.isEmpty()) {
|
||||
return new ArrayList<>();
|
||||
}
|
||||
|
||||
return entities.stream()
|
||||
.map(idGetter)
|
||||
.filter(Objects::nonNull)
|
||||
.distinct()
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
/**
|
||||
* 从实体列表中提取指定字段的ID列表（支持过滤条件）
|
||||
*
|
||||
* @param entities 实体列表
|
||||
* @param idGetter ID获取函数
|
||||
* @param filter 过滤条件
|
||||
* @param <T> 实体类型
|
||||
* @param <ID> ID类型
|
||||
* @return ID列表
|
||||
*/
|
||||
public static <T, ID> List<ID> extractIds(
|
||||
Collection<T> entities,
|
||||
Function<T, ID> idGetter,
|
||||
Function<T, Boolean> filter) {
|
||||
|
||||
if (entities == null || entities.isEmpty()) {
|
||||
return new ArrayList<>();
|
||||
}
|
||||
|
||||
return entities.stream()
|
||||
.filter(filter::apply)
|
||||
.map(idGetter)
|
||||
.filter(Objects::nonNull)
|
||||
.distinct()
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
// ==================== 便捷方法，使用ServiceManager ====================
|
||||
|
||||
/**
|
||||
* 批量获取门店名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getDealershipIdNameMap(Collection<String> dealershipIds) {
|
||||
|
||||
// 批量获取门店信息
|
||||
Map<String, String> dealershipNameMap;
|
||||
if (!dealershipIds.isEmpty()) {
|
||||
LambdaQueryWrapper<Dealership> dealershipQueryWrapper = new LambdaQueryWrapper<>();
|
||||
dealershipQueryWrapper.in(Dealership::getId, dealershipIds);
|
||||
List<Dealership> dealerships = serviceManager.getDealershipService().list(dealershipQueryWrapper);
|
||||
dealershipNameMap = dealerships.stream()
|
||||
.collect(Collectors.toMap(Dealership::getId, Dealership::getDealershipName));
|
||||
} else {
|
||||
dealershipNameMap = new HashMap<>();
|
||||
}
|
||||
return dealershipNameMap;
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取项目名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getProjectIdNameMap(Collection<String> projectIds) {
|
||||
|
||||
Map<String, String> projectNameMap;
|
||||
if (!projectIds.isEmpty()) {
|
||||
LambdaQueryWrapper<ProjectManagement> projectManagementQueryWrapper = new LambdaQueryWrapper<>();
|
||||
projectManagementQueryWrapper.in(ProjectManagement::getId, projectIds);
|
||||
List<ProjectManagement> projectManagementList = serviceManager.getProjectManagementService().list(projectManagementQueryWrapper);
|
||||
projectNameMap = projectManagementList.stream()
|
||||
.collect(Collectors.toMap(ProjectManagement::getId, ProjectManagement::getProjectName));
|
||||
} else {
|
||||
projectNameMap = new HashMap<>();
|
||||
}
|
||||
|
||||
|
||||
return projectNameMap;
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取销售名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getSalesIdNameMap(Collection<String> salesIds) {
|
||||
|
||||
// 批量获取 销售信息
|
||||
Map<String, String> saleNameMap;
|
||||
if (!salesIds.isEmpty()) {
|
||||
LambdaQueryWrapper<SalesManagement> salesQueryWrapper = new LambdaQueryWrapper<>();
|
||||
salesQueryWrapper.in(SalesManagement::getId, salesIds);
|
||||
List<SalesManagement> sales = serviceManager.getSalesManagementService().list(salesQueryWrapper);
|
||||
saleNameMap = sales.stream()
|
||||
.collect(Collectors.toMap(SalesManagement::getId, SalesManagement::getSalesName));
|
||||
} else {
|
||||
saleNameMap = new HashMap<>();
|
||||
}
|
||||
|
||||
|
||||
return saleNameMap;
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取客户名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getCustomerNameMap(Collection<String> customerIds) {
|
||||
return batchGetNameMap(
|
||||
customerIds,
|
||||
serviceManager.getCustomerManagementService(),
|
||||
com.rj.entity.CustomerManagement::getId,
|
||||
com.rj.entity.CustomerManagement::getCustomerName
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取设备名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getDeviceNameMap(Collection<String> deviceIds) {
|
||||
return batchGetNameMap(
|
||||
deviceIds,
|
||||
serviceManager.getDeviceManagementService(),
|
||||
com.rj.entity.DeviceManagement::getId,
|
||||
com.rj.entity.DeviceManagement::getDeviceCode
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取音频名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getAudioNameMap(Collection<String> audioIds) {
|
||||
return batchGetNameMap(
|
||||
audioIds,
|
||||
serviceManager.getAudioManagementService(),
|
||||
com.rj.entity.AudioManagement::getId,
|
||||
com.rj.entity.AudioManagement::getRecordingName
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取视频名称映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<String, String> getVideoNameMap(Collection<String> videoIds) {
|
||||
return batchGetNameMap(
|
||||
videoIds,
|
||||
serviceManager.getVideoManagementService(),
|
||||
com.rj.entity.VideoManagement::getId,
|
||||
com.rj.entity.VideoManagement::getVideoName
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* 批量获取预约信息映射（使用ServiceManager）
|
||||
*/
|
||||
public Map<Long, String> getReservationNameMap(Collection<Long> reservationIds) {
|
||||
return batchGetNameMap(
|
||||
reservationIds,
|
||||
serviceManager.getReservationService(),
|
||||
com.rj.pojo.Reservation::getId,
|
||||
com.rj.pojo.Reservation::getName
|
||||
);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user