com.github.airk.tool.sobitmap.NetworkHunter.java Source code

Introduction

Here is the source code for com.github.airk.tool.sobitmap.NetworkHunter.java

Source

/*
 * Copyright 2015 Kevin Liu
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.github.airk.tool.sobitmap;

import android.net.Uri;
import android.util.Log;

import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.Response;

import java.io.File;
import java.io.IOException;
import java.net.SocketException;
import java.util.concurrent.TimeUnit;

/**
 * Created by kevin on 15/3/24.
 * <p/>
 * Handle network request
 */
final class NetworkHunter extends Hunter {
    private OkHttpClient client;

    NetworkHunter() {
        super();
        client = new OkHttpClient();
        client.setConnectTimeout(15000, TimeUnit.MILLISECONDS);
        client.setReadTimeout(20000, TimeUnit.MILLISECONDS);
        client.setWriteTimeout(20000, TimeUnit.MILLISECONDS);
    }

    @Override
    boolean canHandle(Uri source) {
        String scheme = source.getScheme();
        return scheme.equals("http") || scheme.equals("https");
    }

    @Override
    File preCacheFile() {
        File file = null;
        try {
            Request netReq = new Request.Builder().url(request.source.toString()).build();
            Response response = client.newCall(netReq).execute();
            if (response.code() >= 300) {
                if (SoBitmap.LOG) {
                    Log.d(SoBitmap.TAG, tag() + ": Network error occurred..." + response.code());
                }
                response.body().close();
                request.e = new HuntException(HuntException.REASON_NETWORK_ERROR);
                request.e.setExtra(response.code() + "");
                return null;
            }
            if (SoBitmap.LOG) {
                Log.d(SoBitmap.TAG, tag() + ": Downloading...");
            }
            String size = response.header("content-length");
            if (size != null && !request.options.onlyLevel && Long.parseLong(size) > request.options.maxInput) {
                request.e = new HuntException(HuntException.REASON_TOO_LARGE);
                return null;
            }
            file = new File(request.cacheDir, request.tag);
            Util.inputStreamToFile(file, response.body().byteStream());
            if (SoBitmap.LOG) {
                Log.d(SoBitmap.TAG, tag() + ": Downloaded to file -> " + file.getAbsolutePath());
            }
            if (!file.exists()) { //UNLIKELY
                request.e = new HuntException(HuntException.REASON_FILE_NOT_FOUND);
                request.e.setExtra("Download success but file not found.");
                return null;
            } else {
                if (!request.options.onlyLevel && file.length() / 1024 > request.options.maxInput) {
                    request.e = new HuntException(HuntException.REASON_TOO_LARGE);
                    return null;
                }
            }
        } catch (SocketException e) {
            e.printStackTrace();
            request.e = new HuntException(HuntException.REASON_NETWORK_ERROR);
        } catch (IOException ignore) {
            request.e = new HuntException(HuntException.REASON_IO_EXCEPTION);
        }
        return file;
    }

    @Override
    void cleanup(File cacheFile) {
        if (SoBitmap.LOG) {
            Log.d(SoBitmap.TAG, tag() + ": Cache file cleanup.");
        }
        cacheFile.delete();
    }

    @Override
    String tag() {
        return "NetworkHunter";
    }

}