From d7626830d4aa52a3fdf12c57df4db0f161f7580f Mon Sep 17 00:00:00 2001
From: "Jason T. Greene" <jason.greene@redhat.com>
Date: Fri, 12 Oct 2018 14:35:40 -0500
Subject: [PATCH] Implement env based override for DatasourceProducer

---
 .../jboss/shamrock/agroal/runtime/DataSourceProducer.java | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/agroal/runtime/src/main/java/org/jboss/shamrock/agroal/runtime/DataSourceProducer.java b/agroal/runtime/src/main/java/org/jboss/shamrock/agroal/runtime/DataSourceProducer.java
index a151aa209aea1..3f79628e6163e 100644
--- a/agroal/runtime/src/main/java/org/jboss/shamrock/agroal/runtime/DataSourceProducer.java
+++ b/agroal/runtime/src/main/java/org/jboss/shamrock/agroal/runtime/DataSourceProducer.java
@@ -79,8 +79,14 @@ public AgroalDataSource getDatasource() throws SQLException {
                 throw new RuntimeException("Driver is an XA datasource and xa has been configured");
             }
         }
+
+        String targetUrl = System.getenv("DATASOURCE_URL");
+        if (targetUrl == null || targetUrl.isEmpty()) {
+            targetUrl = url;
+        }
+
         AgroalDataSourceConfigurationSupplier dataSourceConfiguration = new AgroalDataSourceConfigurationSupplier();
-        dataSourceConfiguration.connectionPoolConfiguration().connectionFactoryConfiguration().jdbcUrl(url);
+        dataSourceConfiguration.connectionPoolConfiguration().connectionFactoryConfiguration().jdbcUrl(targetUrl);
         dataSourceConfiguration.connectionPoolConfiguration().connectionFactoryConfiguration().connectionProviderClass(providerClass);
 
         if (jta || xa) {