Dianne Hackborn | b5a380d | 2015-05-20 18:18:46 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 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.server.am; |
| 18 | |
| 19 | import android.app.ActivityOptions; |
| 20 | import android.app.PendingIntent; |
| 21 | import android.content.Context; |
| 22 | import android.content.Intent; |
| 23 | import android.os.Bundle; |
| 24 | import android.os.SystemClock; |
| 25 | import android.util.ArrayMap; |
| 26 | import android.util.MutableLong; |
| 27 | import android.util.TimeUtils; |
Yi Jin | 148d7f4 | 2017-11-28 14:23:56 -0800 | [diff] [blame] | 28 | import android.util.proto.ProtoOutputStream; |
| 29 | import android.util.proto.ProtoUtils; |
| 30 | |
Dianne Hackborn | b5a380d | 2015-05-20 18:18:46 -0700 | [diff] [blame] | 31 | import java.io.PrintWriter; |
| 32 | |
| 33 | /** |
| 34 | * Tracks the time a user spent in an app. |
| 35 | */ |
| 36 | public class AppTimeTracker { |
| 37 | private final PendingIntent mReceiver; |
| 38 | |
| 39 | private long mTotalTime; |
| 40 | private final ArrayMap<String, MutableLong> mPackageTimes = new ArrayMap<>(); |
| 41 | |
| 42 | private long mStartedTime; |
| 43 | private String mStartedPackage; |
| 44 | private MutableLong mStartedPackageTime; |
| 45 | |
| 46 | public AppTimeTracker(PendingIntent receiver) { |
| 47 | mReceiver = receiver; |
| 48 | } |
| 49 | |
| 50 | public void start(String packageName) { |
| 51 | long now = SystemClock.elapsedRealtime(); |
| 52 | if (mStartedTime == 0) { |
| 53 | mStartedTime = now; |
| 54 | } |
| 55 | if (!packageName.equals(mStartedPackage)) { |
| 56 | if (mStartedPackageTime != null) { |
| 57 | long elapsedTime = now - mStartedTime; |
| 58 | mStartedPackageTime.value += elapsedTime; |
| 59 | mTotalTime += elapsedTime; |
| 60 | } |
| 61 | mStartedPackage = packageName; |
| 62 | mStartedPackageTime = mPackageTimes.get(packageName); |
| 63 | if (mStartedPackageTime == null) { |
| 64 | mStartedPackageTime = new MutableLong(0); |
| 65 | mPackageTimes.put(packageName, mStartedPackageTime); |
| 66 | } |
| 67 | } |
| 68 | } |
| 69 | |
| 70 | public void stop() { |
| 71 | if (mStartedTime != 0) { |
| 72 | long elapsedTime = SystemClock.elapsedRealtime() - mStartedTime; |
| 73 | mTotalTime += elapsedTime; |
| 74 | if (mStartedPackageTime != null) { |
| 75 | mStartedPackageTime.value += elapsedTime; |
| 76 | } |
| 77 | mStartedPackage = null; |
| 78 | mStartedPackageTime = null; |
| 79 | } |
| 80 | } |
| 81 | |
| 82 | public void deliverResult(Context context) { |
| 83 | stop(); |
| 84 | Bundle extras = new Bundle(); |
Dianne Hackborn | 67ba2c7 | 2015-06-05 14:23:38 -0700 | [diff] [blame] | 85 | extras.putLong(ActivityOptions.EXTRA_USAGE_TIME_REPORT, mTotalTime); |
Dianne Hackborn | b5a380d | 2015-05-20 18:18:46 -0700 | [diff] [blame] | 86 | Bundle pkgs = new Bundle(); |
| 87 | for (int i=mPackageTimes.size()-1; i>=0; i--) { |
| 88 | pkgs.putLong(mPackageTimes.keyAt(i), mPackageTimes.valueAt(i).value); |
| 89 | } |
Dianne Hackborn | 67ba2c7 | 2015-06-05 14:23:38 -0700 | [diff] [blame] | 90 | extras.putBundle(ActivityOptions.EXTRA_USAGE_TIME_REPORT_PACKAGES, pkgs); |
Dianne Hackborn | b5a380d | 2015-05-20 18:18:46 -0700 | [diff] [blame] | 91 | Intent fillinIntent = new Intent(); |
| 92 | fillinIntent.putExtras(extras); |
| 93 | try { |
| 94 | mReceiver.send(context, 0, fillinIntent); |
| 95 | } catch (PendingIntent.CanceledException e) { |
| 96 | } |
| 97 | } |
| 98 | |
| 99 | public void dumpWithHeader(PrintWriter pw, String prefix, boolean details) { |
| 100 | pw.print(prefix); pw.print("AppTimeTracker #"); |
| 101 | pw.print(Integer.toHexString(System.identityHashCode(this))); |
| 102 | pw.println(":"); |
| 103 | dump(pw, prefix + " ", details); |
| 104 | } |
| 105 | |
| 106 | public void dump(PrintWriter pw, String prefix, boolean details) { |
| 107 | pw.print(prefix); pw.print("mReceiver="); pw.println(mReceiver); |
| 108 | pw.print(prefix); pw.print("mTotalTime="); |
| 109 | TimeUtils.formatDuration(mTotalTime, pw); |
| 110 | pw.println(); |
| 111 | for (int i = 0; i < mPackageTimes.size(); i++) { |
| 112 | pw.print(prefix); pw.print("mPackageTime:"); pw.print(mPackageTimes.keyAt(i)); |
| 113 | pw.print("="); |
| 114 | TimeUtils.formatDuration(mPackageTimes.valueAt(i).value, pw); |
| 115 | pw.println(); |
| 116 | } |
| 117 | if (details && mStartedTime != 0) { |
| 118 | pw.print(prefix); pw.print("mStartedTime="); |
| 119 | TimeUtils.formatDuration(SystemClock.elapsedRealtime(), mStartedTime, pw); |
| 120 | pw.println(); |
| 121 | pw.print(prefix); pw.print("mStartedPackage="); pw.println(mStartedPackage); |
| 122 | } |
| 123 | } |
Yi Jin | 148d7f4 | 2017-11-28 14:23:56 -0800 | [diff] [blame] | 124 | |
| 125 | void writeToProto(ProtoOutputStream proto, long fieldId, boolean details) { |
| 126 | final long token = proto.start(fieldId); |
| 127 | proto.write(AppTimeTrackerProto.RECEIVER, mReceiver.toString()); |
| 128 | proto.write(AppTimeTrackerProto.TOTAL_DURATION_MS, mTotalTime); |
| 129 | for (int i=0; i<mPackageTimes.size(); i++) { |
| 130 | final long ptoken = proto.start(AppTimeTrackerProto.PACKAGE_TIMES); |
| 131 | proto.write(AppTimeTrackerProto.PackageTime.PACKAGE, mPackageTimes.keyAt(i)); |
| 132 | proto.write(AppTimeTrackerProto.PackageTime.DURATION_MS, mPackageTimes.valueAt(i).value); |
| 133 | proto.end(ptoken); |
| 134 | } |
| 135 | if (details && mStartedTime != 0) { |
| 136 | ProtoUtils.toDuration(proto, AppTimeTrackerProto.STARTED_TIME, |
| 137 | mStartedTime, SystemClock.elapsedRealtime()); |
| 138 | proto.write(AppTimeTrackerProto.STARTED_PACKAGE, mStartedPackage); |
| 139 | } |
| 140 | proto.end(token); |
| 141 | } |
Dianne Hackborn | b5a380d | 2015-05-20 18:18:46 -0700 | [diff] [blame] | 142 | } |