@@ -89,6 +89,7 @@ export class KuzuDriver {
8989
9090 async installUDF ( ) : Promise < void > {
9191 var res = await this . conn ?. execute ( 'CREATE MACRO elementid(x) AS CAST(ID(x),"STRING")' ) ;
92+
9293 console . log ( res . toString ( ) ) ;
9394 }
9495
@@ -351,6 +352,12 @@ export class KuzuDriver {
351352 }
352353
353354 async query ( queryScript : string ) : Promise < any > {
355+ if ( queryScript === 'CALL kuzu.meta.schema' ) {
356+ return await this . querySchema ( ) ;
357+ }
358+ if ( queryScript === 'CALL kuzu.meta.statistics' ) {
359+ return await this . getCount ( ) ;
360+ }
354361 console . time ( 'Query cost' ) ;
355362 const queryResult = await this . conn ?. execute ( queryScript ) ;
356363 console . timeEnd ( 'Query cost' ) ;
@@ -483,7 +490,6 @@ export class KuzuDriver {
483490 const edge_count = parseInt ( edge_res . toString ( ) . split ( '\n' ) [ 1 ] , 10 ) ;
484491 const countArray = [ vertex_count , edge_count ] ;
485492
486- console . log ( countArray ) ;
487493 return countArray ;
488494 }
489495
@@ -511,7 +517,7 @@ export class KuzuDriver {
511517 console . log ( `Open dataset ${ this . curDataset } ` ) ;
512518
513519 //@ts -ignore
514- this . db = await this . kuzuEngine . Database ( this . curDataset , 0 , 10 , false , false , 4194304 * 16 * 4 ) ;
520+ this . db = await this . kuzuEngine . Database ( this . curDataset , 0 , 10 , false , false , 4194304 * 16 * 4 * 4 ) ;
515521 //@ts -ignore
516522 this . conn = await this . kuzuEngine . Connection ( this . db ) ;
517523
0 commit comments