add sslmode for postgress DE-6429

razorpay · Aug 7, 2024 · a9e1a85 · a9e1a85
1 parent 6f1b8a3
commit a9e1a85
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sql-delta-import/src/main/scala/com/razorpay/spark/jdbc/JDBCImport.scala b/sql-delta-import/src/main/scala/com/razorpay/spark/jdbc/JDBCImport.scala
@@ -141,8 +141,11 @@ class JDBCImport(
     val schema = importConfig.schema
 
     val connectionUrl = s"jdbc:$dbType://$host:$port/$database"
-
-    connectionUrl
+    if (dbType == Constants.POSTGRESQL) {
+      s"$connectionUrl?sslmode=disable"
+    } else {
+      connectionUrl
+    }
   }
 
   private lazy val sourceDataframe = readJDBCSourceInParallel()