#2270 configuring host

Stirling-Tools · Dec 13, 2024 · 5615b37 · 5615b37
1 parent 69604ea
commit 5615b37
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/main/java/stirling/software/SPDF/config/security/database/DatabaseConfig.java b/src/main/java/stirling/software/SPDF/config/security/database/DatabaseConfig.java
@@ -15,6 +15,8 @@
 @Configuration
 public class DatabaseConfig {
 
+    public static final String DOCKER_HOST = "://db:";
+    public static final String LOCALHOST = "://localhost:";
     private final ApplicationProperties applicationProperties;
 
     public DatabaseConfig(@Autowired ApplicationProperties applicationProperties) {
@@ -27,16 +29,20 @@ public Connection connection() throws SQLException {
                 applicationProperties.getSystem().getDatasource();
 
         DataSourceBuilder<?> dataSourceBuilder = DataSourceBuilder.create();
-        dataSourceBuilder.driverClassName("org.postgresql.Driver");
-        // todo: revert back to ->
-        // dataSourceBuilder.driverClassName(getDriverClassName(datasource.getDriver()));
-        dataSourceBuilder.url(datasource.getUrl());
+        dataSourceBuilder.driverClassName(getDriverClassName(datasource.getDriver()));
+        dataSourceBuilder.url(getUrlForProfile(datasource));
         dataSourceBuilder.username(datasource.getUsername());
         dataSourceBuilder.password(datasource.getPassword());
 
         return dataSourceBuilder.build().getConnection();
     }
 
+    private String getUrlForProfile(ApplicationProperties.Datasource datasource) {
+        return applicationProperties.getSystem().getProfilesActive().equals("default")
+                ? datasource.getUrl().replace(DOCKER_HOST, LOCALHOST)
+                : datasource.getUrl();
+    }
+
     private String getDriverClassName(ApplicationProperties.Driver driverName) {
         switch (driverName) {
             case ORACLE -> {