Improve configurability of performance script

rubensworks · Apr 14, 2023 · 4e7435c · 4e7435c
1 parent ac753e4
commit 4e7435c
Show file tree

Hide file tree

Showing 4 changed files with 208 additions and 159 deletions.
diff --git a/package.json b/package.json
@@ -64,7 +64,8 @@
     "pre-commit": "^1.2.2",
     "streamify-array": "^1.0.1",
     "ts-jest": "^29.0.0",
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "yargs": "^17.7.1"
   },
   "jest": {
     "transform": {

diff --git a/perf/PerformanceTest.ts b/perf/PerformanceTest.ts
@@ -13,40 +13,36 @@ import { RdfStore } from '../lib/RdfStore';
  */
 export class PerformanceTest {
   public constructor(
-    public readonly approaches: {
-      name: string;
-      options: {
-        type: 'rdfstore';
-        options: IRdfStoreOptions<any, any>;
-      } | {
-        type: 'n3';
-      };
-    }[],
+    public readonly approaches: IPerformanceTestApproach[],
     public readonly dimension = 256,
     public readonly prefix = 'http://example.org/#',
     public readonly dataFactory: RDF.DataFactory = new DataFactory(),
   ) {}
 
-  public async run(): Promise<void> {
+  public async run(scope: 'all' | 'triples' | 'quads' | 'quoted'): Promise<void> {
     for (const approach of this.approaches) {
       console.log(`\n# ${approach.name}\n`);
 
-      let store = approach.options.type === 'n3' ? new Store() : new RdfStore(approach.options.options);
-      this.addTriplesToDefaultGraph(this.dimension, store);
-      this.findTriplesNoVariables(this.dimension, store);
-      this.findTriples1Variable(this.dimension, store);
-      this.findTriples2Variables(this.dimension, store);
-      await this.findTriples1VariableStream(this.dimension, <any> store);
-      this.countTriples1Variable(this.dimension, store);
-      console.log();
+      if (scope === 'all' || scope === 'triples') {
+        const store = approach.options.type === 'n3' ? new Store() : new RdfStore(approach.options.options);
+        this.addTriplesToDefaultGraph(this.dimension, store);
+        this.findTriplesNoVariables(this.dimension, store);
+        this.findTriples1Variable(this.dimension, store);
+        this.findTriples2Variables(this.dimension, store);
+        await this.findTriples1VariableStream(this.dimension, <any>store);
+        this.countTriples1Variable(this.dimension, store);
+        console.log();
+      }
 
-      store = approach.options.type === 'n3' ? new Store() : new RdfStore(approach.options.options);
-      this.addQuadsToGraphs(this.dimension / 4, store);
-      this.findQuadsInGraphs(this.dimension / 4, store);
-      console.log();
+      if (scope === 'all' || scope === 'quads') {
+        const store = approach.options.type === 'n3' ? new Store() : new RdfStore(approach.options.options);
+        this.addQuadsToGraphs(this.dimension / 4, store);
+        this.findQuadsInGraphs(this.dimension / 4, store);
+        console.log();
+      }
 
-      if (approach.options.type !== 'n3') {
-        store = new RdfStore(approach.options.options);
+      if ((scope === 'all' || scope === 'quoted') && approach.options.type !== 'n3') {
+        const store = new RdfStore(approach.options.options);
         this.addQuotedTriplesToGraphs(this.dimension / 2, store);
         this.findQuotedTriplesInGraphs(this.dimension / 2, store);
       }
@@ -268,3 +264,13 @@ export class PerformanceTest {
   }
 }
 /* eslint-enable no-console */
+
+export interface IPerformanceTestApproach {
+  name: string;
+  options: {
+    type: 'rdfstore';
+    options: IRdfStoreOptions<any, any>;
+  } | {
+    type: 'n3';
+  };
+}