Spring Boot-如何创建不同JpaRepository接口的工厂

Question

我有一个服务层，可以像这样自动连接存储库

@Service
public class OrderService {

    @Autowired
    private OrderRepository repository;

    public List<Order> retrieveOrders(String storeId) {
        try {

            return repository.getOrders(storeId)

        } catch(Exception e) {
            ...
        }
    }
}

我有了新的要求，根据商店的数量，存储库查询逻辑应该改变。我正在考虑抽象我的存储库并使用工厂模式。当前存储库是一个接口，并扩展了JpaRepository此外，它的所有函数都使用@Query注释来定义JPQL。这是一个只有一个函数的例子，但是在我的实际存储库中有几个函数

public interface OrderRepository extends JpaRepository {

    @Query("select o " +
        " from Orders o " +
        " where o.storeId = :storeId")
    public List<Order> getOrders(@Param(value = "storeId") String storeId);

}

新要求指出，对于特定的 storeId，SQL 需要更改为如下所示的内容：

public interface OrderRepositoryStore886 extends JpaRepository {

    @Query("select o " +
        " from Order o " +
        " where o.storeId = :storeId " +
        "     and o.status IN (1,3,10,15) " +
        "     and EXISTS (SELECT c from CollabOrder WHERE c.orderId = o.orderId)")
    public List<Order> getOrders(@Param(value = "storeId") String storeId);

}

但是，所有剩余的存储都应该使用第一个SQL查询。

我正在尝试弄清楚如何根据 storeId 有条件地@Autowire存储库。我正在考虑创建一个工厂，但我不知道如何创建一个返回接口对象的工厂（我对 java 仍然相当陌生，这可能微不足道，但我还没有弄清楚）。如果我尝试将存储库接口转换为类，它会迫使我覆盖所有抽象的 JpaRepository 方法，这是不必要的额外代码。

Answer 1

你已经走上正轨了。我会解决这个问题，但是创建一个返回正确bean接口的工厂方法。

这种方法的唯一缺点是你不能自动连接你的豆子。您应该即时实例化它们。

create table orders
(
  id                      integer      not null,
  constraint pk_orders primary key(id)
);

@Entity
@Table(name = "orders")
@ToString
@Getter
@Setter
public class Order {

    @Id
    private long id;
}


public interface OrderRepository extends JpaRepository<Order, Long> {
    @Query("SELECT t FROM Order t")
    Order getOrder(String storeId);
}


@Repository
public interface OrderRepositoryStore100 extends OrderRepository {
    @Query("SELECT t FROM Order t WHERE t.id = 100")
    Order getOrder(@Param(value = "id") String storeId);
}

@Repository
public interface OrderRepositoryStore286 extends OrderRepository {
    @Query("SELECT t FROM Order t WHERE t.id = 286")
    Order getOrder(@Param(value = "id") String storeId);
}

public class OrderRepositoryStoreFactory {

    private static final String REPOSITORY_BASE_NAME = "orderRepositoryStore";

    public static OrderRepository getRepository(ApplicationContext context, String storeId) {

        return (OrderRepository) context.getBean(REPOSITORY_BASE_NAME + storeId);
    }
}

@Service
public class OrderService {

    @Autowired
    private ApplicationContext context;

    public Order retrieveOrders(String storeId) {
        try {

            OrderRepository repository = OrderRepositoryStoreFactory.getRepository(context, storeId);

            return repository.getOrder(storeId);

        } catch(Exception e) {
            return null;
        }
    }
}



@SpringBootApplication
public class FactoryApplication implements CommandLineRunner {

    @Autowired
    private ApplicationContext context;

    @Autowired
    private OrderService orderService;

    public static void main(String[] args) {
        SpringApplication.run(FactoryApplication.class, args);
    }

    @Override
    public void run(String... args) throws Exception {
        Order order = orderService.retrieveOrders("100");
        System.out.println(order);

        order = orderService.retrieveOrders("286");
        System.out.println(order);
    }
}

下面是输出：

2020-05-03 20：23：34.567 信息 33955 --- [ main] com.factory.FactoryApplication ： 在 3.003 秒内启动工厂应用程序（JVM 运行为 3.615） 订单（id=100） 订单（id=286）

希望这有帮助