|
25 | 25 | import java.util.Arrays; |
26 | 26 | import java.util.Map; |
27 | 27 | import java.util.Set; |
28 | | -import org.apache.commons.lang3.StringUtils; |
29 | 28 | import org.apache.gravitino.Catalog; |
30 | | -import org.apache.gravitino.auth.AuthenticatorType; |
31 | 29 | import org.apache.gravitino.client.DefaultOAuth2TokenProvider; |
32 | 30 | import org.apache.gravitino.client.GravitinoAdminClient; |
33 | 31 | import org.apache.gravitino.client.GravitinoMetalake; |
@@ -66,7 +64,7 @@ private GravitinoCatalogManager( |
66 | 64 |
|
67 | 65 | // Only OAuth is explicitly configured; otherwise follow Flink security (Kerberos if enabled, |
68 | 66 | // simple auth otherwise). |
69 | | - if (AuthenticatorType.OAUTH.name().equalsIgnoreCase(authType)) { |
| 67 | + if (GravitinoCatalogStoreFactoryOptions.OAUTH2.equalsIgnoreCase(authType)) { |
70 | 68 | this.gravitinoClient = buildOAuthClient(gravitinoUri, gravitinoClientConfig); |
71 | 69 | } else { |
72 | 70 | if (authType != null) { |
@@ -247,14 +245,19 @@ private static GravitinoAdminClient buildOAuthClient( |
247 | 245 | String credential = config.get(GravitinoCatalogStoreFactoryOptions.OAUTH2_CREDENTIAL); |
248 | 246 | String path = config.get(GravitinoCatalogStoreFactoryOptions.OAUTH2_TOKEN_PATH); |
249 | 247 | String scope = config.get(GravitinoCatalogStoreFactoryOptions.OAUTH2_SCOPE); |
250 | | - Preconditions.checkArgument( |
251 | | - StringUtils.isNoneBlank(serverUri, credential, path, scope), |
252 | | - String.format( |
253 | | - "OAuth2 authentication requires: %s, %s, %s, and %s", |
| 248 | + |
| 249 | + // Remove OAuth config from client config map, Because |
| 250 | + // GravitinoAdminClient.builder().withClientConfig() does not recognize these keys. |
| 251 | + Set<String> oauthConfigKeys = |
| 252 | + Sets.newHashSet( |
| 253 | + GravitinoCatalogStoreFactoryOptions.AUTH_TYPE, |
254 | 254 | GravitinoCatalogStoreFactoryOptions.OAUTH2_SERVER_URI, |
255 | 255 | GravitinoCatalogStoreFactoryOptions.OAUTH2_CREDENTIAL, |
256 | 256 | GravitinoCatalogStoreFactoryOptions.OAUTH2_TOKEN_PATH, |
257 | | - GravitinoCatalogStoreFactoryOptions.OAUTH2_SCOPE)); |
| 257 | + GravitinoCatalogStoreFactoryOptions.OAUTH2_SCOPE); |
| 258 | + for (String key : oauthConfigKeys) { |
| 259 | + config.remove(key); |
| 260 | + } |
258 | 261 |
|
259 | 262 | DefaultOAuth2TokenProvider provider = |
260 | 263 | DefaultOAuth2TokenProvider.builder() |
|
0 commit comments