理想的方法是,从请求中捕获架构相关信息并将其保存在 ThreadLocal 中,并在请求连接时设置架构。
不幸的是,当我尝试这种方法时,我发现 setSchema 方法尚未在驱动程序中实现。但我找到了另一种方法(hack)来解决这个问题。
JDBI 提供了语句定位器,我们可以在这里使用它来解决这个问题。
假设我们在查询参数中发送模式名称,我们可以使用 jersey 请求过滤器来获取模式名称。
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;
}
}
假设我们要访问所有模式中的“用户”表,这样编写查询。
@OverrideStatementLocatorWith(SchemaReWriter.class)
public interface UserDao {
@SqlQuery("select * from :schema.users")
public List<User> getAllUsers();
}
不要忘记使用 StatementRewriter 注释 Dao。就这样。您无需担心多个架构。