Capitalize page permission key conf

trinodb · May 22, 2024 · ca07cfd · ca07cfd
1 parent c00dc26
commit ca07cfd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/gateway-ha/src/main/java/io/trino/gateway/ha/security/LbFormAuthManager.java b/gateway-ha/src/main/java/io/trino/gateway/ha/security/LbFormAuthManager.java
@@ -27,6 +27,7 @@
 import io.trino.gateway.ha.domain.request.RestLoginRequest;
 
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -48,7 +49,10 @@ public LbFormAuthManager(FormAuthConfiguration configuration,
             Map<String, String> pagePermissions)
     {
         this.presetUsers = presetUsers;
-        this.pagePermissions = pagePermissions;
+        this.pagePermissions = new HashMap<>();
+        if (pagePermissions != null) {
+            pagePermissions.forEach((key, value) -> this.pagePermissions.put(key.toUpperCase(), value));
+        }
 
         if (configuration != null) {
             this.lbKeyProvider = new LbKeyProvider(configuration

diff --git a/gateway-ha/src/main/java/io/trino/gateway/ha/security/LbOAuthManager.java b/gateway-ha/src/main/java/io/trino/gateway/ha/security/LbOAuthManager.java
@@ -40,6 +40,7 @@
 import java.net.URI;
 import java.security.interfaces.RSAPublicKey;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -61,7 +62,10 @@ public class LbOAuthManager
     public LbOAuthManager(OAuthConfiguration configuration, Map<String, String> pagePermissions)
     {
         this.oauthConfig = configuration;
-        this.pagePermissions = pagePermissions;
+        this.pagePermissions = new HashMap<>();
+        if (pagePermissions != null) {
+            pagePermissions.forEach((key, value) -> this.pagePermissions.put(key.toUpperCase(), value));
+        }
     }
 
     public String getUserIdField()