Changed setGetTableFunction() to be a noop if tables is already set.

aws · Jan 30, 2023 · 9a61559 · 9a61559
1 parent c5899db
commit 9a61559
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/software/amazon/documentdb/jdbc/metadata/DocumentDbSchema.java b/src/main/java/software/amazon/documentdb/jdbc/metadata/DocumentDbSchema.java
@@ -133,7 +133,9 @@ public void setGetTableFunction(
             @NonNull final Function<String, DocumentDbSchemaTable> getTableFunction,
             @NonNull final Function<Set<String>, Map<String, DocumentDbSchemaTable>> getRemainingTablesFunction)
             throws IllegalStateException {
-        if (this.tables != null || this.tableReferences == null) {
+        if (this.tables != null) {
+            return;
+        } else if (this.tableReferences == null) {
             throw new IllegalStateException(
                     SqlError.lookup(SqlError.INVALID_STATE_SET_TABLE_FUNCTION));
         }