1 /*
2  * Copyright (C) 2018 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 
17 package com.android.settingslib.net;
18 
19 import android.app.usage.NetworkStats;
20 import android.app.usage.NetworkStatsManager;
21 import android.content.Context;
22 import android.net.NetworkTemplate;
23 import android.util.Log;
24 
25 import androidx.loader.content.AsyncTaskLoader;
26 
27 /**
28  * Loader for retrieving the network stats summary for all UIDs.
29  */
30 public class NetworkStatsSummaryLoader extends AsyncTaskLoader<NetworkStats> {
31 
32     private static final String TAG = "NetworkDetailLoader";
33     private final NetworkStatsManager mNetworkStatsManager;
34     private final long mStart;
35     private final long mEnd;
36     private final NetworkTemplate mNetworkTemplate;
37 
NetworkStatsSummaryLoader(Builder builder)38     private NetworkStatsSummaryLoader(Builder builder) {
39         super(builder.mContext);
40         mStart = builder.mStart;
41         mEnd = builder.mEnd;
42         mNetworkTemplate = builder.mNetworkTemplate;
43         mNetworkStatsManager = (NetworkStatsManager)
44                 builder.mContext.getSystemService(Context.NETWORK_STATS_SERVICE);
45     }
46 
47     @Override
onStartLoading()48     protected void onStartLoading() {
49         super.onStartLoading();
50         forceLoad();
51     }
52 
53     @Override
loadInBackground()54     public NetworkStats loadInBackground() {
55         try {
56             return mNetworkStatsManager.querySummary(mNetworkTemplate, mStart, mEnd);
57         } catch (RuntimeException e) {
58             Log.e(TAG, "Exception querying network detail.", e);
59             return null;
60         }
61     }
62 
63     @Override
onStopLoading()64     protected void onStopLoading() {
65         super.onStopLoading();
66         cancelLoad();
67     }
68 
69     @Override
onReset()70     protected void onReset() {
71         super.onReset();
72         cancelLoad();
73     }
74 
75     public static class Builder {
76         private final Context mContext;
77         private long mStart;
78         private long mEnd;
79         private NetworkTemplate mNetworkTemplate;
80 
Builder(Context context)81         public Builder(Context context) {
82             mContext = context;
83         }
84 
setStartTime(long start)85         public Builder setStartTime(long start) {
86             mStart = start;
87             return this;
88         }
89 
setEndTime(long end)90         public Builder setEndTime(long end) {
91             mEnd = end;
92             return this;
93         }
94 
95         /**
96          * Set {@link NetworkTemplate} for builder
97          */
setNetworkTemplate(NetworkTemplate template)98         public Builder setNetworkTemplate(NetworkTemplate template) {
99             mNetworkTemplate = template;
100             return this;
101         }
102 
build()103         public NetworkStatsSummaryLoader build() {
104             return new NetworkStatsSummaryLoader(this);
105         }
106     }
107 }
108