From 463e6ebf4b4e8209a7dbb3bf7818544fa68dca5f Mon Sep 17 00:00:00 2001
From: Pere Miquel Brull <peremiquelbrull@gmail.com>
Date: Mon, 1 Sep 2025 11:04:24 +0200
Subject: [PATCH] Fix aurora driver settings

---
 conf/openmetadata.yaml                                           | 1 +
 .../openmetadata/service/jdbi3/HikariCPDataSourceFactory.java    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/conf/openmetadata.yaml b/conf/openmetadata.yaml
index 534263bf6da..26be21921a2 100644
--- a/conf/openmetadata.yaml
+++ b/conf/openmetadata.yaml
@@ -253,6 +253,7 @@ database:
     readWriteSplitting.connectionPoolSize: ${DB_AWS_RWS_CONNECTION_POOL_SIZE:-10}
     readWriteSplitting.maxIdleTime: ${DB_AWS_RWS_MAX_IDLE_TIME:-300000}
     readWriteSplittingConnectionStrategy: ${DB_AWS_RWS_CONNECTION_STRATEGY:-leastConnections}
+    readWriteSplitting.splitSqlByReadWritePattern: ${DB_AWS_RWS_SPLIT_SQL:-true}
     
     # Failover configuration
     failover.enableClusterAwareFailover: ${DB_AWS_FAILOVER_CLUSTER_AWARE:-true}
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/HikariCPDataSourceFactory.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/HikariCPDataSourceFactory.java
index 83e2c3202d3..a8c9d7980d8 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/HikariCPDataSourceFactory.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/HikariCPDataSourceFactory.java
@@ -248,6 +248,7 @@ public class HikariCPDataSourceFactory extends DataSourceFactory {
     props.put("readWriteSplitting.connectionPoolSize", "10");
     props.put("readWriteSplitting.maxIdleTime", "300000");
     props.put("readWriteSplittingConnectionStrategy", "leastConnections");
+    props.put("readWriteSplitting.splitSqlByReadWritePattern", "true");
 
     // Default failover configuration
     props.put("failover.enableClusterAwareFailover", "true");