Set the origin in the QueryContext

polypheny · Dec 13, 2023 · bba6e49 · bba6e49
1 parent e782eb8
commit bba6e49
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/...e/src/main/java/org/polypheny/db/protointerface/statementProcessing/MongoImplementer.java b/...e/src/main/java/org/polypheny/db/protointerface/statementProcessing/MongoImplementer.java
@@ -53,8 +53,13 @@ public void implement( PIStatement piStatement ) throws PIServiceException {
                     9001
             );
         }
-        String query = piStatement.getQuery();
-        List<ImplementationContext> implementations = LanguageManager.getINSTANCE().anyPrepareQuery( QueryContext.builder().query( query ).language( QueryLanguage.from( "mql" ) ).namespaceId( piStatement.getNamespace().id ).build(), statement );
+        QueryContext context = QueryContext.builder()
+                .query( piStatement.getQuery() )
+                .language( QueryLanguage.from( "mql" ) )
+                .namespaceId( piStatement.getNamespace().id )
+                .origin( ORIGIN )
+                .build();
+        List<ImplementationContext> implementations = LanguageManager.getINSTANCE().anyPrepareQuery( context, statement );
 
         piStatement.setImplementation( implementations.get( 0 ).getImplementation() );
     }

diff --git a/...ace/src/main/java/org/polypheny/db/protointerface/statementProcessing/SqlImplementer.java b/...ace/src/main/java/org/polypheny/db/protointerface/statementProcessing/SqlImplementer.java
@@ -53,8 +53,13 @@ public void implement( PIStatement piStatement ) throws PIServiceException {
                     9003
             );
         }
-        String query = piStatement.getQuery();
-        List<ImplementationContext> implementations = LanguageManager.getINSTANCE().anyPrepareQuery( QueryContext.builder().query( query ).language( QueryLanguage.from( "sql" ) ).namespaceId( piStatement.getNamespace().id ).build(), statement );
+        QueryContext context = QueryContext.builder()
+                .query( piStatement.getQuery() )
+                .language( QueryLanguage.from( "sql" ) )
+                .namespaceId( piStatement.getNamespace().id )
+                .origin( ORIGIN )
+                .build();
+        List<ImplementationContext> implementations = LanguageManager.getINSTANCE().anyPrepareQuery( context, statement );
         piStatement.setImplementation( implementations.get( 0 ).getImplementation() );
     }
 

diff --git a/...c/main/java/org/polypheny/db/protointerface/statementProcessing/StatementImplementer.java b/...c/main/java/org/polypheny/db/protointerface/statementProcessing/StatementImplementer.java
@@ -21,12 +21,16 @@
 
 public abstract class StatementImplementer {
 
-    protected boolean hasInvalidLanguage(PIStatement piStatement) {
+    protected final String ORIGIN = "Proto-Interface";
+
+
+    protected boolean hasInvalidLanguage( PIStatement piStatement ) {
         return piStatement.getLanguage() != getLanguage();
     }
 
+
     abstract QueryLanguage getLanguage();
 
-    abstract void implement(PIStatement piStatement );
+    abstract void implement( PIStatement piStatement );
 
 }