Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2013 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.launcher3; |
| 18 | |
| 19 | import android.app.ActivityManager; |
| 20 | import android.app.Service; |
| 21 | import android.content.Context; |
| 22 | import android.content.Intent; |
| 23 | import android.net.Uri; |
| 24 | import android.os.*; |
| 25 | import android.util.Log; |
| 26 | import android.util.LongSparseArray; |
| 27 | |
| 28 | import java.util.ArrayList; |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 29 | import java.util.List; |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 30 | |
| 31 | public class MemoryTracker extends Service { |
| 32 | public static final String TAG = MemoryTracker.class.getSimpleName(); |
| 33 | public static final String ACTION_START_TRACKING = "com.android.launcher3.action.START_TRACKING"; |
| 34 | |
| 35 | private static final long UPDATE_RATE = 5000; |
| 36 | |
| 37 | private static final int MSG_START = 1; |
| 38 | private static final int MSG_STOP = 2; |
| 39 | private static final int MSG_UPDATE = 3; |
| 40 | |
| 41 | public static class ProcessMemInfo { |
| 42 | public int pid; |
| 43 | public String name; |
Daniel Sandler | 8540bb8 | 2013-06-26 01:39:02 -0400 | [diff] [blame] | 44 | public long startTime; |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 45 | public long currentPss, currentUss; |
| 46 | public long[] pss = new long[256]; |
| 47 | public long[] uss = new long[256]; |
| 48 | //= new Meminfo[(int) (30 * 60 / (UPDATE_RATE / 1000))]; // 30 minutes |
| 49 | public long max = 1; |
| 50 | public int head = 0; |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 51 | public ProcessMemInfo(int pid, String name, long start) { |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 52 | this.pid = pid; |
| 53 | this.name = name; |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 54 | this.startTime = start; |
Daniel Sandler | 8540bb8 | 2013-06-26 01:39:02 -0400 | [diff] [blame] | 55 | } |
| 56 | public long getUptime() { |
| 57 | return System.currentTimeMillis() - startTime; |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 58 | } |
| 59 | }; |
| 60 | public final LongSparseArray<ProcessMemInfo> mData = new LongSparseArray<ProcessMemInfo>(); |
| 61 | public final ArrayList<Long> mPids = new ArrayList<Long>(); |
| 62 | private int[] mPidsArray = new int[0]; |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 63 | private final Object mLock = new Object(); |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 64 | |
| 65 | Handler mHandler = new Handler() { |
| 66 | @Override |
| 67 | public void handleMessage(Message m) { |
| 68 | switch (m.what) { |
| 69 | case MSG_START: |
| 70 | mHandler.removeMessages(MSG_UPDATE); |
| 71 | mHandler.sendEmptyMessage(MSG_UPDATE); |
| 72 | break; |
| 73 | case MSG_STOP: |
| 74 | mHandler.removeMessages(MSG_UPDATE); |
| 75 | break; |
| 76 | case MSG_UPDATE: |
| 77 | update(); |
| 78 | mHandler.removeMessages(MSG_UPDATE); |
| 79 | mHandler.sendEmptyMessageDelayed(MSG_UPDATE, UPDATE_RATE); |
| 80 | break; |
| 81 | } |
| 82 | } |
| 83 | }; |
| 84 | |
| 85 | ActivityManager mAm; |
| 86 | |
Daniel Sandler | 8540bb8 | 2013-06-26 01:39:02 -0400 | [diff] [blame] | 87 | public static void startTrackingMe(Context context, String name) { |
| 88 | context.startService(new Intent(context, MemoryTracker.class) |
| 89 | .setAction(MemoryTracker.ACTION_START_TRACKING) |
| 90 | .putExtra("pid", android.os.Process.myPid()) |
| 91 | .putExtra("name", name) |
| 92 | ); |
| 93 | } |
| 94 | |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 95 | public ProcessMemInfo getMemInfo(int pid) { |
| 96 | return mData.get(pid); |
| 97 | } |
| 98 | |
| 99 | public int[] getTrackedProcesses() { |
| 100 | return mPidsArray; |
| 101 | } |
| 102 | |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 103 | public void startTrackingProcess(int pid, String name, long start) { |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 104 | synchronized (mLock) { |
| 105 | final Long lpid = new Long(pid); |
Daniel Sandler | 8540bb8 | 2013-06-26 01:39:02 -0400 | [diff] [blame] | 106 | |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 107 | if (mPids.contains(lpid)) return; |
Daniel Sandler | 8540bb8 | 2013-06-26 01:39:02 -0400 | [diff] [blame] | 108 | |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 109 | mPids.add(lpid); |
| 110 | final int N = mPids.size(); |
| 111 | mPidsArray = new int[N]; |
| 112 | StringBuffer sb = new StringBuffer("Now tracking processes: "); |
| 113 | for (int i=0; i<N; i++) { |
| 114 | final int p = mPids.get(i).intValue(); |
| 115 | mPidsArray[i] = p; |
| 116 | sb.append(p); sb.append(" "); |
| 117 | } |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 118 | mData.put(pid, new ProcessMemInfo(pid, name, start)); |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 119 | Log.v(TAG, sb.toString()); |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 120 | } |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 121 | } |
| 122 | |
| 123 | void update() { |
Daniel Sandler | 2566b57 | 2013-07-02 14:17:31 -0500 | [diff] [blame] | 124 | synchronized (mLock) { |
| 125 | Debug.MemoryInfo[] dinfos = mAm.getProcessMemoryInfo(mPidsArray); |
| 126 | for (int i=0; i<dinfos.length; i++) { |
| 127 | Debug.MemoryInfo dinfo = dinfos[i]; |
| 128 | final long pid = mPids.get(i).intValue(); |
| 129 | final ProcessMemInfo info = mData.get(pid); |
| 130 | info.head = (info.head+1) % info.pss.length; |
| 131 | info.pss[info.head] = info.currentPss = dinfo.getTotalPss(); |
| 132 | info.uss[info.head] = info.currentUss = dinfo.getTotalPrivateDirty(); |
| 133 | if (info.currentPss > info.max) info.max = info.currentPss; |
| 134 | if (info.currentUss > info.max) info.max = info.currentUss; |
| 135 | // Log.v(TAG, "update: pid " + pid + " pss=" + info.currentPss + " uss=" + info.currentUss); |
| 136 | if (info.currentPss == 0) { |
| 137 | Log.v(TAG, "update: pid " + pid + " has pss=0, it probably died"); |
| 138 | mData.remove(pid); |
| 139 | } |
| 140 | } |
| 141 | for (int i=mPids.size()-1; i>=0; i--) { |
| 142 | final long pid = mPids.get(i).intValue(); |
| 143 | if (mData.get(pid) == null) { |
| 144 | mPids.remove(i); |
| 145 | } |
Daniel Sandler | 0becf1f | 2013-06-27 22:39:26 -0400 | [diff] [blame] | 146 | } |
| 147 | } |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | @Override |
| 151 | public void onCreate() { |
| 152 | mAm = (ActivityManager) getSystemService(Context.ACTIVITY_SERVICE); |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 153 | |
| 154 | // catch up in case we crashed but other processes are still running |
| 155 | List<ActivityManager.RunningServiceInfo> svcs = mAm.getRunningServices(256); |
| 156 | for (ActivityManager.RunningServiceInfo svc : svcs) { |
| 157 | if (svc.service.getPackageName().equals(getPackageName())) { |
| 158 | startTrackingProcess(svc.pid, svc.process, System.currentTimeMillis() - (SystemClock.elapsedRealtime() - svc.activeSince)); |
| 159 | } |
| 160 | } |
| 161 | |
| 162 | List<ActivityManager.RunningAppProcessInfo> procs = mAm.getRunningAppProcesses(); |
| 163 | for (ActivityManager.RunningAppProcessInfo proc : procs) { |
| 164 | final String pname = proc.processName; |
| 165 | if (pname.startsWith(getPackageName())) { |
| 166 | startTrackingProcess(proc.pid, pname, System.currentTimeMillis()); |
| 167 | } |
| 168 | } |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 169 | } |
| 170 | |
| 171 | @Override |
| 172 | public void onDestroy() { |
| 173 | mHandler.sendEmptyMessage(MSG_STOP); |
| 174 | } |
| 175 | |
| 176 | @Override |
| 177 | public int onStartCommand(Intent intent, int flags, int startId) { |
| 178 | Log.i(TAG, "Received start id " + startId + ": " + intent); |
| 179 | |
Daniel Sandler | 4988340 | 2013-06-25 13:20:23 -0400 | [diff] [blame] | 180 | if (intent != null) { |
| 181 | if (ACTION_START_TRACKING.equals(intent.getAction())) { |
| 182 | final int pid = intent.getIntExtra("pid", -1); |
| 183 | final String name = intent.getStringExtra("name"); |
Daniel Sandler | 0adfc5a | 2013-07-07 17:16:49 -0500 | [diff] [blame^] | 184 | final long start = intent.getLongExtra("start", System.currentTimeMillis()); |
| 185 | startTrackingProcess(pid, name, start); |
Daniel Sandler | 4988340 | 2013-06-25 13:20:23 -0400 | [diff] [blame] | 186 | } |
Daniel Sandler | b9eb286 | 2013-06-14 20:17:30 -0400 | [diff] [blame] | 187 | } |
| 188 | |
| 189 | mHandler.sendEmptyMessage(MSG_START); |
| 190 | |
| 191 | return START_STICKY; |
| 192 | } |
| 193 | |
| 194 | public class MemoryTrackerInterface extends Binder { |
| 195 | MemoryTracker getService() { |
| 196 | return MemoryTracker.this; |
| 197 | } |
| 198 | } |
| 199 | |
| 200 | private final IBinder mBinder = new MemoryTrackerInterface(); |
| 201 | |
| 202 | public IBinder onBind(Intent intent) { |
| 203 | mHandler.sendEmptyMessage(MSG_START); |
| 204 | |
| 205 | return mBinder; |
| 206 | } |
| 207 | } |