Add schema to SessionConfigurationContext

prestodb · Jan 15, 2025 · fed9c5d · fed9c5d
1 parent 2827fbb
commit fed9c5d
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 0 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/Session.java b/presto-main/src/main/java/com/facebook/presto/Session.java
@@ -803,6 +803,11 @@ public Identity getIdentity()
             return this.identity;
         }
 
+        public Optional getSchema()
+        {
+            return Optional.ofNullable(this.schema);
+        }
+
         public Optional<String> getSource()
         {
             return Optional.ofNullable(this.source);

diff --git a/presto-main/src/main/java/com/facebook/presto/server/SessionPropertyDefaults.java b/presto-main/src/main/java/com/facebook/presto/server/SessionPropertyDefaults.java
@@ -118,6 +118,7 @@ public void applyDefaultProperties(
 
         SessionConfigurationContext context = new SessionConfigurationContext(
                 sessionBuilder.getIdentity().getUser(),
+                sessionBuilder.getSchema(),
                 sessionBuilder.getSource(),
                 sessionBuilder.getClientTags(),
                 queryType,

diff --git a/...ty-managers/src/test/java/com/facebook/presto/session/TestFileSessionPropertyManager.java b/...ty-managers/src/test/java/com/facebook/presto/session/TestFileSessionPropertyManager.java
@@ -40,6 +40,7 @@ public class TestFileSessionPropertyManager
 {
     private static final SessionConfigurationContext CONTEXT = new SessionConfigurationContext(
             "user",
+            Optional.empty(),
             Optional.of("source"),
             ImmutableSet.of("tag1", "tag2"),
             Optional.of(QueryType.DATA_DEFINITION.toString()),

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/session/SessionConfigurationContext.java b/presto-spi/src/main/java/com/facebook/presto/spi/session/SessionConfigurationContext.java
@@ -25,6 +25,7 @@
 public final class SessionConfigurationContext
 {
     private final String user;
+    private final Optional<String> schema;
     private final Optional<String> source;
     private final Set<String> clientTags;
     private final Optional<String> queryType;
@@ -34,6 +35,7 @@ public final class SessionConfigurationContext
 
     public SessionConfigurationContext(
             String user,
+            Optional<String> schema,
             Optional<String> source,
             Set<String> clientTags,
             Optional<String> queryType,
@@ -42,6 +44,7 @@ public SessionConfigurationContext(
             String prestoServerVersion)
     {
         this.user = requireNonNull(user, "user is null");
+        this.schema = requireNonNull(schema, "schema is null");
         this.source = requireNonNull(source, "source is null");
         this.clientTags = unmodifiableSet(new HashSet<>(requireNonNull(clientTags, "clientTags is null")));
         this.queryType = requireNonNull(queryType, "queryType is null");
@@ -55,6 +58,11 @@ public String getUser()
         return user;
     }
 
+    public Optional<String> getSchema()
+    {
+        return schema;
+    }
+
     public Optional<String> getSource()
     {
         return source;