91 lines
3.4 KiB
Java
91 lines
3.4 KiB
Java
/*
|
|
* Twidere - Twitter client for Android
|
|
*
|
|
* Copyright (C) 2012-2015 Mariotaku Lee <mariotaku.lee@gmail.com>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package org.mariotaku.twidere.util.net;
|
|
|
|
import android.content.ContentResolver;
|
|
import android.content.ContentValues;
|
|
import android.content.Context;
|
|
|
|
import com.squareup.okhttp.Interceptor;
|
|
import com.squareup.okhttp.OkHttpClient;
|
|
import com.squareup.okhttp.Request;
|
|
import com.squareup.okhttp.RequestBody;
|
|
import com.squareup.okhttp.Response;
|
|
import com.squareup.okhttp.ResponseBody;
|
|
|
|
import org.mariotaku.twidere.model.RequestType;
|
|
import org.mariotaku.twidere.provider.TwidereDataStore.NetworkUsages;
|
|
import org.mariotaku.twidere.util.Utils;
|
|
|
|
import java.io.IOException;
|
|
|
|
/**
|
|
* Created by mariotaku on 15/6/24.
|
|
*/
|
|
public class NetworkUsageUtils {
|
|
public static void initForHttpClient(Context context, OkHttpClient client) {
|
|
client.networkInterceptors().add(new NetworkUsageInterceptor(context));
|
|
}
|
|
|
|
private static int sNetworkType;
|
|
|
|
public static void setNetworkType(int networkType) {
|
|
NetworkUsageUtils.sNetworkType = networkType;
|
|
}
|
|
|
|
private static class NetworkUsageInterceptor implements Interceptor {
|
|
private final Context context;
|
|
|
|
public NetworkUsageInterceptor(Context context) {
|
|
this.context = context;
|
|
setNetworkType(Utils.getActiveNetworkType(context));
|
|
}
|
|
|
|
@Override
|
|
public Response intercept(Chain chain) throws IOException {
|
|
final Request request = chain.request();
|
|
final Object tag = request.tag();
|
|
if (!(tag instanceof RequestType)) return chain.proceed(request);
|
|
final ContentValues values = new ContentValues();
|
|
values.put(NetworkUsages.TIME_IN_HOURS, System.currentTimeMillis() / 1000 / 60 / 60);
|
|
values.put(NetworkUsages.KILOBYTES_SENT, getBodyLength(request.body()) / 1024.0);
|
|
values.put(NetworkUsages.REQUEST_TYPE, ((RequestType) tag).getName());
|
|
values.put(NetworkUsages.REQUEST_NETWORK, sNetworkType);
|
|
final Response response = chain.proceed(request);
|
|
values.put(NetworkUsages.KILOBYTES_RECEIVED, getBodyLength(response.body()) / 1024.0);
|
|
final ContentResolver cr = context.getContentResolver();
|
|
cr.insert(NetworkUsages.CONTENT_URI, values);
|
|
return response;
|
|
}
|
|
|
|
private long getBodyLength(RequestBody body) throws IOException {
|
|
if (body == null) return 0;
|
|
final long length = body.contentLength();
|
|
return length > 0 ? length : 0;
|
|
}
|
|
|
|
private long getBodyLength(ResponseBody body) throws IOException {
|
|
if (body == null) return 0;
|
|
final long length = body.contentLength();
|
|
return length > 0 ? length : 0;
|
|
}
|
|
}
|
|
}
|