|
2 | 2 |
|
3 | 3 | var fs = require('fs'); |
4 | 4 | var Sequelize = require('sequelize-cockroachdb'); |
5 | | -var sequelize = new Sequelize(process.env.ADDR, { |
6 | | - dialectOptions: {cockroachdbTelemetryDisabled : true} |
7 | | -}); |
| 5 | + |
| 6 | +if (process.env.ADDR === undefined) { |
| 7 | + throw new Error("ADDR (database URL) must be specified."); |
| 8 | +} |
| 9 | + |
| 10 | +var url = new URL(process.env.ADDR); |
| 11 | +var opts = { |
| 12 | + dialect: "postgres", |
| 13 | + username: url.username, |
| 14 | + host: url.hostname, |
| 15 | + port: url.port, |
| 16 | + database: url.pathname.substring(1), // ignore leading '/' |
| 17 | + dialectOptions: { |
| 18 | + cockroachdbTelemetryDisabled: true, |
| 19 | + ssl: {}, |
| 20 | + }, |
| 21 | + logging: false, |
| 22 | +}; |
| 23 | + |
| 24 | +if (url.password) { |
| 25 | + opts.password = url.password |
| 26 | +} |
| 27 | +if (url.searchParams.has("options")) { |
| 28 | + var pgOpts = url.searchParams.get("options") |
| 29 | + var cluster = pgOpts.match(/cluster=([^\s]+)/)[1] |
| 30 | + opts.database = `${cluster}.${opts.database}` |
| 31 | +} |
| 32 | +if (url.searchParams.get("sslmode") === "disable") { |
| 33 | + delete opts.dialectOptions.ssl |
| 34 | +} else { |
| 35 | + if (url.searchParams.has("sslrootcert")) { |
| 36 | + opts.dialectOptions.ssl.ca = fs.readFileSync(url.searchParams.get("sslrootcert").toString()) |
| 37 | + } |
| 38 | + if (url.searchParams.has("sslcert")) { |
| 39 | + opts.dialectOptions.ssl.cert = fs.readFileSync(url.searchParams.get("sslcert").toString()) |
| 40 | + } |
| 41 | + if (url.searchParams.has("sslkey")) { |
| 42 | + opts.dialectOptions.ssl.key = fs.readFileSync(url.searchParams.get("sslkey").toString()) |
| 43 | + } |
| 44 | +} |
| 45 | +var sequelize = new Sequelize(opts); |
8 | 46 | var DataTypes = Sequelize.DataTypes; |
9 | 47 |
|
10 | 48 | if (!Sequelize.supportsCockroachDB) { |
11 | 49 | throw new Error("CockroachDB dialect for Sequelize not installed"); |
12 | 50 | } |
13 | 51 |
|
14 | | -if (process.env.ADDR === undefined) { |
15 | | - throw new Error("ADDR (database URL) must be specified."); |
16 | | -} |
17 | | - |
18 | 52 | module.exports.Customer = sequelize.define('customer', { |
19 | 53 | name: DataTypes.STRING |
20 | 54 | }, { |
|
0 commit comments