333

schangxiang@126.com

2025-09-19 18966e02fb573c7e2bb0c6426ed792b38b910940

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
"use strict";
 
var fs = require("fs");
var async = require("async");
var Benchmark = require("benchmark");
var Connection = require("../lib/tedious").Connection;
 
function createConnection(cb) {
  var config = JSON.parse(fs.readFileSync(require('os').homedir() + '/.tedious/test-connection.json', 'utf8')).config;
 
  var connection = new Connection(config);
  connection.on("connect", function() {
    cb(connection);
  });
}
 
function createBenchmark(test) {
  if (process.argv.indexOf("--profile") != -1) {
    process.nextTick(function() {
      runProfile(test);
    });
  } else {
    process.nextTick(function() {
      runBenchmark(test);
    });
  }
}
 
function runBenchmark(test) {
  var memStart, memMax = memStart = process.memoryUsage().rss;
 
  test.setup(function(err) {
    if (err) throw err;
 
    var bm = new Benchmark(test.name, {
      defer: true,
      fn: function(deferred) {
        test.exec(function(err) {
          if (err) throw err;
 
          memMax = Math.max(memMax, process.memoryUsage().rss);
 
          deferred.resolve();
        });
      }
    });
 
    bm.on("complete", function(event) {
      console.log(String(event.target))
      console.log("Memory:", (memMax - memStart)/1024/1024, "MiB")
 
      test.teardown(function(err) {
        if (err) throw err;
      });
    });
 
    bm.run({ "async": true });
  });
}
 
function runProfile(test) {
  test.setup(function(err) {
    if (err) throw err;
 
    async.timesSeries(test.profileIterations, function(n, done) {
      async.setImmediate(function() {
        console.log("[Iteration " +  n + "]");
        test.exec(done);
      });
    }, function(err) {
      if (err) throw err;
 
      test.teardown(function(err) {
        if (err) throw err;
      });
    });
  });
}
 
module.exports.createBenchmark = createBenchmark;
module.exports.createConnection = createConnection;