Spring Batch JPA继承场景下如何实现无需判断实体类型的通用保存方法

方案思路 #

利用Spring的依赖注入特性收集所有子类对应的服务实例，提前构建实体类型和服务的映射关系，运行时直接根据实体类型匹配对应服务，全程无需手动编写类型判断分支。

具体实现步骤 #

1. 扩展基础服务接口，增加类型匹配方法 #

在AbstractBaseService中新增supports方法，用于判断当前服务是否支持处理传入的实体类型：

public interface AbstractBaseService<T extends AbstractBaseDocument, ID extends Serializable> {
    T saveAndFlush(T entity);
    // 新增：判断当前服务是否支持处理指定类型的实体
    boolean supports(Class<?> entityClazz);
}

2. 各子类服务实现类型匹配方法 #

分别在SaillieDocumentServiceImpl和InscriptionDocumentServiceImpl中实现上述方法：

// SaillieDocumentServiceImpl 中新增
@Override
public boolean supports(Class<?> entityClazz) {
    return SaillieDocument.class.isAssignableFrom(entityClazz);
}

// InscriptionDocumentServiceImpl 中新增
@Override
public boolean supports(Class<?> entityClazz) {
    return InscriptionDocument.class.isAssignableFrom(entityClazz);
}

3. 改造Writer类，实现通用保存逻辑 #

注入所有基础服务的实现类，提前构建类型和服务的映射缓存，运行时直接匹配调用：

// 注入所有AbstractBaseService的实现实例
@Autowired
private List<AbstractBaseService<? extends AbstractBaseDocument, ? extends Serializable>> allBaseServices;

// 实体类型 -> 对应服务的缓存映射
private Map<Class<? extends AbstractBaseDocument>, AbstractBaseService> entityServiceCache;

// 类初始化时构建缓存
@PostConstruct
public void initEntityServiceCache() {
    entityServiceCache = new HashMap<>();
    for (AbstractBaseService service : allBaseServices) {
        // 这里直接用supports方法匹配，不用硬编码判断服务类型
        if (service.supports(SaillieDocument.class)) {
            entityServiceCache.put(SaillieDocument.class, service);
        } else if (service.supports(InscriptionDocument.class)) {
            entityServiceCache.put(InscriptionDocument.class, service);
        }
        // 后续新增子类只需要在这里加一行映射即可，不需要修改保存逻辑
    }
}

@Override
public void write(List<? extends AbstractBaseDocument> documents) throws Exception {
    for (AbstractBaseDocument document : documents) {
        log.info(">>> Sauvegarde dans la table temporaire {}", document.toString());
        // 直接根据实体类型获取对应服务
        AbstractBaseService matchedService = entityServiceCache.get(document.getClass());
        if (matchedService == null) {
            throw new IllegalArgumentException("未找到适配实体类型的服务：" + document.getClass().getName());
        }
        // 无需强转，直接调用保存方法
        matchedService.saveAndFlush(document);
    }
}

优化提示 #

如果不想每次新增子类都修改initEntityServiceCache方法，可以通过反射获取每个服务的泛型参数类型自动构建映射，进一步降低耦合度，适配更多子类扩展场景。

内容的提问来源于stack exchange，提问作者anthony44