pigallery2/benchmark/index.ts

import {Config} from '../src/common/config/private/Config';
import * as path from 'path';
import {ProjectPath} from '../src/backend/ProjectPath';
import {BenchmarkResult, Benchmarks} from './Benchmarks';
import {SearchTypes} from '../src/common/entities/AutoCompleteItem';
import {Utils} from '../src/common/Utils';
import {DiskMangerWorker} from '../src/backend/model/threading/DiskMangerWorker';

const config: { path: string, system: string } = require(path.join(__dirname, 'config.json'));
Config.Server.imagesFolder = config.path;
const dbPath = path.join(__dirname, 'test.db');
ProjectPath.reset();
const RUNS = 50;

let resultsText = '';
const printLine = (text: string) => {
  resultsText += text + '\n';
};

const printHeader = async () => {
  const dt = new Date();
  printLine('## PiGallery2 v' + require('./../package.json').version +
    ', ' + Utils.zeroPrefix(dt.getDate(), 2) +
    '.' + Utils.zeroPrefix(dt.getMonth() + 1, 2) +
    '.' + dt.getFullYear());
  printLine('**System**: ' + config.system);
  const dir = await DiskMangerWorker.scanDirectory('./');
  printLine('**Gallery**: directories: ' +
    dir.directories.length +
    ' media: ' + dir.media.length +
    // @ts-ignore
    ', faces: ' + dir.media.reduce((p, c) => p + (c.metadata.faces || []).length, 0));
};


const printTableHeader = () => {
  printLine('| action | action details | average time | details |');
  printLine('|:------:|:--------------:|:------------:|:-------:|');
};
const printResult = (result: BenchmarkResult, action: string, actionDetails: string = '') => {
  console.log('benchmarked: ' + action);
  let details = '-';
  if (result.items) {
    details = 'items: ' + result.items;
  }
  if (result.media) {
    details = 'media: ' + result.media + ', directories:' + result.directories;
  }
  printLine('| ' + action + ' | ' + actionDetails +
    ' | ' + (result.duration).toFixed(1) + 'ms | ' + details + ' |');
};

const run = async () => {
  const start = Date.now();
  const bm = new Benchmarks(RUNS, dbPath);

  // header
  await printHeader();
  printTableHeader();
  printResult(await bm.bmScanDirectory(), 'Scanning directory');
  printResult(await bm.bmSaveDirectory(), 'Saving directory');
  printResult(await bm.bmListDirectory(), 'Listing Directory');
  (await bm.bmAllSearch('a')).forEach(res => {
    if (res.searchType !== null) {
      printResult(res.result, 'searching', '`a` as `' + SearchTypes[res.searchType] + '`');
    } else {
      printResult(res.result, 'searching', '`a` as `any`');
    }
  });
  printResult(await bm.bmInstantSearch('a'), 'instant search', '`a`');
  printResult(await bm.bmAutocomplete('a'), 'auto complete', '`a`');
  console.log(resultsText);
  console.log('run for : ' + ((Date.now() - start)).toFixed(1) + 'ms');
};

run();