testBigBookOut.js

const utils = require('./utils/utils');
const HrStopwatch = require('./utils/hr-stopwatch');
const ColumnSum = require('./utils/column-sum');

const Excel = require('../excel');

const {Workbook} = Excel;
const {WorkbookWriter} = Excel.stream.xlsx;

if (process.argv[2] === 'help') {
  console.log('Usage:');
  console.log('    node testBigBookOut filename count writer strings styles');
  console.log('Where:');
  console.log('    writer is one of [stream, document]');
  console.log('    strings is one of [shared, own]');
  console.log('    styles is one of [styled, plain]');
  // eslint-disable-next-line no-process-exit
  process.exit(0);
}

const filename = process.argv[2];
const count = process.argv.length > 3 ? parseInt(process.argv[3], 10) : 1000;
const writer = process.argv.length > 4 ? process.argv[4] : 'stream';
const strings = process.argv.length > 5 ? process.argv[5] : 'own';
const styles = process.argv.length > 6 ? process.argv[6] : 'styled';

const useStream = writer === 'stream';

const options = {
  writer: useStream ? 'stream' : 'document',
  filename,
  useSharedStrings: strings === 'shared',
  useStyles: styles === 'styled',
};
console.log(JSON.stringify(options, null, '  '));

const stopwatch = new HrStopwatch();
stopwatch.start();

const wb = useStream ? new WorkbookWriter(options) : new Workbook();
const ws = wb.addWorksheet('blort');

const fonts = {
  arialBlackUI14: {
    name: 'Arial Black',
    family: 2,
    size: 14,
    underline: true,
    italic: true,
  },
  comicSansUdB16: {
    name: 'Comic Sans MS',
    family: 4,
    size: 8,
    underline: 'double',
    bold: true,
  },
};

ws.columns = [
  {header: 'Col 1', key: 'key', width: 25},
  {header: 'Col 2', key: 'name', width: 32},
  {header: 'Col 3', key: 'age', width: 21},
  {header: 'Col 4', key: 'addr1', width: 18},
  {header: 'Col 5', key: 'addr2', width: 8},
  {header: 'Col 6', key: 'num1', width: 8},
  {header: 'Col 7', key: 'num2', width: 8},
  {
    header: 'Col 8',
    key: 'num3',
    width: 32,
    style: {font: fonts.comicSansUdB16},
  },
  {header: 'Col 9', key: 'date', width: 12},
  {header: 'Col 10', key: 'num4', width: 12},
];

const colCount = new ColumnSum([3, 6, 7, 8, 10]);

ws.getRow(1).font = fonts.arialBlackUI14;

let t1 = 0;
let t2 = 0;
const sw = new HrStopwatch();
const today = new Date().getTime();
let iCount = 0;

function addRow() {
  sw.start();
  const row = ws.addRow({
    key: iCount,
    name: utils.randomName(5),
    age: utils.randomNum(100),
    addr1: utils.randomName(16),
    addr2: utils.randomName(10),
    num1: utils.randomNum(10000),
    num2: utils.randomNum(100000),
    num3: utils.randomNum(1000000),
    date: new Date(today + (iCount * 86400000)),
    num4: utils.randomNum(1000),
  });
  const lap = sw.span;
  colCount.add(row);
  row.commit();
  const end = sw.span;

  t1 += lap;
  t2 += end - lap;
}

function scheduleRow(callback) {
  if (iCount++ < count) {
    if (iCount % 100000 === 0) {
      console.log(iCount);
    }
    setImmediate(() => {
      addRow();
      scheduleRow(callback);
    });
  } else {
    setImmediate(callback);
  }
}

function allDone() {
  console.log(`addRow avg ${(t1 * 1000000) / count}\xB5s`);
  console.log(`commit avg ${(t2 * 1000000) / count}\xB5s`);
  sw.start();
  (useStream ? wb.commit() : wb.xlsx.writeFile(filename, options))
    .then(() => {
      console.log(`Commit/writeFile: ${sw}`);
      stopwatch.stop();
      console.log('Done.');
      console.log(`Sums: ${colCount}`);
      console.log(`Time: ${stopwatch}`);
    })
    .catch(error => {
      console.log(error.message);
    });
}

scheduleRow(allDone);