Merge branch 'override-schema-service' of https://github.com/normanj-…

…bitquill/amazon-documentdb-jdbc-driver into override-schema-service
aws · Jan 30, 2023 · 337f412 · 337f412
2 parents 701e8b0 + b7e4346
commit 337f412
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));
         }