tedious/benchmarks/bulk-load/stream.js

A TDS driver, for connecting to MS SQLServer databases.

const { createBenchmark, createConnection } = require('../common');

const { Request, TYPES } = require('../../src/tedious');
const { Readable } = require('readable-stream');

const bench = createBenchmark(main, {
  n: [10, 100],
  size: [
    10,
    100,
    1000,
    10000
  ]
});

function main({ n, size }) {
  createConnection(function(connection) {
    const request = new Request(`
      CREATE TABLE "#tmpTestTable" (
        "id" int NOT NULL
      )
    `, (err) => {
      if (err) {
        throw err;
      }

      let i = 0;

      bench.start();

      (function cb() {
        const bulkLoad = connection.newBulkLoad('#tmpTestTable', (err) => {
          if (err) {
            throw err;
          }

          if (i++ === n) {
            bench.end(n);

            connection.close();

            return;
          }

          cb();
        });

        bulkLoad.addColumn('id', TYPES.Int, { nullable: false });

        Readable.from((function*() {
          for (let j = 0; j < size; j++) {
            yield [j];
          }
        })()).pipe(bulkLoad.getRowStream());

        connection.execBulkLoad(bulkLoad);
      })();
    });

    connection.execSqlBatch(request);
  });
}