add enmu DatastoreBackend, add javadoc

Author: felicity3786

gateway-ha/src/main/java/io/trino/gateway/ha/config/DataStoreBackend.java

@@ -0,0 +1,21 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.trino.gateway.ha.config;
+
+public enum DataStoreBackend {
+    ORACLE,
+    MYSQL,
+    POSTGRES,
+    H2
+}

gateway-ha/src/main/java/io/trino/gateway/ha/config/DataStoreConfiguration.java

@@ -111,7 +111,7 @@ public MySqlConfiguration getMySqlConfiguration()
         return mySqlConfiguration;
     }
 
-    public void setMysqlConfiguration(MySqlConfiguration mySqlConfig)
+    public void setMySqlConfiguration(MySqlConfiguration mySqlConfig)
     {
         this.mySqlConfiguration = mySqlConfig;
     }

gateway-ha/src/main/java/io/trino/gateway/ha/config/HaGatewayConfiguration.java

@@ -43,7 +43,6 @@ public class HaGatewayConfiguration
     private List<String> statementPaths = ImmutableList.of(V1_STATEMENT_PATH);
     private boolean includeClusterHostInResponse;
     private ProxyResponseConfiguration proxyResponseConfiguration = new ProxyResponseConfiguration();
-    private boolean oracleBackend;
 
     private RequestAnalyzerConfig requestAnalyzerConfig = new RequestAnalyzerConfig();
 
@@ -298,14 +297,4 @@ public HaGatewayConfigurationException(String message)
             super(message);
         }
     }
-
-    public boolean isOracleBackend()
-    {
-        return oracleBackend;
-    }
-
-    public void setOracleBackend(boolean oracleBackend)
-    {
-        this.oracleBackend = oracleBackend;
-    }
 }

gateway-ha/src/test/java/io/trino/gateway/ha/router/BaseTestQueryHistoryManager.java

@@ -14,7 +14,6 @@
 package io.trino.gateway.ha.router;
 
 import io.trino.gateway.ha.config.DataStoreConfiguration;
-import io.trino.gateway.ha.config.HaGatewayConfiguration;
 import io.trino.gateway.ha.domain.response.DistributionResponse;
 import io.trino.gateway.ha.persistence.FlywayMigration;
 import io.trino.gateway.ha.persistence.JdbcConnectionManager;
@@ -51,10 +50,7 @@ void setUp()
                 true);
         FlywayMigration.migrate(config);
         JdbcConnectionManager jdbcConnectionManager = createTestingJdbcConnectionManager(container, config);
-        HaGatewayConfiguration haGatewayConfiguration = new HaGatewayConfiguration();
-        haGatewayConfiguration.setDataStore(config);
-        haGatewayConfiguration.setOracleBackend(container.getJdbcUrl().startsWith("jdbc:oracle"));
-        queryHistoryManager = new HaQueryHistoryManager(jdbcConnectionManager.getJdbi(), config);
+        queryHistoryManager = new HaQueryHistoryManager(jdbcConnectionManager.getJdbi(), jdbcConnectionManager.getConfiguration());
     }
 
     @AfterAll