SpringBoot中基于设计模式和配置动态选择Repository策略

// BaseRepository.java
public interface BaseRepository {
    MyPojo save(MyPojo pojo); // 假设所有Repository都有一个save方法
    // 可以在这里添加其他通用方法
}

// ARepository.java
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.stereotype.Repository;

@Repository("repoA") // 指定唯一的bean名称
public interface ARepository extends JpaRepository<MyPojo, String>, BaseRepository {
    // JpaRepository提供了save方法，所以这里无需额外实现
    // 如果没有继承CrudRepository等，需要自行实现save方法
    @Override
    default MyPojo save(MyPojo pojo) {
        // 示例：实际应调用JpaRepository的save方法
        return JpaRepository.super.save(pojo);
    }
}

// BRepository.java
@Repository("repoB") // 指定唯一的bean名称
public interface BRepository extends JpaRepository<MyPojo, String>, BaseRepository {
    @Override
    default MyPojo save(MyPojo pojo) {
        return JpaRepository.super.save(pojo);
    }
}

// 假设ElasticRepository不继承CrudRepository，需要自己实现save
// ElasticRepository.java
@Repository("elasticRepository")
public class ElasticRepository implements BaseRepository {
    // 注入Elasticsearch客户端等
    // ...
    @Override
    public MyPojo save(MyPojo pojo) {
        // 实现Elasticsearch的保存逻辑
        System.out.println("Saving to Elastic: " + pojo.getValue());
        return pojo;
    }
}

// MongoDbRepository.java
@Repository("mongoDbRepository")
public interface MongoDbRepository extends MongoRepository<MyPojo, String>, BaseRepository {
    @Override
    default MyPojo save(MyPojo pojo) {
        return MongoRepository.super.save(pojo);
    }
}

// RedisRepository.java
@Repository("redisRepository")
public class RedisRepository implements BaseRepository {
    // 注入RedisTemplate等
    // ...
    @Override
    public MyPojo save(MyPojo pojo) {
        // 实现Redis的保存逻辑
        System.out.println("Saving to Redis: " + pojo.getValue());
        return pojo;
    }
}

// BaseRepositoryFactory.java
public interface BaseRepositoryFactory {
    BaseRepository getBaseRepository(String whereToSave);
}

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

@Configuration
public class AppConfig {

    @Bean
    public ServiceLocatorFactoryBean baseRepositoryBean() {
        ServiceLocatorFactoryBean serviceLocatorFactoryBean = new ServiceLocatorFactoryBean();
        // 设置工厂接口，ServiceLocatorFactoryBean会自动查找Spring上下文中匹配的Bean
        serviceLocatorFactoryBean.setServiceLocatorInterface(BaseRepositoryFactory.class);
        return serviceLocatorFactoryBean;
    }
}

// ControllerVersionThree.java
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 ControllerVersionThree {

    @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 {
            // 根据whereToSave参数动态获取Repository实例
            BaseRepository repository = baseRepositoryFactory.getBaseRepository(whereToSave);
            if (repository != null) {
                return repository.save(myPojo).toString();
            } else {
                return "Unknown destination or repository not found for: " + whereToSave;
            }
        } catch (Exception e) {
            // 处理Repository未找到或保存失败的情况
            return "Error saving to " + whereToSave + ": " + e.getMessage();
        }
    }
}

// MyRequest.java
public class MyRequest {
    private String whereToSave;
    private String value;
    // Getters and Setters
    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; }
}

// MyPojo.java
public class MyPojo {
    private String id;
    private String value;
    // Constructor
    public MyPojo(String id, String value) {
        this.id = id;
        this.value = value;
    }
    // Getters and Setters
    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 + '\'' + '}'; }
}