You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@drill.apache.org by cg...@apache.org on 2022/05/03 14:20:40 UTC

[drill] branch master updated: [MINOR UPDATE] Fix default auth mode in AuthMode.parseOrDefault and test the case of JDBC. (#2528)

This is an automated email from the ASF dual-hosted git repository.

cgivre pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/drill.git


The following commit(s) were added to refs/heads/master by this push:
     new d21b92da41 [MINOR UPDATE] Fix default auth mode in AuthMode.parseOrDefault and test the case of JDBC. (#2528)
d21b92da41 is described below

commit d21b92da4103d8c1ee695cf75f3edfef15f3efe1
Author: James Turton <91...@users.noreply.github.com>
AuthorDate: Tue May 3 16:20:33 2022 +0200

    [MINOR UPDATE] Fix default auth mode in AuthMode.parseOrDefault and test the case of JDBC. (#2528)
---
 .../org/apache/drill/exec/store/http/HttpStoragePluginConfig.java  | 2 +-
 .../java/org/apache/drill/exec/store/jdbc/JdbcStorageConfig.java   | 2 +-
 .../test/java/org/apache/drill/exec/store/jdbc/TestDataSource.java | 3 ++-
 .../java/org/apache/drill/common/logical/StoragePluginConfig.java  | 7 ++++---
 4 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/contrib/storage-http/src/main/java/org/apache/drill/exec/store/http/HttpStoragePluginConfig.java b/contrib/storage-http/src/main/java/org/apache/drill/exec/store/http/HttpStoragePluginConfig.java
index 3af4ed2bed..6bd9475199 100644
--- a/contrib/storage-http/src/main/java/org/apache/drill/exec/store/http/HttpStoragePluginConfig.java
+++ b/contrib/storage-http/src/main/java/org/apache/drill/exec/store/http/HttpStoragePluginConfig.java
@@ -82,7 +82,7 @@ public class HttpStoragePluginConfig extends CredentialedStoragePluginConfig {
         normalize(proxyPassword),
         credentialsProvider),
         credentialsProvider == null,
-        AuthMode.parseOrDefault(authMode)
+        AuthMode.parseOrDefault(authMode, AuthMode.SHARED_USER)
     );
     this.cacheResults = cacheResults != null && cacheResults;
 
diff --git a/contrib/storage-jdbc/src/main/java/org/apache/drill/exec/store/jdbc/JdbcStorageConfig.java b/contrib/storage-jdbc/src/main/java/org/apache/drill/exec/store/jdbc/JdbcStorageConfig.java
index d6deb64964..b8d915c49e 100644
--- a/contrib/storage-jdbc/src/main/java/org/apache/drill/exec/store/jdbc/JdbcStorageConfig.java
+++ b/contrib/storage-jdbc/src/main/java/org/apache/drill/exec/store/jdbc/JdbcStorageConfig.java
@@ -70,7 +70,7 @@ public class JdbcStorageConfig extends CredentialedStoragePluginConfig {
     super(
       CredentialProviderUtils.getCredentialsProvider(username, password, credentialsProvider),
       credentialsProvider == null,
-      AuthMode.parseOrDefault(authMode)
+      AuthMode.parseOrDefault(authMode, AuthMode.SHARED_USER)
     );
     this.driver = driver;
     this.url = url;
diff --git a/contrib/storage-jdbc/src/test/java/org/apache/drill/exec/store/jdbc/TestDataSource.java b/contrib/storage-jdbc/src/test/java/org/apache/drill/exec/store/jdbc/TestDataSource.java
index 2e345ead40..a9c35d771d 100644
--- a/contrib/storage-jdbc/src/test/java/org/apache/drill/exec/store/jdbc/TestDataSource.java
+++ b/contrib/storage-jdbc/src/test/java/org/apache/drill/exec/store/jdbc/TestDataSource.java
@@ -55,12 +55,13 @@ public class TestDataSource extends BaseTest {
   @Test
   public void testInitWithoutUserAndPassword() {
     JdbcStorageConfig config = new JdbcStorageConfig(
-      DRIVER, url, null, null, false, false, null, null, AuthMode.SHARED_USER.name(), 1000);
+      DRIVER, url, null, null, false, false, null, null, null, 1000);
     try (HikariDataSource dataSource = JdbcStoragePlugin.initDataSource(config, null)) {
       assertEquals(DRIVER, dataSource.getDriverClassName());
       assertEquals(url, dataSource.getJdbcUrl());
       assertNull(dataSource.getUsername());
       assertNull(dataSource.getPassword());
+      assertEquals(config.getAuthMode(), AuthMode.SHARED_USER);
     }
   }
 
diff --git a/logical/src/main/java/org/apache/drill/common/logical/StoragePluginConfig.java b/logical/src/main/java/org/apache/drill/common/logical/StoragePluginConfig.java
index 32154fa394..b5c265e010 100644
--- a/logical/src/main/java/org/apache/drill/common/logical/StoragePluginConfig.java
+++ b/logical/src/main/java/org/apache/drill/common/logical/StoragePluginConfig.java
@@ -32,7 +32,8 @@ public abstract class StoragePluginConfig {
   // comparisons; doing so will break the plugin registry.
   protected Boolean enabled;
 
-  protected AuthMode authMode;
+  // The overridable default plugin auth mode is DRILL_PROCESS
+  protected AuthMode authMode = AuthMode.DRILL_PROCESS;
 
   /**
    * Check for enabled status of the plugin
@@ -105,8 +106,8 @@ public abstract class StoragePluginConfig {
      */
     USER_TRANSLATION;
 
-    public static AuthMode parseOrDefault(String authMode) {
-      return !Strings.isNullOrEmpty(authMode) ? AuthMode.valueOf(authMode.toUpperCase()) : DRILL_PROCESS;
+    public static AuthMode parseOrDefault(String authMode, AuthMode defavlt) {
+      return !Strings.isNullOrEmpty(authMode) ? AuthMode.valueOf(authMode.toUpperCase()) : defavlt;
     }
   }
 }