Skip to content

Commit

Permalink
refactor(server): telemetry ports env
Browse files Browse the repository at this point in the history
  • Loading branch information
jrasm91 committed Oct 17, 2024
1 parent 0eabb3e commit 78e9b2f
Show file tree
Hide file tree
Showing 6 changed files with 80 additions and 18 deletions.
15 changes: 12 additions & 3 deletions server/src/interfaces/config.interface.ts
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,11 @@ export interface EnvData {
thirdPartySupportUrl?: string;
};

bull: {
config: QueueOptions;
queues: RegisterQueueOptions[];
};

database: {
url?: string;
host: string;
Expand All @@ -45,6 +50,10 @@ export interface EnvData {
server: string;
};

network: {
trustedProxies: string[];
};

resourcePaths: {
lockFile: string;
geodata: {
Expand All @@ -62,9 +71,9 @@ export interface EnvData {

redis: RedisOptions;

bull: {
config: QueueOptions;
queues: RegisterQueueOptions[];
telemetry: {
apiPort: number;
microservicesPort: number;
};

storage: {
Expand Down
40 changes: 40 additions & 0 deletions server/src/repositories/config.repository.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,9 @@ const resetEnv = () => {
for (const env of [
'IMMICH_WORKERS_INCLUDE',
'IMMICH_WORKERS_EXCLUDE',
'IMMICH_TRUSTED_PROXIES',
'IMMICH_API_METRICS_PORT',
'IMMICH_MICROSERVICES_METRICS_PORT',

'DB_URL',
'DB_HOSTNAME',
Expand Down Expand Up @@ -178,4 +181,41 @@ describe('getEnv', () => {
expect(getEnv).toThrowError('Invalid worker(s) found: api,microservices,randomservice');
});
});

describe('network', () => {
it('should return default network options', () => {
const { network } = getEnv();
expect(network).toEqual({
trustedProxies: [],
});
});

it('should parse trusted proxies', () => {
process.env.IMMICH_TRUSTED_PROXIES = '10.1.0.0,10.2.0.0, 169.254.0.0/16';
const { network } = getEnv();
expect(network).toEqual({
trustedProxies: ['10.1.0.0', '10.2.0.0', '169.254.0.0/16'],
});
});
});

describe('telemetry', () => {
it('should return default ports', () => {
const { telemetry } = getEnv();
expect(telemetry).toEqual({
apiPort: 8081,
microservicesPort: 8082,
});
});

it('should parse custom ports', () => {
process.env.IMMICH_API_METRICS_PORT = '2001';
process.env.IMMICH_MICROSERVICES_METRICS_PORT = '2002';
const { telemetry } = getEnv();
expect(telemetry).toEqual({
apiPort: 2001,
microservicesPort: 2002,
});
});
});
});
12 changes: 12 additions & 0 deletions server/src/repositories/config.repository.ts
Original file line number Diff line number Diff line change
Expand Up @@ -117,6 +117,13 @@ const getEnv = (): EnvData => {

licensePublicKey: isProd ? productionKeys : stagingKeys,

network: {
trustedProxies: (process.env.IMMICH_TRUSTED_PROXIES ?? '')
.split(',')
.map((value) => value.trim())
.filter(Boolean),
},

redis: redisConfig,

resourcePaths: {
Expand All @@ -138,6 +145,11 @@ const getEnv = (): EnvData => {
ignoreMountCheckErrors: process.env.IMMICH_IGNORE_MOUNT_CHECK_ERRORS === 'true',
},

telemetry: {
apiPort: Number(process.env.IMMICH_API_METRICS_PORT || '') || 8081,
microservicesPort: Number(process.env.IMMICH_MICROSERVICES_METRICS_PORT || '') || 8082,
},

workers,

noColor: !!process.env.NO_COLOR,
Expand Down
16 changes: 4 additions & 12 deletions server/src/workers/api.ts
Original file line number Diff line number Diff line change
Expand Up @@ -10,25 +10,17 @@ import { ImmichEnvironment } from 'src/enum';
import { IConfigRepository } from 'src/interfaces/config.interface';
import { ILoggerRepository } from 'src/interfaces/logger.interface';
import { WebSocketAdapter } from 'src/middleware/websocket.adapter';
import { ConfigRepository } from 'src/repositories/config.repository';
import { ApiService } from 'src/services/api.service';
import { isStartUpError } from 'src/services/storage.service';
import { otelStart } from 'src/utils/instrumentation';
import { useSwagger } from 'src/utils/misc';

function parseTrustedProxy(input?: string) {
if (!input) {
return [];
}
// Split on ',' char to allow multiple IPs
return input.split(',');
}

async function bootstrap() {
process.title = 'immich-api';
const otelPort = Number.parseInt(process.env.IMMICH_API_METRICS_PORT ?? '8081');
const trustedProxies = parseTrustedProxy(process.env.IMMICH_TRUSTED_PROXIES ?? '');

otelStart(otelPort);
const { telemetry, network } = new ConfigRepository().getEnv();
otelStart(telemetry.apiPort);

const app = await NestFactory.create<NestExpressApplication>(ApiModule, { bufferLogs: true });
const logger = await app.resolve<ILoggerRepository>(ILoggerRepository);
Expand All @@ -39,7 +31,7 @@ async function bootstrap() {

logger.setContext('Bootstrap');
app.useLogger(logger);
app.set('trust proxy', ['loopback', 'linklocal', 'uniquelocal', ...trustedProxies]);
app.set('trust proxy', ['loopback', 'linklocal', 'uniquelocal', ...network.trustedProxies]);
app.set('etag', 'strong');
app.use(cookieParser());
app.use(json({ limit: '10mb' }));
Expand Down
6 changes: 3 additions & 3 deletions server/src/workers/microservices.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,13 @@ import { serverVersion } from 'src/constants';
import { IConfigRepository } from 'src/interfaces/config.interface';
import { ILoggerRepository } from 'src/interfaces/logger.interface';
import { WebSocketAdapter } from 'src/middleware/websocket.adapter';
import { ConfigRepository } from 'src/repositories/config.repository';
import { isStartUpError } from 'src/services/storage.service';
import { otelStart } from 'src/utils/instrumentation';

export async function bootstrap() {
const otelPort = Number.parseInt(process.env.IMMICH_MICROSERVICES_METRICS_PORT ?? '8082');

otelStart(otelPort);
const { telemetry } = new ConfigRepository().getEnv();
otelStart(telemetry.microservicesPort);

const app = await NestFactory.create(MicroservicesModule, { bufferLogs: true });
const logger = await app.resolve(ILoggerRepository);
Expand Down
9 changes: 9 additions & 0 deletions server/test/repositories/config.repository.mock.ts
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,10 @@ const envData: EnvData = {
server: 'server-public-key',
},

network: {
trustedProxies: [],
},

redis: {
host: 'redis',
port: 6379,
Expand All @@ -56,6 +60,11 @@ const envData: EnvData = {
ignoreMountCheckErrors: false,
},

telemetry: {
apiPort: 8081,
microservicesPort: 8082,
},

workers: [ImmichWorker.API, ImmichWorker.MICROSERVICES],

noColor: false,
Expand Down

0 comments on commit 78e9b2f

Please sign in to comment.