-
Notifications
You must be signed in to change notification settings - Fork 1
/
sql.ts
70 lines (59 loc) · 1.45 KB
/
sql.ts
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
import * as _ from "lodash";
import * as Sequelize from "sequelize";
export const toSQL = (value: any) => {
if (_.isDate(value)) {
return value;
}
if (_.isObject(value) || _.isArray(value)) {
return "__json:" + JSON.stringify(value);
}
return value;
};
export const createTable = async (
url: string,
structure: any,
data: any[][],
indexes = [],
clear = true,
name = "data"
) => {
const typeMap = {
date: Sequelize.DATE,
number: Sequelize.DECIMAL,
boolean: Sequelize.BOOLEAN,
string: Sequelize.STRING,
integer: Sequelize.INTEGER,
object: Sequelize.STRING
};
let options = {
logging: null
};
if (_.startsWith(url, "postgres://") && url.indexOf("ssl=true") !== -1) {
options["native"] = true;
}
const sequelize = new Sequelize(url, options);
const sequelizeStructure = _.mapValues(structure, type => typeMap[type]);
const headers = _.keys(sequelizeStructure);
const sequelizeOptions = {
timestamps: false,
tableName: name,
version: false,
indexes: indexes
};
const Model = sequelize.define(name, sequelizeStructure, sequelizeOptions);
if (clear) {
await Model.drop();
}
await Model.sync();
await Model.bulkCreate(
data.map(row => {
return _.zipObject(headers, row.map(toSQL));
})
);
return new Promise<{
sequelize: Sequelize.Sequelize;
model: typeof Model;
}>(resolve => {
resolve({ sequelize: sequelize, model: Model });
});
};