2017-08-15 19 views
1

我想刷新一个JWT令牌,当我得到一个特定的异常,当它是另一个异常,我的ErrorHandler应该处理它们。角2 http异常处理程序和jwt刷新

我有一段代码,一个是令牌刷新工作的地方,另一个是代码处理异常处理程序的代码段,但我不能以工作方式组合它们。

问题是我不能抛出异常,并用我的ErrorHandler在observable中捕获异常。

这里是我可以刷新令牌的代码。当它失败时,它会检查错误代码是否为token_expired,何时它会刷新令牌并重试请求。

export class HttpErrorService extends Http { 

    constructor(backend: XHRBackend, defaultOptions: RequestOptions) { 
    super(backend, defaultOptions); 
    } 

    request(url: string | Request, options?: RequestOptionsArgs, disableRefresh = false): Observable<Response> { 
     return super.request(url, options).catch((error: Response) => { 
     // Refresh token on token_expired exception. 
     if (!disableRefresh && error.status === 401 && error.json().error.code === 'token_expired') { 
      return this.renewToken().flatMap((response) => { 
      const res = response.json(); 
      // Replace the token in storage. 
      localStorage.setItem('__token', res.data.token); 

      // Replace request the token with the new one. 
      if (url instanceof Request) { 
       url.headers.set('Authorization', 'Bearer ' + res.data.token); 
      } else if (options) { 
       options.headers.set('Authorization', 'Bearer ' + res.data.token); 
      } 

      // To prevent a loop disable refreshing at the next request. 
      return this.request(url, options, true); 
      }); 
     } 

     // Here I want to throw the exception. 
     // I need to be able to catch it with my exception handler. 
     // throw error; doesn't work. 
     return Observable.throw(error); 
     }); 
    } 

    private getBaseUrl(): string { 
    return environment.base_uri; 
    }; 

    renewToken(): Observable<Response> { 
    const headers = new Headers(); 
    headers.append('Authorization', 'Bearer ' + localStorage.getItem('__token')) 

    return this.post(this.getBaseUrl() + '/auth/refresh', {}, {headers: headers}); 
    } 
} 

上述唯一的坏处是我无法在异常处理程序中捕捉异常。

以下代码可能会抛出ErrorHandler可以捕获的异常。但我不知道我怎么能刷新令牌一个电话...

export class HttpErrorService extends Http { 

    constructor(backend: XHRBackend, defaultOptions: RequestOptions) { 
    super(backend, defaultOptions); 
    } 

    request(url: string | Request, options?: RequestOptionsArgs, disableRefresh = false): Observable<Response> { 
    return Observable.create(observer => { 
     super.request(url, options).subscribe(
     res => observer.next(res), 
     err => { 
      if (!disableRefresh && err.status === 401 && err.json().error.code === 'token_expired') { 
      // I can't return this.renewToken()... 
      } 
      observer.error(err); 
      throw new HttpException(err); // this is getting catched by the ErrorHandler 
     }, 
     () => observer.complete); 
    }); 
    } 

    private getBaseUrl(): string { 
    return environment.base_uri; 
    }; 

    renewToken(): Observable<Response> { 
    const headers = new Headers(); 
    headers.append('Authorization', 'Bearer ' + localStorage.getItem('__token')) 

    return this.post(this.getBaseUrl() + '/auth/refresh', {}, {headers: headers}); 
    } 
} 

我的错误处理程序是只包含一个console.log()
https://angular.io/api/core/ErrorHandler

我该如何得到这个工作?

+0

也可以添加实际调用请求函数和ErrorHandler的代码吗? – trungk18

+0

@ trungk18我重写了Http类,所以每个http请求都使用请求函数。它是默认的Http库。 ErrorHandler只是一个'console.log();' –

+0

您好,您可以尝试“抛出Observable.throw(错误)”而不是“返回Observable.throw(错误)”在您的第一块代码? – trungk18

回答

0

几个小时后,我终于找到了解决方案!

export class HttpErrorService extends Http { 

    constructor(backend: XHRBackend, defaultOptions: RequestOptions) { 
    super(backend, defaultOptions); 
    } 

    request(url: string | Request, options?: RequestOptionsArgs, disableRefresh = false): Observable<Response> { 
    return Observable.create(observer => { 
     super.request(url, options).retryWhen(attempts => this.retryRequest(attempts)).catch((error: Response) => { 
     // Refresh token on token_expired exception. 
     if (!disableRefresh && error.status === 401 && error.json().error.code === 'token_expired') { 
      return this.renewToken().flatMap((response) => { 
      const res = response.json(); 
      // Replace the token in storage. 
      localStorage.setItem('__token', res.data.token); 

      // Replace request the token with the new one. 
      if (url instanceof Request) { 
       url.headers.set('Authorization', 'Bearer ' + res.data.token); 
      } else if (options) { 
       options.headers.set('Authorization', 'Bearer ' + res.data.token); 
      } 

      // To prevent a loop disable refreshing at the next request. 
      return this.request(url, options, true); 
      }); 
     } 

     throw Observable.throw(error); 
     }).subscribe(
     res => observer.next(res), 
     err => { 
      observer.error(err); 
      throw new HttpException(err); 
     } 
    ); 
    }); 
    } 

    private getBaseUrl(): string { 
    return environment.base_uri; 
    }; 

    renewToken(): Observable<Response> { 
    const headers = new Headers(); 
    headers.append('Authorization', 'Bearer ' + localStorage.getItem('__token')) 

    return this.post(this.getBaseUrl() + '/auth/refresh', {}, {headers: headers}); 
    } 

    retryRequest(attempts: any) { 
    let count = 0; 

    return attempts.flatMap(error => { 
     return ++count >= 3 ? Observable.throw(error) : Observable.timer(count * 1000); 
    }); 
    } 

}