make a use query after receiving the initialize schema in planetscale

prisma · Nov 6, 2023 · 4a4f468 · 4a4f468
1 parent 662c7bb
commit 4a4f468
Showing 1 changed file with 16 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,20 @@ 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("---- Changing database used by proxy to the one specified in the URL: "+databaseName)
+        let request = {
+            method: 'executeRaw',
+            params: {
+                query: 'USE ' + databaseName + ';',
+            }
+        };
+        await engine.query(JSON.stringify(request), "", "")
+    }
+}
+
 async function adapterFromEnv(url: string): Promise<DriverAdapter> {
     const adapter = process.env.DRIVER_ADAPTER ?? ''