I need to perform a request, if my token is expired I need to refresh it and retry the request.
This is how I'm trying to do it, at the moment I can refresh the token but it throws me a NetworkOnMainThreadException. It finishes the request,update the token and reaches logs, but that exception its killing me. How can I avoid that?
public Observable<Estabelecimento> listarEstabelecimentos() {
return Observable.defer(this::getListarEstabelecimentoObservable)
.retryWhen(throwableObservable -> throwableObservable.flatMap(
throwable -> {
if (throwable instanceof UnauthorizedException) {
return mRequestManager.getTokenObservable(AutoAtendimentoApplication.getContext())
.doOnNext(response -> /* log stuff */)
.flatMap((Function<AuthResponse, ObservableSource<?>>) response2 ->
getListarEstabelecimentoObservable()
.doOnNext(estabelecimento ->
/* log stuff */)
)
);
}
return Observable.error(throwable);
}));
}
NetWorkErrorHandler:
public <T> T processError(Response<T> response ) {
switch (response.code()) {
case 401:
throw new UnauthorizedException();
default:
return response.body();
}
}
Token:
private Observable<AuthResponse> getToken(Context context,
@GrantType.GrantTypeDef String grantType, @Nullable String refreshToken) {
SessionManager sessionManager = SessionManager.getInstance(context);
Usuario usuario = sessionManager.getUser();
AuthRequest request = new AuthRequest(usuario.getUsername(),
usuario.getPassword(), grantType, refreshToken);
return mAuthAPIService.getToken(request)
.observeOn(AndroidSchedulers.mainThread())
.subscribeOn(Schedulers.io())
.map(response -> storeTokens(response, context));
}
Edit 1: I'm pretty sure the problem happens inside the flatmap and put subscribeOn does not solve the problem
Edit 2: Code update