chore: update Hibernate sample code (#368)

Update Hibernate sample code to reflect changes in the wrapper source code.

examples/HibernateExample/build.gradle

@@ -13,9 +13,9 @@ dependencies {
     annotationProcessor 'org.projectlombok:lombok:1.18.24'
     implementation 'org.hibernate:hibernate-core:6.1.3.Final'
     implementation 'jakarta.persistence:jakarta.persistence-api:3.1.0'
-    runtimeOnly 'org.postgresql:postgresql:42.5.1'
-    runtimeOnly 'mysql:mysql-connector-java:8.0.30'
-    runtimeOnly project(':aws-advanced-jdbc-wrapper')
+    runtimeOnly 'org.postgresql:postgresql:42.5.4'
+    runtimeOnly 'mysql:mysql-connector-java:8.0.32'
+    implementation project(':aws-advanced-jdbc-wrapper')
     testImplementation 'org.junit.jupiter:junit-jupiter-api:5.9.0'
     testRuntimeOnly 'org.junit.jupiter:junit-jupiter-engine:5.9.0'
 }

examples/HibernateExample/src/main/java/software/amazon/JpaExample.java

@@ -16,68 +16,77 @@
 
 package software.amazon;
 
+import java.time.Duration;
+import java.time.Instant;
+import org.hibernate.Hibernate;
 import jakarta.persistence.EntityManager;
 import jakarta.persistence.EntityManagerFactory;
 import jakarta.persistence.Persistence;
-import org.hibernate.Hibernate;
 import software.amazon.entity.Address;
 import software.amazon.entity.User;
 
-import java.sql.SQLException;
-
 public class JpaExample {
-    private static EntityManagerFactory entityManagerFactory = null;
 
-    public static void main(String[] args) {
-        try {
-            Address address = new Address();
-            address.setTown("Orangeville");
-            address.setStreet("Faulkner");
-            address.setCountry("Canada");
-            address.setPostal("L1L2M2");
-            insertAddress(address);
-            User user = new User();
-            user.setFirst("Dave");
-            user.setLast("Cramer");
-            user.setAddress(address);
-            insertUser(user);
+  private static final EntityManagerFactory entityManagerFactory = Persistence.createEntityManagerFactory("Atlas");
+
+  public static void main(String[] args) {
+    final Instant start = Instant.now();
+    final Duration threshold = Duration.ofMinutes(5);
+
+    int i = 0;
+    try {
+      // Keep inserting data for 5 minutes.
+      while (Duration.between(start, Instant.now()).toMinutes() > threshold.toMinutes()) {
+        Address address = new Address();
+        address.setTown("Orangeville");
+        address.setStreet("Faulkner");
+        address.setCountry("Canada");
+        address.setPostal("L1L2M2");
+
+        insertAddress(address);
 
-            while (true) {
-                User user1 = getUser(user.getId());
-            }
-        } catch (Exception ex ){
-            // exception should be a TransactionStateUnknownException on failover
-            ex.printStackTrace();
-        }
+        User user = new User();
+        user.setFirst("First" + i);
+        user.setLast("Last" + i);
+        user.setAddress(address);
+        insertUser(user);
+
+        i++;
+      }
+    } catch (Exception ex) {
+      // Exception should be a FailoverSuccessSQLException if failover occurred.
+      // For more information regarding FailoverSuccessSQLException please visit the driver's documentation.
+      ex.printStackTrace();
     }
-    private static void insertUser(User user) throws Exception {
-        try (EntityManager entityManager = openEntityManager()) {
-            entityManager.getTransaction().begin();
-            entityManager.persist(user);
-            entityManager.getTransaction().commit();
-        }
+  }
+
+  private static void insertUser(User user) {
+    try (EntityManager entityManager = openEntityManager()) {
+      entityManager.getTransaction().begin();
+      entityManager.persist(user);
+      entityManager.getTransaction().commit();
     }
-    private static void insertAddress(Address address) throws Exception {
-        try (EntityManager entityManager = openEntityManager()) {
-            entityManager.getTransaction().begin();
-            entityManager.persist(address);
-            entityManager.getTransaction().commit();
-        }
+  }
+
+  private static void insertAddress(Address address) {
+    try (EntityManager entityManager = openEntityManager()) {
+      entityManager.getTransaction().begin();
+      entityManager.persist(address);
+      entityManager.getTransaction().commit();
     }
-    private static User getUser(int id) throws SQLException {
-        try (EntityManager entityManager = openEntityManager()) {
+  }
 
-            User user = entityManager.find(User.class, id);
-            Hibernate.initialize(user);
+  private static User getUser(int id) {
+    try (EntityManager entityManager = openEntityManager()) {
 
-            return user;
-        }
-    }
+      User user = entityManager.find(User.class, id);
+      Hibernate.initialize(user);
 
-    private static EntityManager openEntityManager() {
-        if (entityManagerFactory == null) {
-            entityManagerFactory = Persistence.createEntityManagerFactory( "Atlas" );
-        }
-        return entityManagerFactory.createEntityManager();
+      return user;
     }
+  }
+
+  private static EntityManager openEntityManager() {
+    return entityManagerFactory.createEntityManager();
+  }
 }

examples/HibernateExample/src/main/resources/META-INF/persistence.xml

@@ -9,13 +9,12 @@
 
     <properties>
 
-        <property name="hibernate.dialect" value="software.amazon.utility.AuroraPostgreSQLDialect"/>
+        <property name="hibernate.dialect" value="org.hibernate.dialect.PostgreSQLDialect"/>
         <property name="hibernate.connection.driver_class" value="software.amazon.jdbc.Driver"/>
-        <property name="hibernate.connection.url" value="jdbc:aws-wrapper:postgresql://hostname.amazonaws.com/hibernate_orm_test"/>
+        <property name="hibernate.connection.url" value="jdbc:aws-wrapper:postgresql://db.cluster-xyz.region.rds.amazonaws.com/hibernate_orm_test"/>
         <property name="hibernate.connection.username" value="hibernate_orm_test"/>
         <property name="hibernate.connection.password" value="hibernate_orm_test"/>
         <property name="hibernate.connection.jdbcProtocol" value="jdbc:postgresql://"/>
-        <property name="hibernate.connection.wrapperPlugins" value="failover"/>
 
         <property name="hibernate.connection.pool_size" value="5"/>
 

examples/HikariExample/build.gradle.kts

@@ -15,8 +15,8 @@
  */
 
 dependencies {
-    implementation("org.postgresql:postgresql:42.5.0")
-    implementation("mysql:mysql-connector-java:8.0.31")
+    implementation("org.postgresql:postgresql:42.5.4")
+    implementation("mysql:mysql-connector-java:8.0.32")
     implementation(project(":aws-advanced-jdbc-wrapper"))
     implementation("com.zaxxer:HikariCP:4.0.3")
 }