| /* |
| * Copyright (C) 2013 The Android Open Source Project |
| * |
| * 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.android.launcher3.testing; |
| |
| import android.app.ActivityManager; |
| import android.app.Service; |
| import android.content.Context; |
| import android.content.Intent; |
| import android.os.Binder; |
| import android.os.Debug; |
| import android.os.Handler; |
| import android.os.IBinder; |
| import android.os.Message; |
| import android.os.SystemClock; |
| import android.util.Log; |
| import android.util.LongSparseArray; |
| |
| import com.android.launcher3.util.TestingUtils; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| public class MemoryTracker extends Service { |
| public static final String TAG = MemoryTracker.class.getSimpleName(); |
| |
| private static final long UPDATE_RATE = 5000; |
| |
| private static final int MSG_START = 1; |
| private static final int MSG_STOP = 2; |
| private static final int MSG_UPDATE = 3; |
| |
| public static class ProcessMemInfo { |
| public int pid; |
| public String name; |
| public long startTime; |
| public long currentPss, currentUss; |
| public long[] pss = new long[256]; |
| public long[] uss = new long[256]; |
| //= new Meminfo[(int) (30 * 60 / (UPDATE_RATE / 1000))]; // 30 minutes |
| public long max = 1; |
| public int head = 0; |
| public ProcessMemInfo(int pid, String name, long start) { |
| this.pid = pid; |
| this.name = name; |
| this.startTime = start; |
| } |
| public long getUptime() { |
| return System.currentTimeMillis() - startTime; |
| } |
| }; |
| public final LongSparseArray<ProcessMemInfo> mData = new LongSparseArray<ProcessMemInfo>(); |
| public final ArrayList<Long> mPids = new ArrayList<Long>(); |
| private int[] mPidsArray = new int[0]; |
| private final Object mLock = new Object(); |
| |
| Handler mHandler = new Handler() { |
| @Override |
| public void handleMessage(Message m) { |
| switch (m.what) { |
| case MSG_START: |
| mHandler.removeMessages(MSG_UPDATE); |
| mHandler.sendEmptyMessage(MSG_UPDATE); |
| break; |
| case MSG_STOP: |
| mHandler.removeMessages(MSG_UPDATE); |
| break; |
| case MSG_UPDATE: |
| update(); |
| mHandler.removeMessages(MSG_UPDATE); |
| mHandler.sendEmptyMessageDelayed(MSG_UPDATE, UPDATE_RATE); |
| break; |
| } |
| } |
| }; |
| |
| ActivityManager mAm; |
| |
| public ProcessMemInfo getMemInfo(int pid) { |
| return mData.get(pid); |
| } |
| |
| public int[] getTrackedProcesses() { |
| return mPidsArray; |
| } |
| |
| public void startTrackingProcess(int pid, String name, long start) { |
| synchronized (mLock) { |
| final Long lpid = Long.valueOf(pid); |
| |
| if (mPids.contains(lpid)) return; |
| |
| mPids.add(lpid); |
| updatePidsArrayL(); |
| |
| mData.put(pid, new ProcessMemInfo(pid, name, start)); |
| } |
| } |
| |
| void updatePidsArrayL() { |
| final int N = mPids.size(); |
| mPidsArray = new int[N]; |
| StringBuffer sb = new StringBuffer("Now tracking processes: "); |
| for (int i=0; i<N; i++) { |
| final int p = mPids.get(i).intValue(); |
| mPidsArray[i] = p; |
| sb.append(p); sb.append(" "); |
| } |
| Log.v(TAG, sb.toString()); |
| } |
| |
| void update() { |
| synchronized (mLock) { |
| Debug.MemoryInfo[] dinfos = mAm.getProcessMemoryInfo(mPidsArray); |
| for (int i=0; i<dinfos.length; i++) { |
| Debug.MemoryInfo dinfo = dinfos[i]; |
| if (i > mPids.size()) { |
| Log.e(TAG, "update: unknown process info received: " + dinfo); |
| break; |
| } |
| final long pid = mPids.get(i).intValue(); |
| final ProcessMemInfo info = mData.get(pid); |
| info.head = (info.head+1) % info.pss.length; |
| info.pss[info.head] = info.currentPss = dinfo.getTotalPss(); |
| info.uss[info.head] = info.currentUss = dinfo.getTotalPrivateDirty(); |
| if (info.currentPss > info.max) info.max = info.currentPss; |
| if (info.currentUss > info.max) info.max = info.currentUss; |
| // Log.v(TAG, "update: pid " + pid + " pss=" + info.currentPss + " uss=" + info.currentUss); |
| if (info.currentPss == 0) { |
| Log.v(TAG, "update: pid " + pid + " has pss=0, it probably died"); |
| mData.remove(pid); |
| } |
| } |
| for (int i=mPids.size()-1; i>=0; i--) { |
| final long pid = mPids.get(i).intValue(); |
| if (mData.get(pid) == null) { |
| mPids.remove(i); |
| updatePidsArrayL(); |
| } |
| } |
| } |
| } |
| |
| @Override |
| public void onCreate() { |
| mAm = (ActivityManager) getSystemService(Context.ACTIVITY_SERVICE); |
| |
| // catch up in case we crashed but other processes are still running |
| List<ActivityManager.RunningServiceInfo> svcs = mAm.getRunningServices(256); |
| for (ActivityManager.RunningServiceInfo svc : svcs) { |
| if (svc.service.getPackageName().equals(getPackageName())) { |
| Log.v(TAG, "discovered running service: " + svc.process + " (" + svc.pid + ")"); |
| startTrackingProcess(svc.pid, svc.process, |
| System.currentTimeMillis() - (SystemClock.elapsedRealtime() - svc.activeSince)); |
| } |
| } |
| |
| List<ActivityManager.RunningAppProcessInfo> procs = mAm.getRunningAppProcesses(); |
| for (ActivityManager.RunningAppProcessInfo proc : procs) { |
| final String pname = proc.processName; |
| if (pname.startsWith(getPackageName())) { |
| Log.v(TAG, "discovered other running process: " + pname + " (" + proc.pid + ")"); |
| startTrackingProcess(proc.pid, pname, System.currentTimeMillis()); |
| } |
| } |
| } |
| |
| @Override |
| public void onDestroy() { |
| mHandler.sendEmptyMessage(MSG_STOP); |
| } |
| |
| @Override |
| public int onStartCommand(Intent intent, int flags, int startId) { |
| Log.v(TAG, "Received start id " + startId + ": " + intent); |
| |
| if (intent != null) { |
| if (TestingUtils.ACTION_START_TRACKING.equals(intent.getAction())) { |
| final int pid = intent.getIntExtra("pid", -1); |
| final String name = intent.getStringExtra("name"); |
| final long start = intent.getLongExtra("start", System.currentTimeMillis()); |
| startTrackingProcess(pid, name, start); |
| } |
| } |
| |
| mHandler.sendEmptyMessage(MSG_START); |
| |
| return START_STICKY; |
| } |
| |
| public class MemoryTrackerInterface extends Binder { |
| MemoryTracker getService() { |
| return MemoryTracker.this; |
| } |
| } |
| |
| private final IBinder mBinder = new MemoryTrackerInterface(); |
| |
| public IBinder onBind(Intent intent) { |
| mHandler.sendEmptyMessage(MSG_START); |
| |
| return mBinder; |
| } |
| } |