1/*
2 * Copyright (C) 2011 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.settingslib.net;
18
19import static android.net.NetworkStats.SET_DEFAULT;
20import static android.net.NetworkStats.SET_FOREGROUND;
21import static android.net.NetworkStats.TAG_NONE;
22import static android.net.NetworkStatsHistory.FIELD_RX_BYTES;
23import static android.net.NetworkStatsHistory.FIELD_TX_BYTES;
24import static android.text.format.DateUtils.HOUR_IN_MILLIS;
25
26import android.content.AsyncTaskLoader;
27import android.content.Context;
28import android.net.INetworkStatsSession;
29import android.net.NetworkStatsHistory;
30import android.net.NetworkTemplate;
31import android.os.Bundle;
32import android.os.RemoteException;
33
34import com.android.settingslib.AppItem;
35
36/**
37 * Loader for historical chart data for both network and UID details.
38 */
39public class ChartDataLoader extends AsyncTaskLoader<ChartData> {
40    private static final String KEY_TEMPLATE = "template";
41    private static final String KEY_APP = "app";
42    private static final String KEY_FIELDS = "fields";
43
44    private final INetworkStatsSession mSession;
45    private final Bundle mArgs;
46
47    public static Bundle buildArgs(NetworkTemplate template, AppItem app) {
48        return buildArgs(template, app, FIELD_RX_BYTES | FIELD_TX_BYTES);
49    }
50
51    public static Bundle buildArgs(NetworkTemplate template, AppItem app, int fields) {
52        final Bundle args = new Bundle();
53        args.putParcelable(KEY_TEMPLATE, template);
54        args.putParcelable(KEY_APP, app);
55        args.putInt(KEY_FIELDS, fields);
56        return args;
57    }
58
59    public ChartDataLoader(Context context, INetworkStatsSession session, Bundle args) {
60        super(context);
61        mSession = session;
62        mArgs = args;
63    }
64
65    @Override
66    protected void onStartLoading() {
67        super.onStartLoading();
68        forceLoad();
69    }
70
71    @Override
72    public ChartData loadInBackground() {
73        final NetworkTemplate template = mArgs.getParcelable(KEY_TEMPLATE);
74        final AppItem app = mArgs.getParcelable(KEY_APP);
75        final int fields = mArgs.getInt(KEY_FIELDS);
76
77        try {
78            return loadInBackground(template, app, fields);
79        } catch (RemoteException e) {
80            // since we can't do much without history, and we don't want to
81            // leave with half-baked UI, we bail hard.
82            throw new RuntimeException("problem reading network stats", e);
83        }
84    }
85
86    private ChartData loadInBackground(NetworkTemplate template, AppItem app, int fields)
87            throws RemoteException {
88        final ChartData data = new ChartData();
89        data.network = mSession.getHistoryForNetwork(template, fields);
90
91        if (app != null) {
92            // load stats for current uid and template
93            final int size = app.uids.size();
94            for (int i = 0; i < size; i++) {
95                final int uid = app.uids.keyAt(i);
96                data.detailDefault = collectHistoryForUid(
97                        template, uid, SET_DEFAULT, data.detailDefault);
98                data.detailForeground = collectHistoryForUid(
99                        template, uid, SET_FOREGROUND, data.detailForeground);
100            }
101
102            if (size > 0) {
103                data.detail = new NetworkStatsHistory(data.detailForeground.getBucketDuration());
104                data.detail.recordEntireHistory(data.detailDefault);
105                data.detail.recordEntireHistory(data.detailForeground);
106            } else {
107                data.detailDefault = new NetworkStatsHistory(HOUR_IN_MILLIS);
108                data.detailForeground = new NetworkStatsHistory(HOUR_IN_MILLIS);
109                data.detail = new NetworkStatsHistory(HOUR_IN_MILLIS);
110            }
111        }
112
113        return data;
114    }
115
116    @Override
117    protected void onStopLoading() {
118        super.onStopLoading();
119        cancelLoad();
120    }
121
122    @Override
123    protected void onReset() {
124        super.onReset();
125        cancelLoad();
126    }
127
128    /**
129     * Collect {@link NetworkStatsHistory} for the requested UID, combining with
130     * an existing {@link NetworkStatsHistory} if provided.
131     */
132    private NetworkStatsHistory collectHistoryForUid(
133            NetworkTemplate template, int uid, int set, NetworkStatsHistory existing)
134            throws RemoteException {
135        final NetworkStatsHistory history = mSession.getHistoryForUid(
136                template, uid, set, TAG_NONE, FIELD_RX_BYTES | FIELD_TX_BYTES);
137
138        if (existing != null) {
139            existing.recordEntireHistory(history);
140            return existing;
141        } else {
142            return history;
143        }
144    }
145}
146