@@ -20,7 +20,7 @@ export class MiddlewareControl {
2020 * @private
2121 * A member holding map of MiddlewareOptions
2222 */
23- private middlewareOptions : Map < string , MiddlewareOptions > ;
23+ private middlewareOptions : Map < Function , MiddlewareOptions > ;
2424
2525 /**
2626 * @public
@@ -30,10 +30,10 @@ export class MiddlewareControl {
3030 * @returns The instance of MiddlewareControl
3131 */
3232 public constructor ( middlewareOptions : MiddlewareOptions [ ] = [ ] ) {
33- this . middlewareOptions = new Map < string , MiddlewareOptions > ( ) ;
33+ this . middlewareOptions = new Map < Function , MiddlewareOptions > ( ) ;
3434 for ( const option of middlewareOptions ) {
35- const name = option . constructor . name ;
36- this . middlewareOptions . set ( name , option ) ;
35+ const fn = option . constructor ;
36+ this . middlewareOptions . set ( fn , option ) ;
3737 }
3838 }
3939
@@ -43,8 +43,8 @@ export class MiddlewareControl {
4343 * @param {string } name - The class name of the strongly typed option class
4444 * @returns The middleware option
4545 */
46- public getMiddlewareOptions ( name : string ) : MiddlewareOptions {
47- return this . middlewareOptions . get ( name ) ;
46+ public getMiddlewareOptions ( fn : Function ) : MiddlewareOptions {
47+ return this . middlewareOptions . get ( fn ) ;
4848 }
4949
5050 /**
@@ -54,7 +54,7 @@ export class MiddlewareControl {
5454 * @param {MiddlewareOptions } option - The strongly typed middleware option
5555 * @returns nothing
5656 */
57- public setMiddlewareOptions ( name : string , option : MiddlewareOptions ) : void {
58- this . middlewareOptions . set ( name , option ) ;
57+ public setMiddlewareOptions ( fn : Function , option : MiddlewareOptions ) : void {
58+ this . middlewareOptions . set ( fn , option ) ;
5959 }
6060}
0 commit comments