Cara ideal untuk melakukannya adalah, menangkap informasi terkait skema dari permintaan dan menyimpannya di ThreadLocal dan mengatur skema setiap kali koneksi diminta. Sayangnya ketika saya mencoba pendekatan ini, saya menemukan metode setSchema belum diterapkan di driver. Tetapi saya menemukan cara lain (retas) untuk menyelesaikan ini. JDBI menyediakan Locator pernyataan yang dapat kita gunakan di sini untuk menyelesaikan masalah ini.
Katakanlah kita mengirim nama skema dalam Parameter kueri, kita dapat menggunakan filter permintaan jersey untuk mendapatkan nama skema.
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;
}
}
Ini akan mendapatkan nama skema pada setiap permintaan. Daftarkan filer ini pada bootstrap aplikasi Anda.
environment.jersey().property(ResourceConfig.PROPERTY_CONTAINER_REQUEST_FILTERS, asList(new SchemaNameFilter()));
Sekarang kita perlu menulis bagian kedua, di mana kita harus menggunakan informasi skema ini. Sertakan SchemaRewriter ini,
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;
}
}
Katakanlah kita ingin mengakses tabel "pengguna" yang ada di semua skema, tulis kueri seperti ini.
@OverrideStatementLocatorWith(SchemaReWriter.class)
public interface UserDao {
@SqlQuery("select * from :schema.users")
public List<User> getAllUsers();
}
Jangan lupa untuk membubuhi keterangan Dao dengan StatementRewriter. Itu saja. Anda tidak perlu khawatir tentang banyak skema.
