1. 程式人生 > >RxJava(五) onErrorResumeNext操作符實現app與伺服器間token機制

RxJava(五) onErrorResumeNext操作符實現app與伺服器間token機制

RxJava系列文章目錄導讀:

一、需求場景:

在開發App的時候, 很多公司的提api介面, 請求的的時候都需要帶有token, 該token在使用者第一次啟動app或者登陸的時候去獲取. 以後的所有請求都需要帶該Token 如果token過期, 伺服器將返回401, 這時候就需要去請求獲取token的介面, 如果獲取成功接著在請求原來的介面. 這個時候就兩個回撥的嵌套了. 實現起來比較費勁, 而且也不夠優雅. 程式碼的可維護性變得很差. 可以使用 onErrorResumeNext 來處理這樣的業務邏輯.

例如:請求一個使用者資訊介面,如果token沒有過期,返回使用者資訊,如果token過期,伺服器返回401,客戶端發一個獲取新token的請求,成功後,再去請求使用者資訊介面。

二、如何使用onErrorResumeNext解決

使用Retrofit來訪問伺服器


    private static RestAdapter restAdapter = new RestAdapter
            .Builder()
            .setLogLevel(RestAdapter.LogLevel.FULL)
            .setEndpoint(BASE_URL)
            .setErrorHandler(new NetWorkErrorHandler())
            .setRequestInterceptor(requestInterceptor)
            .build();


    public
static <S> S createService(Class<S> serviceClazz) { return restAdapter.create(serviceClazz); }

如果伺服器返回401,我們要去請求新的token,下面來判斷錯誤型別:

NetWorkErrorHandler

private static class NetWorkErrorHandler implements ErrorHandler {
    @Override
    public Throwable handleError(RetrofitError error) {
        retrofit.client.Response r = error.getResponse();
        if
(r != null && r.getStatus() == 401) { Log.e("ErrorHandler", "---------> access deny code=401"); // User Custom Exception return new AccessDenyException(error.getMessage()); } return error.getCause(); } }

UserApi

public interface UserApi {

    @GET("/token")
    AuthToken refreshToken();
}

伺服器端程式碼邏輯

伺服器端使用Java web+Tomcat來實現的. 如果需要可以把伺服器部署在你的本地機器上, github地址
伺服器端的基本邏輯:客戶端請伺服器api,伺服器判斷客戶端帶過來的token,如果過期則返回401,提示沒有許可權訪問;如果是請求token介面,則返回token,有效期為10s。

客戶端App的實現

以一個請求使用者資訊介面為例

Observable<Response> observable = userApi.getUserInfo();
        observable.onErrorResumeNext(refreshTokenAndRetry(observable))//also use retryWhen to implement it
                .subscribeOn(Schedulers.io())
                .observeOn(AndroidSchedulers.mainThread())
                .subscribe(new Observer<Response>() {
                    @Override
                    public void onCompleted() {
                        loading = false;
                        appendText(tvLogs, "task completed-----");
                        //hideLoadingDialog();
                    }

                    @Override
                    public void onError(Throwable t) {
                        //hideLoadingDialog();
                        t.printStackTrace();
                        loading = false;
                        appendText(tvLogs, t.getClass().getName() + "\n" + t.getMessage());
                        NetErrorType.ErrorType error = NetErrorType.getErrorType(t);
                        appendText(tvLogs, error.msg);
                    }

                    public void onNext(Response response) {
                        String content = new String(((TypedByteArray) response.getBody()).getBytes());
                        appendText(tvLogs, "receiver data: " + content);
                    }
                });

核心程式碼


    private <T> Func1<Throwable, ? extends Observable<? extends T>> refreshTokenAndRetry(final Observable<T> toBeResumed) {
        return new Func1<Throwable, Observable<? extends T>>() {
            @Override
            public Observable<? extends T> call(Throwable throwable) {
                throwable.printStackTrace();
                // Here check if the error thrown really is a 401
                if (isHttp401Error(throwable)) {
                    return createTokenObvervable().flatMap(new Func1<AuthToken, Observable<? extends T>>() {
                        @Override
                        public Observable<? extends T> call(AuthToken token) {
                            appendText(tvLogs, "refresh token success,token's validity is 10s\nResume last request");
                            return toBeResumed;
                        }
                    });
                }
                // re-throw this error because it's not recoverable from here
                return Observable.error(throwable);
            }

            public boolean isHttp401Error(Throwable throwable) {
                return throwable instanceof AccessDenyException;
            }

        };
    }

請求token api 的Observable


    public Observable<AuthToken> createTokenObvervable() {
        return Observable.create(new Observable.OnSubscribe<AuthToken>() {
            @Override
            public void call(Subscriber<? super AuthToken> observer) {
                try {
                    if (!observer.isUnsubscribed()) {
                        appendText(tvLogs, "God!!! Token is out of date. \nstart refresh token......");
                        observer.onNext(userApi.refreshToken());
                        observer.onCompleted();
                    }
                } catch (Exception e) {
                    observer.onError(e);
                }
            }
        }).subscribeOn(Schedulers.io());
    }

執行效果:

這裡寫圖片描述