/
create-test-tables.js
90 lines (78 loc) · 2.04 KB
/
create-test-tables.js
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
82
83
84
85
86
87
88
89
90
'use strict'
var fs = require('fs')
var args = require('../test/cli')
var pg = require('../lib')
var people = [
{ name: 'Aaron', age: 10 },
{ name: 'Brian', age: 20 },
{ name: 'Chris', age: 30 },
{ name: 'David', age: 40 },
{ name: 'Elvis', age: 50 },
{ name: 'Frank', age: 60 },
{ name: 'Grace', age: 70 },
{ name: 'Haley', age: 80 },
{ name: 'Irma', age: 90 },
{ name: 'Jenny', age: 100 },
{ name: 'Kevin', age: 110 },
{ name: 'Larry', age: 120 },
{ name: 'Michelle', age: 130 },
{ name: 'Nancy', age: 140 },
{ name: 'Olivia', age: 150 },
{ name: 'Peter', age: 160 },
{ name: 'Quinn', age: 170 },
{ name: 'Ronda', age: 180 },
{ name: 'Shelley', age: 190 },
{ name: 'Tobias', age: 200 },
{ name: 'Uma', age: 210 },
{ name: 'Veena', age: 220 },
{ name: 'Wanda', age: 230 },
{ name: 'Xavier', age: 240 },
{ name: 'Yoyo', age: 250 },
{ name: 'Zanzabar', age: 260 },
]
var config = {
host: args.host,
port: args.port,
user: args.user,
password: args.password,
database: args.database,
}
if (process.env.PGSSLMODE) {
config.ssl = {}
if (process.env.PGSSLROOTCERT) {
config.ssl.ca = fs.readFileSync(process.env.PGSSLROOTCERT)
}
if (process.env.PGSSLCERT) {
config.ssl.cert = fs.readFileSync(process.env.PGSSLCERT)
}
if (process.env.PGSSLKEY) {
config.ssl.key = fs.readFileSync(process.env.PGSSLKEY)
}
}
var con = new pg.Client(config)
con.connect((err) => {
if (err) {
throw err
}
con.query(
'DROP TABLE IF EXISTS person;' + ' CREATE TABLE person (id serial, name varchar(10), age integer)',
(err) => {
if (err) {
throw err
}
console.log('Created table person')
console.log('Filling it with people')
con.query(
'INSERT INTO person (name, age) VALUES' +
people.map((person) => ` ('${person.name}', ${person.age})`).join(','),
(err, result) => {
if (err) {
throw err
}
console.log(`Inserted ${result.rowCount} people`)
con.end()
}
)
}
)
})