use workerInfo to get the expected value

src/handlers/route_handler.ts

@@ -264,7 +264,7 @@ export class RouteHandler {
  }
  }
 
- static addExpected(type: string, className: string, workerName: string, expectedValue: any) {
+ static addExpected(type: "body" | "query" | "param", className: string, workerName: string, expectedValue: any) {
 
  for (const prop in expectedValue) {
  const propValue = expectedValue[prop];
@@ -359,18 +359,6 @@ export class RouteHandler {
  }
  }
 
- static getExpectedQuery(controllerName: string, workerName: string) {
- return routerCollection.get(controllerName).workers.get(workerName).expectedQuery;
- }
-
- static getExpectedBody(controllerName: string, workerName: string) {
- return routerCollection.get(controllerName).workers.get(workerName).expectedBody;
- }
-
- static getExpectedParam(controllerName: string, workerName: string) {
- return routerCollection.get(controllerName).workers.get(workerName).expectedParam;
- }
-
  static addRouteToCache(url: string, route: IRouteMatch) {
  routeCache.set(url, route);
  }

src/helpers/validate_body_guard.ts

@@ -5,7 +5,7 @@ import { executeValidate } from "./execute_validate";
 export class ValidateBodyGuard extends Guard {
  async check() {
  const componentProp = this['componentProp_'];
- const expectedBody = RouteHandler.getExpectedBody(componentProp.controllerName, this.workerName);
+ const expectedBody = componentProp.workerInfo.expectedBody;
  if (expectedBody == null) return;
  const validationResult = await executeValidate(expectedBody, this.body);
  if (validationResult.error) {

src/helpers/validate_param_guard.ts

@@ -5,7 +5,7 @@ import { executeValidate } from "./execute_validate";
 export class ValidateParamGuard extends Guard {
  async check() {
  const componentProp = this['componentProp_'];
- const expectedValue = RouteHandler.getExpectedParam(componentProp.controllerName, this.workerName);
+ const expectedValue = componentProp.workerInfo.expectedParam;
  if (expectedValue == null) return;
  const validationResult = await executeValidate(expectedValue, this.param);
  if (validationResult.error) {

src/helpers/validate_query_shield.ts

@@ -1,11 +1,10 @@
 import { Shield } from "../abstracts";
-import { RouteHandler } from "../handlers";
 import { executeValidate } from "./execute_validate";
 
 export class ValidateQueryShield extends Shield {
  async protect() {
  const componentProp = this['componentProp_'];
- const expectedQuery = RouteHandler.getExpectedQuery(componentProp.controllerName, this.workerName);
+ const expectedQuery = componentProp.workerInfo.expectedQuery;
  if (expectedQuery == null) return;
  const validationResult = await executeValidate(expectedQuery, this.query);
  if (validationResult.error) {

src/interfaces/worker_info.ts

@@ -10,5 +10,6 @@ export type IWorkerInfo = {
  values: any[];
  expectedQuery?: any;
  expectedBody?: any;
+ expectedParam?: any;
  cache?: ICacheOptionStored;
 };

src/models/worker_info.ts

@@ -22,6 +22,7 @@ export class WorkerInfo implements IWorkerInfo {
  this.values = value.values;
  this.expectedQuery = value.expectedQuery;
  this.expectedBody = value.expectedBody;
+ this.expectedParam = value.expectedParam;
  this.pattern = value.pattern;
  this.cache = value.cache;
  }