-
Notifications
You must be signed in to change notification settings - Fork 13
/
SequelizeProvider.ts
109 lines (91 loc) · 3.22 KB
/
SequelizeProvider.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import { StorageProviderConstructor } from '../types/StorageProviderConstructor';
import { IStorageProvider } from './interface/IStorageProvider';
import { StorageProvider } from './StorageProvider';
import { Database } from './Database';
import { Util } from '../util/Util';
import * as Sequelize from 'sequelize';
/**
* Represents a Sequelize Model instance which represents a single
* db table entry from the table controlled by the `SequelizeProvider`.
*/
interface StorageEntry extends Sequelize.Model
{
key: string;
value: string;
}
/**
* Represents a Model cast type that fixes some issues with Model
* methods that have `this` types declared
*/
type SequelizeModel = (new () => Sequelize.Model) & typeof Sequelize.Model;
export enum Dialect { Postgres, SQLite, MSSQL, MySQL }
export function SequelizeProvider(url: string, dialect: Dialect, debug: boolean): StorageProviderConstructor
{
return class extends StorageProvider implements IStorageProvider
{
private readonly _backend: Database;
private readonly _model: SequelizeModel;
public constructor(name: string)
{
super();
// Lazy load sequelize
const packages: string[] = ['sequelize-typescript', 'sequelize'];
const seq: typeof Sequelize = Util.lazyLoad<typeof Sequelize>(...packages);
this._backend = Database.instance(url, debug);
this._model = class extends seq.Model {};
const dataTypes: typeof Sequelize.DataTypes =
(seq as any)['DataTypes']
? (seq as any)['DataTypes']
: (seq as any)['DataType'];
this._model.init(
{
key: { type: dataTypes.STRING, allowNull: false, primaryKey: true },
value: [Dialect.Postgres, Dialect.SQLite, Dialect.MSSQL].includes(dialect)
? dataTypes.TEXT
: dataTypes.TEXT('long')
},
{
modelName: name,
timestamps: false,
freezeTableName: true,
sequelize: this._backend.db
}
);
}
public async init(): Promise<void>
{
await this._backend.init();
await this._backend.db.sync();
}
public async keys(): Promise<string[]>
{
return (await this._model.findAll() as StorageEntry[]).map(r => r.key);
}
public async get(key: string): Promise<string | undefined>
{
if (typeof key === 'undefined') throw new TypeError('Key must be provided');
if (typeof key !== 'string') throw new TypeError('Key must be a string');
const entry: StorageEntry = await this._model.findByPk(key) as StorageEntry;
if (entry === null) return;
return entry.value;
}
public async set(key: string, value: string): Promise<void>
{
if (typeof key === 'undefined') throw new TypeError('Key must be provided');
if (typeof key !== 'string') throw new TypeError('Key must be a string');
if (typeof value === 'undefined') throw new TypeError('Value must be provided');
if (typeof value !== 'string') throw new TypeError('Value must be a string');
await this._model.upsert({ key, value });
}
public async remove(key: string): Promise<void>
{
if (typeof key === 'undefined') throw new TypeError('Key must be provided');
if (typeof key !== 'string') throw new TypeError('Key must be a string');
await this._model.destroy({ where: { key } });
}
public async clear(): Promise<void>
{
await this._model.destroy({ where: {} });
}
};
}