com.github.scribejava.httpclient.ahc.OAuthAsyncCompletionHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scribejava-httpclient-ahc Show documentation

There is a newer version: 8.3.3

Show newest version

package com.github.scribejava.httpclient.ahc;

import com.github.scribejava.core.model.OAuthAsyncRequestCallback;
import com.github.scribejava.core.model.OAuthRequest;
import com.github.scribejava.core.model.Response;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import org.asynchttpclient.AsyncCompletionHandler;

public class OAuthAsyncCompletionHandler extends AsyncCompletionHandler {

    private final OAuthAsyncRequestCallback callback;
    private final OAuthRequest.ResponseConverter converter;

    public OAuthAsyncCompletionHandler(OAuthAsyncRequestCallback callback,
            OAuthRequest.ResponseConverter converter) {
        this.callback = callback;
        this.converter = converter;
    }

    @Override
    public T onCompleted(org.asynchttpclient.Response ahcResponse) {
        try {
            final Map headersMap = new HashMap<>();
            for (Map.Entry header : ahcResponse.getHeaders()) {
                headersMap.put(header.getKey(), header.getValue());
            }

            final Response response = new Response(ahcResponse.getStatusCode(), ahcResponse.getStatusText(), headersMap,
                    ahcResponse.getResponseBodyAsStream());

            @SuppressWarnings("unchecked")
            final T t = converter == null ? (T) response : converter.convert(response);
            if (callback != null) {
                callback.onCompleted(t);
            }
            return t;
        } catch (IOException | RuntimeException e) {
            onThrowable(e);
            return null;
        }
    }

    @Override
    public void onThrowable(Throwable t) {
        if (callback != null) {
            callback.onThrowable(t);
        }
    }
};