Spring Boot中基于设计模式与配置实现动态Repository选择

public interface BaseRepository {
    /**
     * 保存MyPojo对象
     * @param myPojo 待保存的对象
     * @return 保存后的MyPojo对象
     */
    MyPojo save(MyPojo myPojo);
}

import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;

// 示例：Elasticsearch Repository
@Repository("elasticRepository") // 指定Bean名称
public interface ElasticRepository extends JpaRepository<MyPojo, String>, BaseRepository {
    @Override
    default MyPojo save(MyPojo myPojo) {
        // 实际的Elasticsearch保存逻辑，可能需要转换为Elasticsearch特定的文档
        System.out.println("Saving MyPojo to Elasticsearch: " + myPojo.getId());
        // JpaRepository的save方法返回的是实体本身，这里为简化示例直接返回
        return JpaRepository.super.save(myPojo);
    }
}

// 示例：MongoDB Repository
import org.springframework.data.mongodb.repository.MongoRepository;

@Repository("mongoDbRepository") // 指定Bean名称
public interface MongoDbRepository extends MongoRepository<MyPojo, String>, BaseRepository {
    @Override
    default MyPojo save(MyPojo myPojo) {
        // 实际的MongoDB保存逻辑
        System.out.println("Saving MyPojo to MongoDB: " + myPojo.getId());
        return MongoRepository.super.save(myPojo);
    }
}

// 示例：Redis Repository
// 假设RedisRepository没有直接继承Spring Data Redis的接口，而是自定义实现
@Repository("redisRepository") // 指定Bean名称
public class RedisRepositoryImpl implements BaseRepository {
    // 注入RedisTemplate或其他Redis客户端
    // @Autowired private RedisTemplate<String, MyPojo> redisTemplate;

    @Override
    public MyPojo save(MyPojo myPojo) {
        // 实际的Redis保存逻辑
        System.out.println("Saving MyPojo to Redis: " + myPojo.getId());
        // redisTemplate.opsForValue().set(myPojo.getId(), myPojo);
        return myPojo;
    }
}

public interface BaseRepositoryFactory {
    /**
     * 根据存储目标名称获取对应的BaseRepository实例。
     * ServiceLocatorFactoryBean会根据方法参数（通常是beanName）查找对应的Bean。
     *
     * @param whereToSave 存储目标的名称，应与具体Repository的Bean名称匹配
     * @return 对应的BaseRepository实例
     */
    BaseRepository getBaseRepository(String whereToSave);
}

import org.springframework.beans.factory.config.ServiceLocatorFactoryBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class RepositoryFactoryConfig {

    @Bean
    public ServiceLocatorFactoryBean baseRepositoryBean() {
        ServiceLocatorFactoryBean serviceLocatorFactoryBean = new ServiceLocatorFactoryBean();
        // 指定ServiceLocatorFactoryBean需要实现的工厂接口
        serviceLocatorFactoryBean.setServiceLocatorInterface(BaseRepositoryFactory.class);
        // 可以选择性配置keyMethod，默认为"get<ServiceLocatorInterfaceMethodName>"
        // 例如，如果方法是getBaseRepository(String key)，默认查找的是key对应的bean
        return serviceLocatorFactoryBean;
    }
}

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.UUID;

@RestController
public class MyController {

    // 注入我们配置的Repository工厂
    @Autowired
    private BaseRepositoryFactory baseRepositoryFactory;

    @PostMapping(path = "/save")
    public String save(@RequestBody MyRequest myRequest) {
        String whereToSave = myRequest.getWhereToSave(); // 从请求中获取存储目标
        MyPojo myPojo = new MyPojo(UUID.randomUUID().toString(), myRequest.getValue());

        try {
            // 通过工厂动态获取对应的Repository
            BaseRepository selectedRepository = baseRepositoryFactory.getBaseRepository(whereToSave);
            // 使用获取到的Repository保存数据
            MyPojo savedPojo = selectedRepository.save(myPojo);
            return "Successfully saved to " + whereToSave + ": " + savedPojo.getId();
        } catch (Exception e) {
            // 处理Repository未找到或保存失败的情况
            return "Failed to save to " + whereToSave + ": " + e.getMessage();
        }
    }
}

// 假设MyRequest和MyPojo的定义如下：
class MyRequest {
    private String whereToSave;
    private String value;

    public String getWhereToSave() { return whereToSave; }
    public void setWhereToSave(String whereToSave) { this.whereToSave = whereToSave; }
    public String getValue() { return value; }
    public void setValue(String value) { this.value = value; }
}

class MyPojo {
    private String id;
    private String value;

    public MyPojo(String id, String value) {
        this.id = id;
        this.value = value;
    }

    public String getId() { return id; }
    public void setId(String id) { this.id = id; }
    public String getValue() { return value; }
    public void setValue(String value) { this.value = value; }

    @Override
    public String toString() {
        return "MyPojo{" +
               "id='" + id + '\'' +
               ", value='" + value + '\'' +
               '}';
    }
}