Tutorial: Getting Started with the AWS Advanced JDBC Driver, Spring Boot and Hibernate for load-balanced write and read-only connections (Two Datasources)
In this tutorial, you will set up a Spring Boot and Hibernate application with the AWS Advanced JDBC Driver, and use two datasources to fetch and update data from an Aurora PostgreSQL database. One datasource is configured to provide a writer connection. The other datasource is configured to provide a reader connection to off load a writer node from read-only queries. Both datasources provide pooled connections through AWS Advanced JDBC Driver internal connection pool configuration.
Note: this tutorial was written using the following technologies:
- Spring Boot 2.7.1
- Hibernate
- AWS JDBC Driver 2.3.2
- Postgresql 42.5.4
- Gradle 7
- Java 11
You will progress through the following sections:
- Create a Gradle Spring Boot project
- Add the required Gradle dependencies
- Configure the AWS Advanced JDBC Driver
- This tutorial uses the Aurora PostgreSQL database.
Create a Gradle project with the following project hierarchy:
└───src
└───main
├───java
│ └───example
│ ├───data
│ │ ├───Book.java
│ │ ├───BookRepository.java
│ │ └───BookService.java
│ └───spring
│ ├───Config.java
│ ├───LoadBalancedReaderDataSourceContext.java
│ ├───RoutingDataSource.java
│ ├───ShouldRetryTransactionException.java
│ ├───SpringHibernateBalancedReaderTwoDataSourceExampleApplication.java
│ ├───WithLoadBalancedDataSourceInterception.java
│ └───WithLoadBalancedReaderDataSource.java
└───resources
└───application.yml
Note: this sample code assumes the target database contains a table named
Book
that can be generated using the SQL queries provided insrc/main/resources/books.sql
.
In your build.gradle.kts
, add the following dependencies.
dependencies {
implementation("org.springframework.boot:spring-boot-starter-jdbc")
implementation("org.springframework.retry:spring-retry")
implementation("org.postgresql:postgresql")
implementation("software.amazon.jdbc:aws-advanced-jdbc-wrapper:latest")
}
Please note that the sample code inside the AWS JDBC Driver project will use the dependency implementation(project(":aws-advanced-jdbc-wrapper"))
instead of implementation("software.amazon.jdbc:aws-advanced-jdbc-wrapper:latest")
as seen above.
Configure Spring to use the AWS JDBC Driver as the default datasource.
-
In the
application.yml
, add new datasources for Spring:spring: datasource: writer-datasource: url: jdbc:aws-wrapper:postgresql://db-identifier.cluster-XYZ.us-east-2.rds.amazonaws.com:5432/test_db?wrapperProfileName=SF_F0 username: dev_user password: dev_password driver-class-name: software.amazon.jdbc.Driver type: org.springframework.jdbc.datasource.SimpleDriverDataSource load-balanced-reader-datasource: url: jdbc:aws-wrapper:postgresql://db-identifier.cluster-ro-XYZ.us-east-2.rds.amazonaws.com:5432/test_db?wrapperProfileName=SF_F0&readerInitialConnectionHostSelectorStrategy=roundRobin username: dev_user password: dev_password driver-class-name: software.amazon.jdbc.Driver type: org.springframework.jdbc.datasource.SimpleDriverDataSource
-
The datasources mentioned above do not use Hikari datasources which are the default for Spring 2+ applications. The AWS JDBC Driver manages its own internal connection pool (or several connection pools, if needed), which increases overall efficiency and helps facilitate failover support. All necessary configuration parameters are defined in
SF_F0
configuration profile. Other configuration presets fromSF_
family can be used as well. More details are available at Configuration Profiles and Configuration Presets.
Optional configuration parameterreaderInitialConnectionHostSelectorStrategy
in connection string helps to setup a strategy selecting a reader node. Possible values arerandom
,roundRobin
andleastConnections
. More details are available at Reader Selection Strategies. -
Configure Hibernate dialect:
jpa: properties: hibernate: dialect: org.hibernate.dialect.PostgreSQLDialect
-
[Optional] You can enable driver logging by adding the following to
application.yml
:logging: level: software: amazon: jdbc: INFO jdbc.states: INFO example: TRACE
For detailed logs use TRACE
for software.amazon.jdbc
package.
Start the application by running ./gradlew :springhibernatetwodatasource:bootRun
in the terminal. You should see the application making a connection to the database and fetching data from the Book
table.
This tutorial walks through the steps required to add and configure the AWS JDBC Driver to a simple Spring Boot and Hibernate application that load balances connections.