make a use query after receiving the initialize schema in planetscale

prisma · Nov 6, 2023 · 347e535 · 347e535
1 parent 662c7bb
commit 347e535
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/query-engine/driver-adapters/connector-test-kit-executor/src/index.ts b/query-engine/driver-adapters/connector-test-kit-executor/src/index.ts
@@ -97,6 +97,8 @@ async function handleRequest(method: string, params: unknown): Promise<unknown>
             });
             await engine.connect("")
 
+            await useSchema(engine, castParams.url)
+
             state[castParams.schemaId] = {
                 engine,
                 adapter,
@@ -222,6 +224,26 @@ async function initQe(url: string, prismaSchema: string, logCallback: qe.QueryLo
     return [engineInstance, errorCapturingAdapter];
 }
 
+async function useSchema(engine: engines.QueryEngineInstance, url: string): Promise<void> {
+    if (process.env.DRIVER_ADAPTER == 'planetscale') {
+        const databaseName = new URL(url).pathname.replace("/", "");
+        console.error("Using database: ", databaseName)
+        let request = {
+            action: 'executeRaw',
+            query: {
+                arguments: {
+                    query: 'USE '+ databaseName,
+                    parameters: '[]',
+                },
+                selection: {
+                    $scalars: true,
+                },
+            },
+        };
+        await engine.query(JSON.stringify(request), "", "")
+    }
+}
+
 async function adapterFromEnv(url: string): Promise<DriverAdapter> {
     const adapter = process.env.DRIVER_ADAPTER ?? ''