-
-
Notifications
You must be signed in to change notification settings - Fork 55
/
Copy pathproxy-filter.registry.ts
42 lines (36 loc) · 1.32 KB
/
proxy-filter.registry.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
import { Injectable } from '@nestjs/common';
import { Filter } from './interfaces/filter.interface';
import { RouteFilter } from './interfaces/route.interface';
@Injectable()
export class ProxyFilterRegistry {
private readonly filters = new Map<string, Filter>();
private readonly routeFilters = new Map<string, RouteFilter[]>();
public addFilter(name: string, filter: Filter) {
this.filters.set(name, filter);
}
public addRouteFilters(id: string, routeFilters: RouteFilter[]) {
this.routeFilters.set(id, routeFilters);
}
public addRouteFilter(id: string, routeFilter: RouteFilter) {
const routeFilters = this.routeFilters.get(id) || [];
routeFilters.push(routeFilter);
this.routeFilters.set(id, routeFilters);
}
public getFilter(name: string): Filter | undefined {
return this.filters.get(name);
}
public getRouteFilters(id: string): ([RouteFilter, Filter])[] {
const routeFilters = this.routeFilters.get(id);
if (!routeFilters) {
return [];
}
const result = [];
routeFilters.forEach(routeFilter => {
const filter = this.filters.get(routeFilter.name);
if (filter) {
result.push([routeFilter, filter]);
}
});
return result;
}
}