fix: playground csv loading (#3076)

PRQL · Jul 25, 2023 · c0d769a · c0d769a
1 parent 88064da
commit c0d769a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 13 deletions.
diff --git a/web/playground/src/workbench/Workbench.js b/web/playground/src/workbench/Workbench.js
@@ -40,7 +40,9 @@ class Workbench extends React.Component {
   componentDidMount() {
     this.props.setCallables({ loadFile: (f, c) => this.loadFile(f, c) });
 
-    this.duckdb = duckdb.init();
+    if (!this.duckdb) {
+      this.duckdb = duckdb.init();
+    }
   }
 
   beforeEditorMount(monaco) {

diff --git a/web/playground/src/workbench/duckdb.js b/web/playground/src/workbench/duckdb.js
@@ -40,24 +40,18 @@ export const CHINOOK_TABLES = [
 
 async function registerChinook(db) {
   const baseUrl = `${window.location.href}data/chinook`;
-  const http = duckdb.DuckDBDataProtocol.HTTP;
 
   await Promise.all(
-    CHINOOK_TABLES.map((table) =>
-      db.registerFileURL(
-        `${table}.csv`,
-        `${baseUrl}/${table}.csv`,
-        http,
-        false,
-      ),
-    ),
+    CHINOOK_TABLES.map(async (table) => {
+      const res = await fetch(`${baseUrl}/${table}.csv`);
+
+      db.registerFileText(`${table}.csv`, res);
+    }),
   );
 
   const c = await db.connect();
   for (const table of CHINOOK_TABLES) {
-    await c.query(`
-      CREATE TABLE ${table} AS SELECT * FROM read_csv_auto('${table}.csv');
-    `);
+    await c.insertCSVFromPath(`${table}.csv`, { name: table, detect: true });
   }
   c.close();
 }