@visactor/vquery 0.1.33 → 0.1.35

package/dist/db/duckDb.d.ts

@@ -1,3 +1,4 @@
+import { QueryResult } from '../types';
 export declare class DuckDB {
     private db;
     private connection;
@@ -5,7 +6,7 @@ export declare class DuckDB {
     /**
      * @description 初始化 DuckDB 实例
      */
-    instantiate: () => Promise<void>;
+    init: () => Promise<void>;
     /**
      * @description 释放 DuckDB 实例
      */
@@ -25,4 +26,15 @@ export declare class DuckDB {
         dataset: any[];
         table: any;
     }>;
+    /**
+     * 确保一个文件存在，如果不存在，则根据同名文件创建临时表
+     * @param fileName 文件名
+     */
+    private ensureSchema;
+    /**
+     * @description 获取文件的 Schema
+     * @param fileName 文件名
+     * @returns 文件的 Schema
+     */
+    getSchema: (fileName: string) => Promise<QueryResult>;
 }

package/dist/index.cjs

@@ -1,4 +1,7 @@
 "use strict";
+const __rslib_import_meta_url__ = /*#__PURE__*/ function() {
+    return 'undefined' == typeof document ? new (require('url'.replace('', ''))).URL('file:' + __filename).href : document.currentScript && document.currentScript.src || new URL('main.js', document.baseURI).href;
+}();
 var __webpack_require__ = {};
 (()=>{
     __webpack_require__.d = (exports1, definition)=>{
@@ -31,9 +34,18 @@ class DuckDB {
     db = null;
     connection = null;
     constructor(){}
-    instantiate = async ()=>{
-        const JSDELIVR_BUNDLES = (0, duckdb_wasm_namespaceObject.getJsDelivrBundles)();
-        const bundle = await (0, duckdb_wasm_namespaceObject.selectBundle)(JSDELIVR_BUNDLES);
+    init = async ()=>{
+        const MANUAL_BUNDLES = {
+            mvp: {
+                mainModule: new URL('@duckdb/duckdb-wasm/dist/duckdb-mvp.wasm', __rslib_import_meta_url__).href,
+                mainWorker: new URL('@duckdb/duckdb-wasm/dist/duckdb-browser-mvp.worker.js', __rslib_import_meta_url__).toString()
+            },
+            eh: {
+                mainModule: new URL('@duckdb/duckdb-wasm/dist/duckdb-eh.wasm', __rslib_import_meta_url__).href,
+                mainWorker: new URL('@duckdb/duckdb-wasm/dist/duckdb-browser-eh.worker.js', __rslib_import_meta_url__).toString()
+            }
+        };
+        const bundle = await (0, duckdb_wasm_namespaceObject.selectBundle)(MANUAL_BUNDLES);
         const worker_url = URL.createObjectURL(new Blob([
             `importScripts("${bundle.mainWorker}");`
         ], {
@@ -80,6 +92,16 @@ class DuckDB {
             table
         };
     };
+    ensureSchema = async (fileName)=>{
+        if (!this.connection) throw new Error('connection is null');
+        await this.connection.query(`CREATE TEMP TABLE IF NOT EXISTS "${fileName}" AS SELECT * FROM read_csv_auto('${fileName}')`);
+    };
+    getSchema = async (fileName)=>{
+        if (!this.connection) throw new Error('connection is null');
+        await this.ensureSchema(fileName);
+        const result = await this.connection.query(`PRAGMA table_info('${fileName}')`);
+        return result.toArray().map((row)=>row.toJSON());
+    };
 }
 class IndexedDB {
     db = null;
@@ -165,8 +187,8 @@ class VQuery {
         this.duckDB = new DuckDB();
         this.indexedDB = new IndexedDB(dbName);
     }
-    instantiate = async ()=>{
-        await this.duckDB.instantiate();
+    init = async ()=>{
+        await this.duckDB.init();
         await this.indexedDB.open();
     };
     close = async ()=>{
@@ -208,6 +230,7 @@ class VQuery {
             }
         };
     };
+    getSchema = async (fileName)=>this.duckDB.getSchema(fileName);
 }
 exports.VQuery = __webpack_exports__.VQuery;
 for(var __webpack_i__ in __webpack_exports__)if (-1 === [

package/dist/index.d.ts

@@ -1,41 +1 @@
-export declare class VQuery {
-    private duckDB;
-    private indexedDB;
-    constructor(dbName?: string);
-    /**
-     * @description 初始化数据库
-     */
-    instantiate: () => Promise<void>;
-    /**
-     * @description 关闭数据库
-     */
-    close: () => Promise<void>;
-    /**
-     * @description 注册文件
-     * @param fileName 文件名
-     * @param source 文件内容
-     */
-    writeFile: (fileName: string, source: string | ArrayBuffer | Uint8Array | Blob) => Promise<void>;
-    /**
-     * @description 从 IndexedDB 读取文件并注册到 DuckDB
-     * @param fileName 文件名
-     */
-    readFile: (fileName: string) => Promise<void>;
-    /**
-     * @description 列出 IndexedDB 中的所有文件
-     */
-    listFiles: () => Promise<string[]>;
-    /**
-     * @description 执行 SQL 查询
-     * @param sql SQL 语句
-     */
-    query: (sql: string) => Promise<{
-        performance: {
-            startAt: string;
-            endAt: string;
-            duration: number;
-        };
-        dataset: any[];
-        table: any;
-    }>;
-}
+export { VQuery } from './vquery';

package/dist/index.js

@@ -1,11 +1,20 @@
-import { AsyncDuckDB, ConsoleLogger, getJsDelivrBundles, selectBundle } from "@duckdb/duckdb-wasm";
+import { AsyncDuckDB, ConsoleLogger, selectBundle } from "@duckdb/duckdb-wasm";
 class DuckDB {
     db = null;
     connection = null;
     constructor(){}
-    instantiate = async ()=>{
-        const JSDELIVR_BUNDLES = getJsDelivrBundles();
-        const bundle = await selectBundle(JSDELIVR_BUNDLES);
+    init = async ()=>{
+        const MANUAL_BUNDLES = {
+            mvp: {
+                mainModule: new URL('@duckdb/duckdb-wasm/dist/duckdb-mvp.wasm', import.meta.url).href,
+                mainWorker: new URL('@duckdb/duckdb-wasm/dist/duckdb-browser-mvp.worker.js', import.meta.url).toString()
+            },
+            eh: {
+                mainModule: new URL('@duckdb/duckdb-wasm/dist/duckdb-eh.wasm', import.meta.url).href,
+                mainWorker: new URL('@duckdb/duckdb-wasm/dist/duckdb-browser-eh.worker.js', import.meta.url).toString()
+            }
+        };
+        const bundle = await selectBundle(MANUAL_BUNDLES);
         const worker_url = URL.createObjectURL(new Blob([
             `importScripts("${bundle.mainWorker}");`
         ], {
@@ -52,6 +61,16 @@ class DuckDB {
             table
         };
     };
+    ensureSchema = async (fileName)=>{
+        if (!this.connection) throw new Error('connection is null');
+        await this.connection.query(`CREATE TEMP TABLE IF NOT EXISTS "${fileName}" AS SELECT * FROM read_csv_auto('${fileName}')`);
+    };
+    getSchema = async (fileName)=>{
+        if (!this.connection) throw new Error('connection is null');
+        await this.ensureSchema(fileName);
+        const result = await this.connection.query(`PRAGMA table_info('${fileName}')`);
+        return result.toArray().map((row)=>row.toJSON());
+    };
 }
 class IndexedDB {
     db = null;
@@ -137,8 +156,8 @@ class VQuery {
         this.duckDB = new DuckDB();
         this.indexedDB = new IndexedDB(dbName);
     }
-    instantiate = async ()=>{
-        await this.duckDB.instantiate();
+    init = async ()=>{
+        await this.duckDB.init();
         await this.indexedDB.open();
     };
     close = async ()=>{
@@ -180,5 +199,6 @@ class VQuery {
             }
         };
     };
+    getSchema = async (fileName)=>this.duckDB.getSchema(fileName);
 }
 export { VQuery };

package/dist/types/QueryResult.d.ts

@@ -0,0 +1 @@
+export type QueryResult = any;

package/dist/types/index.d.ts

@@ -0,0 +1 @@
+export type { QueryResult } from './QueryResult';

package/dist/vquery.d.ts

@@ -0,0 +1,48 @@
+import { QueryResult } from './types';
+export declare class VQuery {
+    private duckDB;
+    private indexedDB;
+    constructor(dbName?: string);
+    /**
+     * @description 初始化数据库
+     */
+    init: () => Promise<void>;
+    /**
+     * @description 关闭数据库
+     */
+    close: () => Promise<void>;
+    /**
+     * @description 注册文件
+     * @param fileName 文件名
+     * @param source 文件内容
+     */
+    writeFile: (fileName: string, source: string | ArrayBuffer | Uint8Array | Blob) => Promise<void>;
+    /**
+     * @description 从 IndexedDB 读取文件并注册到 DuckDB
+     * @param fileName 文件名
+     */
+    readFile: (fileName: string) => Promise<void>;
+    /**
+     * @description 列出 IndexedDB 中的所有文件
+     */
+    listFiles: () => Promise<string[]>;
+    /**
+     * @description 执行 SQL 查询
+     * @param sql SQL 语句
+     */
+    query: (sql: string) => Promise<{
+        performance: {
+            startAt: string;
+            endAt: string;
+            duration: number;
+        };
+        dataset: any[];
+        table: any;
+    }>;
+    /**
+     * @description 获取文件的 Schema
+     * @param fileName 文件名
+     * @returns 文件的 Schema
+     */
+    getSchema: (fileName: string) => Promise<QueryResult>;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@visactor/vquery",
-  "version": "0.1.33",
+  "version": "0.1.35",
   "type": "module",
   "exports": {
     ".": {