87 lines
2.0 KiB
JavaScript
87 lines
2.0 KiB
JavaScript
|
var Benchmark = require('benchmark');
|
||
|
|
var cp = require('duplex-child-process');
|
||
|
|
var pg = require('pg')
|
||
|
|
|
||
|
|
var copy = require('../').from
|
||
|
|
|
||
|
|
var client = function() {
|
||
|
|
var client = new pg.Client()
|
||
|
|
client.connect()
|
||
|
|
return client
|
||
|
|
}
|
||
|
|
|
||
|
|
var psql = '/opt/postgresql-9.6.1/bin/psql'
|
||
|
|
var limit = 999999;
|
||
|
|
var inStream = function() {
|
||
|
|
return cp.spawn('seq', ['0', ''+limit]);
|
||
|
|
}
|
||
|
|
var suite = new Benchmark.Suite;
|
||
|
|
suite
|
||
|
|
.add({
|
||
|
|
name: 'unix pipe into psql COPY',
|
||
|
|
defer: true,
|
||
|
|
fn: function(d) {
|
||
|
|
var c = client();
|
||
|
|
c.query('DROP TABLE IF EXISTS plugnumber', function() {
|
||
|
|
c.query('CREATE TABLE plugnumber (num int)', function() {
|
||
|
|
c.end();
|
||
|
|
var from = cp.spawn('sh', ['-c', 'seq 0 '+limit+' | '+psql+' postgres -c \'COPY plugnumber FROM STDIN\''])
|
||
|
|
from.on('close', function() {
|
||
|
|
d.resolve();
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.add({
|
||
|
|
name: 'pipe into psql COPY',
|
||
|
|
defer: true,
|
||
|
|
fn: function(d) {
|
||
|
|
var c = client();
|
||
|
|
c.query('DROP TABLE IF EXISTS plugnumber', function() {
|
||
|
|
c.query('CREATE TABLE plugnumber (num int)', function() {
|
||
|
|
c.end();
|
||
|
|
var seq = inStream();
|
||
|
|
var from = cp.spawn(psql, ['postgres', '-c', 'COPY plugnumber FROM STDIN'])
|
||
|
|
seq.pipe(from);
|
||
|
|
from.on('close', function() {
|
||
|
|
d.resolve();
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
})
|
||
|
|
.add({
|
||
|
|
name: 'pipe into pg-copy-stream COPY',
|
||
|
|
defer: true,
|
||
|
|
fn: function(d) {
|
||
|
|
var c = client();
|
||
|
|
c.query('DROP TABLE IF EXISTS plugnumber', function() {
|
||
|
|
c.query('CREATE TABLE plugnumber (num int)', function() {
|
||
|
|
var seq = inStream()
|
||
|
|
var from = c.query(copy('COPY plugnumber FROM STDIN'))
|
||
|
|
seq.pipe(from);
|
||
|
|
from.on('end', function() {
|
||
|
|
c.end();
|
||
|
|
d.resolve();
|
||
|
|
})
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
.on('cycle', function(event) {
|
||
|
|
console.log(String(event.target));
|
||
|
|
})
|
||
|
|
.on('complete', function() {
|
||
|
|
console.log('Fastest is ' + this.filter('fastest').map('name'));
|
||
|
|
});
|
||
|
|
|
||
|
|
|
||
|
|
var c = client()
|
||
|
|
c.query('DROP TABLE IF EXISTS plugnumber', function() {
|
||
|
|
c.end();
|
||
|
|
suite.run();
|
||
|
|
})
|
||
|
|
|