2021-04-22 10:07:10 +02:00
|
|
|
'use strict';
|
|
|
|
|
|
|
|
import EventEmitter from 'events';
|
|
|
|
import { IUnleash } from './types/core';
|
|
|
|
import { IUnleashConfig, IUnleashOptions } from './types/option';
|
|
|
|
import version from './util/version';
|
|
|
|
import migrator from '../migrator';
|
|
|
|
import getApp from './app';
|
|
|
|
import { createMetricsMonitor } from './metrics';
|
|
|
|
import { createStores } from './db';
|
|
|
|
import { createServices } from './services';
|
2021-04-22 15:04:08 +02:00
|
|
|
import { createConfig } from './create-config';
|
2021-04-22 10:07:10 +02:00
|
|
|
import User from './user';
|
|
|
|
|
|
|
|
import permissions from './permissions';
|
|
|
|
import AuthenticationRequired from './authentication-required';
|
|
|
|
import eventType from './event-type';
|
|
|
|
import { addEventHook } from './event-hook';
|
|
|
|
|
|
|
|
async function closeServer(opts): Promise<void> {
|
|
|
|
const { server, metricsMonitor } = opts;
|
|
|
|
|
|
|
|
metricsMonitor.stopMonitoring();
|
|
|
|
|
|
|
|
return new Promise((resolve, reject) => {
|
|
|
|
server.close(err => (err ? reject(err) : resolve()));
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
async function destroyDatabase(stores): Promise<void> {
|
|
|
|
const { db, clientInstanceStore, clientMetricsStore } = stores;
|
|
|
|
|
|
|
|
return new Promise((resolve, reject) => {
|
|
|
|
clientInstanceStore.destroy();
|
|
|
|
clientMetricsStore.destroy();
|
|
|
|
|
|
|
|
db.destroy(error => (error ? reject(error) : resolve()));
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
async function createApp(
|
|
|
|
config: IUnleashConfig,
|
|
|
|
startApp: boolean,
|
|
|
|
): Promise<IUnleash> {
|
|
|
|
// Database dependencies (stateful)
|
|
|
|
const logger = config.getLogger('server-impl.js');
|
|
|
|
const serverVersion = version;
|
|
|
|
const eventBus = new EventEmitter();
|
|
|
|
const stores = createStores(config, eventBus);
|
|
|
|
const services = createServices(stores, config);
|
|
|
|
|
|
|
|
if (!config.server.secret) {
|
|
|
|
const secret = await stores.settingStore.get('unleash.secret');
|
|
|
|
// eslint-disable-next-line no-param-reassign
|
|
|
|
config.server.secret = secret;
|
|
|
|
}
|
|
|
|
const app = getApp(config, stores, services, eventBus);
|
|
|
|
const metricsMonitor = createMetricsMonitor();
|
|
|
|
if (typeof config.eventHook === 'function') {
|
|
|
|
addEventHook(config.eventHook, stores.eventStore);
|
|
|
|
}
|
|
|
|
metricsMonitor.startMonitoring(config, stores, serverVersion, eventBus);
|
|
|
|
const unleash: Omit<IUnleash, 'stop'> = {
|
|
|
|
stores,
|
|
|
|
eventBus,
|
|
|
|
services,
|
|
|
|
app,
|
|
|
|
config,
|
|
|
|
version: serverVersion,
|
|
|
|
};
|
|
|
|
|
|
|
|
if (config.import.file) {
|
|
|
|
await services.stateService.importFile({
|
|
|
|
file: config.import.file,
|
|
|
|
dropBeforeImport: config.import.dropBeforeImport,
|
|
|
|
userName: 'import',
|
|
|
|
keepExisting: config.import.keepExisting,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
return new Promise((resolve, reject) => {
|
|
|
|
if (startApp) {
|
|
|
|
const server = app.listen(config.listen, () =>
|
|
|
|
logger.info('Unleash has started.', server.address()),
|
|
|
|
);
|
|
|
|
const stop = () => {
|
|
|
|
logger.info('Shutting down Unleash...');
|
|
|
|
|
|
|
|
return closeServer({ server, metricsMonitor }).then(() =>
|
|
|
|
destroyDatabase(stores),
|
|
|
|
);
|
|
|
|
};
|
|
|
|
|
|
|
|
server.keepAliveTimeout = config.server.keepAliveTimeout;
|
|
|
|
server.headersTimeout = config.server.headersTimeout;
|
|
|
|
server.on('listening', () => {
|
|
|
|
resolve({ ...unleash, server, stop });
|
|
|
|
});
|
|
|
|
server.on('error', reject);
|
|
|
|
} else {
|
|
|
|
const stop = () => {
|
|
|
|
logger.info('Shutting down Unleash...');
|
|
|
|
metricsMonitor.stopMonitoring();
|
|
|
|
return destroyDatabase(stores);
|
|
|
|
};
|
|
|
|
|
|
|
|
resolve({ ...unleash, stop });
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
|
|
|
async function start(opts: IUnleashOptions = {}): Promise<IUnleash> {
|
|
|
|
const config = createConfig(opts);
|
|
|
|
const logger = config.getLogger('server-impl.js');
|
|
|
|
|
|
|
|
try {
|
|
|
|
if (config.db.disableMigration) {
|
|
|
|
logger.info('DB migrations disabled');
|
|
|
|
} else {
|
|
|
|
await migrator(config);
|
|
|
|
}
|
|
|
|
} catch (err) {
|
|
|
|
logger.error('Failed to migrate db', err);
|
|
|
|
throw err;
|
|
|
|
}
|
|
|
|
|
|
|
|
return createApp(config, true);
|
|
|
|
}
|
|
|
|
|
|
|
|
// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
|
|
|
|
async function create(opts: IUnleashOptions): Promise<IUnleash> {
|
|
|
|
const config = createConfig(opts);
|
|
|
|
const logger = config.getLogger('server-impl.js');
|
|
|
|
|
|
|
|
try {
|
|
|
|
if (config.db.disableMigration) {
|
|
|
|
logger.info('DB migrations disabled');
|
|
|
|
} else {
|
|
|
|
await migrator(config);
|
|
|
|
}
|
|
|
|
} catch (err) {
|
|
|
|
logger.error('Failed to migrate db', err);
|
|
|
|
throw err;
|
|
|
|
}
|
|
|
|
return createApp(config, false);
|
|
|
|
}
|
|
|
|
const serverImpl = {
|
|
|
|
start,
|
|
|
|
create,
|
|
|
|
User,
|
|
|
|
AuthenticationRequired,
|
|
|
|
permissions,
|
|
|
|
eventType,
|
|
|
|
};
|
|
|
|
export default serverImpl;
|
|
|
|
module.exports = serverImpl;
|