mirror of
https://github.com/Unleash/unleash.git
synced 2024-12-22 19:07:54 +01:00
bb58a516bd
https://linear.app/unleash/issue/2-1237/explore-slack-app-addon-scalability-and-limitations Relevant document: https://linear.app/unleash/document/894e12b7-802c-4bc5-8c22-75af0e66fa4b - Implements 30s cache layer for Slack channels; - Adds error logging; - Adds respective tests; - Slight refactors and improvements for overall robustness; --------- Co-authored-by: Gastón Fournier <gaston@getunleash.io>
216 lines
6.2 KiB
TypeScript
216 lines
6.2 KiB
TypeScript
import stoppable, { StoppableServer } from 'stoppable';
|
|
import { promisify } from 'util';
|
|
import version from './util/version';
|
|
import { migrateDb } from '../migrator';
|
|
import getApp from './app';
|
|
import { createMetricsMonitor } from './metrics';
|
|
import { createStores } from './db';
|
|
import { createServices, scheduleServices } from './services';
|
|
import { createConfig } from './create-config';
|
|
import registerGracefulShutdown from './util/graceful-shutdown';
|
|
import { createDb } from './db/db-pool';
|
|
import sessionDb from './middleware/session-db';
|
|
// Types
|
|
import {
|
|
IAuthType,
|
|
IUnleash,
|
|
IUnleashConfig,
|
|
IUnleashOptions,
|
|
IUnleashServices,
|
|
RoleName,
|
|
} from './types';
|
|
|
|
import User, { IUser } from './types/user';
|
|
import ApiUser from './types/api-user';
|
|
import { Logger, LogLevel } from './logger';
|
|
import AuthenticationRequired from './types/authentication-required';
|
|
import Controller from './routes/controller';
|
|
import { IAuthRequest } from './routes/unleash-types';
|
|
import { SimpleAuthSettings } from './types/settings/simple-auth-settings';
|
|
import { Knex } from 'knex';
|
|
import * as permissions from './types/permissions';
|
|
import * as eventType from './types/events';
|
|
import { Db } from './db/db';
|
|
import { defaultLockKey, defaultTimeout, withDbLock } from './util/db-lock';
|
|
|
|
async function createApp(
|
|
config: IUnleashConfig,
|
|
startApp: boolean,
|
|
): Promise<IUnleash> {
|
|
// Database dependencies (stateful)
|
|
const logger = config.getLogger('server-impl.js');
|
|
const serverVersion = config.enterpriseVersion ?? version;
|
|
const db = createDb(config);
|
|
const stores = createStores(config, db);
|
|
const services = createServices(stores, config, db);
|
|
await scheduleServices(services);
|
|
|
|
const metricsMonitor = createMetricsMonitor();
|
|
const unleashSession = sessionDb(config, db);
|
|
|
|
const stopUnleash = async (server?: StoppableServer) => {
|
|
logger.info('Shutting down Unleash...');
|
|
if (server) {
|
|
const stopServer = promisify(server.stop);
|
|
await stopServer();
|
|
}
|
|
services.schedulerService.stop();
|
|
metricsMonitor.stopMonitoring();
|
|
stores.clientInstanceStore.destroy();
|
|
services.clientMetricsServiceV2.destroy();
|
|
services.proxyService.destroy();
|
|
services.addonService.destroy();
|
|
await db.destroy();
|
|
};
|
|
|
|
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 = await getApp(config, stores, services, unleashSession, db);
|
|
|
|
await metricsMonitor.startMonitoring(
|
|
config,
|
|
stores,
|
|
serverVersion,
|
|
config.eventBus,
|
|
services.instanceStatsService,
|
|
db,
|
|
);
|
|
const unleash: Omit<IUnleash, 'stop'> = {
|
|
stores,
|
|
eventBus: config.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,
|
|
});
|
|
}
|
|
|
|
if (
|
|
config.environmentEnableOverrides &&
|
|
config.environmentEnableOverrides?.length > 0
|
|
) {
|
|
await services.environmentService.overrideEnabledProjects(
|
|
config.environmentEnableOverrides,
|
|
);
|
|
}
|
|
|
|
return new Promise((resolve, reject) => {
|
|
if (startApp) {
|
|
const server = stoppable(
|
|
app.listen(config.listen, () =>
|
|
logger.info('Unleash has started.', server.address()),
|
|
),
|
|
config.server.gracefulShutdownTimeout,
|
|
);
|
|
|
|
server.keepAliveTimeout = config.server.keepAliveTimeout;
|
|
server.headersTimeout = config.server.headersTimeout;
|
|
server.on('listening', () => {
|
|
resolve({
|
|
...unleash,
|
|
server,
|
|
stop: () => stopUnleash(server),
|
|
});
|
|
});
|
|
server.on('error', reject);
|
|
} else {
|
|
resolve({ ...unleash, stop: stopUnleash });
|
|
}
|
|
});
|
|
}
|
|
|
|
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 migration: disabled');
|
|
} else {
|
|
logger.debug('DB migration: start');
|
|
if (opts.flagResolver?.isEnabled('migrationLock')) {
|
|
logger.info('Running migration with lock');
|
|
const lock = withDbLock(config.db, {
|
|
lockKey: defaultLockKey,
|
|
timeout: defaultTimeout,
|
|
logger,
|
|
});
|
|
await lock(migrateDb)(config);
|
|
} else {
|
|
await migrateDb(config);
|
|
}
|
|
|
|
logger.debug('DB migration: end');
|
|
}
|
|
} catch (err) {
|
|
logger.error('Failed to migrate db', err);
|
|
throw err;
|
|
}
|
|
|
|
const unleash = await createApp(config, true);
|
|
if (config.server.gracefulShutdownEnable) {
|
|
registerGracefulShutdown(unleash, logger);
|
|
}
|
|
return unleash;
|
|
}
|
|
|
|
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 migrateDb(config);
|
|
}
|
|
} catch (err) {
|
|
logger.error('Failed to migrate db', err);
|
|
throw err;
|
|
}
|
|
return createApp(config, false);
|
|
}
|
|
|
|
export default {
|
|
start,
|
|
create,
|
|
};
|
|
|
|
export {
|
|
start,
|
|
create,
|
|
Controller,
|
|
AuthenticationRequired,
|
|
User,
|
|
ApiUser,
|
|
LogLevel,
|
|
RoleName,
|
|
IAuthType,
|
|
Knex,
|
|
Db,
|
|
permissions,
|
|
eventType,
|
|
};
|
|
|
|
export type {
|
|
Logger,
|
|
IUnleash,
|
|
IUnleashOptions,
|
|
IUnleashConfig,
|
|
IUser,
|
|
IUnleashServices,
|
|
IAuthRequest,
|
|
SimpleAuthSettings,
|
|
};
|