Question

我正在使用DropWizard（使用JDBI）构建Java Rest API，我的要求是我需要使用相同的应用程序查询多个MySQL模式。它基本上是一个包含多个模式的AWS MySQL实例 - 每个客户端一个模式。

我需要的是一种机制，它知道哪个＆＃34;架构＆＃34;根据请求进行查询 - IE：请求所属的客户端。

我知道如何创建DataSource，DAO等（使用本教程：https://dropwizard.github.io/dropwizard/manual/jdbi.html），但不知道如何查询多个模式。

有什么想法吗？

Answer 1

执行此操作的理想方法是，从请求中捕获与架构相关的信息，并将其保存在ThreadLocal中，并在请求连接时设置架构。不幸的是，当我尝试这种方法时，我发现setSchema方法尚未在驱动程序中实现。但我找到了解决这个问题的另一种方法（黑客）。 JDBI提供了语句Locator，我们可以在这里解决这个问题。

假设我们在查询参数中发送模式名称，我们可以使用泽西请求过滤器来获取模式名称。

public class Schema {
    public static ThreadLocal<String> name = new ThreadLocal<>();
}


public class SchemaNameFilter implements ContainerRequestFilter {

    @Override
    public ContainerRequest filter(ContainerRequest request) {
        if(request.getQueryParameters().containsKey("schema")) {
            Schema.name.set(request.getQueryParameters().get("schema").get(0));
        }
        return request;
    }
}

这将获得每个请求的架构名称。在您的应用程序引导程序中注册此文件管理器。

environment.jersey().property(ResourceConfig.PROPERTY_CONTAINER_REQUEST_FILTERS, asList(new SchemaNameFilter()));

现在我们需要编写第二部分，我们应该使用这个架构信息。包括此SchemaRewriter，

public class SchemaReWriter implements StatementLocator {
    @Override
    public String locate(String sql, StatementContext ctx) throws Exception {
        if (nonNull(Schema.name.get())) {
            sql = sql.replaceAll(":schema", Schema.name.get());
        }
        return sql;
    }
}

让我们说我们想访问表＆＃34;用户＆＃34;在所有模式中，写这样的查询。

@OverrideStatementLocatorWith(SchemaReWriter.class)
public interface UserDao {

  @SqlQuery("select * from :schema.users")
  public List<User> getAllUsers();

}

别忘了用StatementRewriter注释Dao。这就是全部。您不必担心多个模式。

Answer 2

最简单的解决方案是使用多个：

  @JsonProperty("database")
    public DataSourceFactory getDataSourceFactory() {
        return database;
    }

配置，例如：

@JsonProperty("products")
@JsonProperty("clients")

然后配置文件：

products:
  # products schema configuration...
clients:
  # clients configuration

使用Dropwizard＆amp; JDBI用多个模式查询数据库？

2 个答案: