Implement class to evaluate current use of Notification object memory (9dc62004) · Commits · e / os / android_frameworks_base

packages/SystemUI/src/com/android/systemui/flags/Flags.java

+3 −0

Original line number	Original line	Diff line number	Diff line
	@@ -68,6 +68,9 @@ public class Flags {

	public static final UnreleasedFlag INSTANT_VOICE_REPLY = new UnreleasedFlag(111, true);		public static final UnreleasedFlag INSTANT_VOICE_REPLY = new UnreleasedFlag(111, true);

			public static final UnreleasedFlag NOTIFICATION_MEMORY_MONITOR_ENABLED = new UnreleasedFlag(112,
			false);

	// next id: 112		// next id: 112

	/***************************************/		/***************************************/

packages/SystemUI/src/com/android/systemui/statusbar/notification/init/NotificationsControllerImpl.kt

+8 −0

Original line number	Original line	Diff line number	Diff line
	@@ -19,6 +19,8 @@ package com.android.systemui.statusbar.notification.init
	import android.service.notification.StatusBarNotification		import android.service.notification.StatusBarNotification
	import com.android.systemui.ForegroundServiceNotificationListener		import com.android.systemui.ForegroundServiceNotificationListener
	import com.android.systemui.dagger.SysUISingleton		import com.android.systemui.dagger.SysUISingleton
			import com.android.systemui.flags.FeatureFlags
			import com.android.systemui.flags.Flags
	import com.android.systemui.people.widget.PeopleSpaceWidgetManager		import com.android.systemui.people.widget.PeopleSpaceWidgetManager
	import com.android.systemui.plugins.statusbar.NotificationSwipeActionHelper.SnoozeOption		import com.android.systemui.plugins.statusbar.NotificationSwipeActionHelper.SnoozeOption
	import com.android.systemui.statusbar.NotificationListener		import com.android.systemui.statusbar.NotificationListener
	@@ -38,6 +40,7 @@ import com.android.systemui.statusbar.notification.collection.notifcollection.No
	import com.android.systemui.statusbar.notification.collection.render.NotifStackController		import com.android.systemui.statusbar.notification.collection.render.NotifStackController
	import com.android.systemui.statusbar.notification.interruption.HeadsUpViewBinder		import com.android.systemui.statusbar.notification.interruption.HeadsUpViewBinder
	import com.android.systemui.statusbar.notification.logging.NotificationLogger		import com.android.systemui.statusbar.notification.logging.NotificationLogger
			import com.android.systemui.statusbar.notification.logging.NotificationMemoryMonitor
	import com.android.systemui.statusbar.notification.row.NotifBindPipelineInitializer		import com.android.systemui.statusbar.notification.row.NotifBindPipelineInitializer
	import com.android.systemui.statusbar.notification.stack.NotificationListContainer		import com.android.systemui.statusbar.notification.stack.NotificationListContainer
	import com.android.systemui.statusbar.phone.CentralSurfaces		import com.android.systemui.statusbar.phone.CentralSurfaces
	@@ -71,6 +74,8 @@ class NotificationsControllerImpl @Inject constructor(
	private val peopleSpaceWidgetManager: PeopleSpaceWidgetManager,		private val peopleSpaceWidgetManager: PeopleSpaceWidgetManager,
	private val bubblesOptional: Optional<Bubbles>,		private val bubblesOptional: Optional<Bubbles>,
	private val fgsNotifListener: ForegroundServiceNotificationListener,		private val fgsNotifListener: ForegroundServiceNotificationListener,
			private val memoryMonitor: Lazy<NotificationMemoryMonitor>,
			private val featureFlags: FeatureFlags
	) : NotificationsController {		) : NotificationsController {

	override fun initialize(		override fun initialize(
	@@ -112,6 +117,9 @@ class NotificationsControllerImpl @Inject constructor(
	notificationLogger.setUpWithContainer(listContainer)		notificationLogger.setUpWithContainer(listContainer)
	peopleSpaceWidgetManager.attach(notificationListener)		peopleSpaceWidgetManager.attach(notificationListener)
	fgsNotifListener.init()		fgsNotifListener.init()
			if (featureFlags.isEnabled(Flags.NOTIFICATION_MEMORY_MONITOR_ENABLED)) {
			memoryMonitor.get().init()
			}
	}		}

	// TODO: Convert all functions below this line into listeners instead of public methods		// TODO: Convert all functions below this line into listeners instead of public methods

packages/SystemUI/src/com/android/systemui/statusbar/notification/logging/NotificationMemory.kt

0 → 100644

+41 −0

Original line number	Original line	Diff line number	Diff line
			/*
			*
			* Copyright (C) 2022 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.systemui.statusbar.notification.logging

			/** Describes usage of a notification. */
			data class NotificationMemoryUsage(
			val packageName: String,
			val notificationId: String,
			val objectUsage: NotificationObjectUsage,
			)

			/**
			* Describes current memory usage of a [android.app.Notification] object.
			*
			* The values are in bytes.
			*/
			data class NotificationObjectUsage(
			val smallIcon: Int,
			val largeIcon: Int,
			val extras: Int,
			val style: String?,
			val styleIcon: Int,
			val bigPicture: Int,
			val extender: Int,
			val hasCustomView: Boolean,
			)

packages/SystemUI/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitor.kt

0 → 100644

+239 −0

Original line number	Original line	Diff line number	Diff line
			/*
			*
			* Copyright (C) 2022 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.systemui.statusbar.notification.logging

			import android.app.Notification
			import android.app.Person
			import android.graphics.Bitmap
			import android.graphics.drawable.Icon
			import android.os.Bundle
			import android.os.Parcel
			import android.os.Parcelable
			import android.util.Log
			import androidx.annotation.WorkerThread
			import androidx.core.util.contains
			import com.android.systemui.Dumpable
			import com.android.systemui.dagger.SysUISingleton
			import com.android.systemui.dump.DumpManager
			import com.android.systemui.statusbar.notification.NotificationUtils
			import com.android.systemui.statusbar.notification.collection.NotifPipeline
			import com.android.systemui.statusbar.notification.collection.NotificationEntry
			import java.io.PrintWriter
			import javax.inject.Inject

			/** This class monitors and logs current Notification memory use. */
			@SysUISingleton
			class NotificationMemoryMonitor
			@Inject
			constructor(
			val notificationPipeline: NotifPipeline,
			val dumpManager: DumpManager,
			) : Dumpable {

			companion object {
			private const val TAG = "NotificationMemMonitor"
			private const val CAR_EXTENSIONS = "android.car.EXTENSIONS"
			private const val CAR_EXTENSIONS_LARGE_ICON = "large_icon"
			private const val TV_EXTENSIONS = "android.tv.EXTENSIONS"
			private const val WEARABLE_EXTENSIONS = "android.wearable.EXTENSIONS"
			private const val WEARABLE_EXTENSIONS_BACKGROUND = "background"
			}

			fun init() {
			Log.d(TAG, "NotificationMemoryMonitor initialized.")
			dumpManager.registerDumpable(javaClass.simpleName, this)
			}

			override fun dump(pw: PrintWriter, args: Array<out String>) {
			currentNotificationMemoryUse().forEach { use -> pw.println(use.toString()) }
			}

			@WorkerThread
			fun currentNotificationMemoryUse(): List<NotificationMemoryUsage> {
			return notificationMemoryUse(notificationPipeline.allNotifs)
			}

			/** Returns a list of memory use entries for currently shown notifications. */
			@WorkerThread
			fun notificationMemoryUse(
			notifications: Collection<NotificationEntry>
			): List<NotificationMemoryUsage> {
			return notifications.asSequence().map { entry ->
			val packageName = entry.sbn.packageName
			val notificationObjectUsage =
			computeNotificationObjectUse(entry.sbn.notification, hashSetOf())
			NotificationMemoryUsage(
			packageName,
			NotificationUtils.logKey(entry.sbn.key),
			notificationObjectUsage)
			}.toList()
			}

			/**
			* Computes the estimated memory usage of a given [Notification] object. It'll attempt to
			* inspect Bitmaps in the object and provide summary of memory usage.
			*/
			private fun computeNotificationObjectUse(
			notification: Notification,
			seenBitmaps: HashSet<Int>
			): NotificationObjectUsage {
			val extras = notification.extras
			val smallIconUse = computeIconUse(notification.smallIcon, seenBitmaps)
			val largeIconUse = computeIconUse(notification.getLargeIcon(), seenBitmaps)

			// Collect memory usage of extra styles

			// Big Picture
			val bigPictureIconUse =
			computeParcelableUse(extras, Notification.EXTRA_PICTURE_ICON, seenBitmaps) +
			computeParcelableUse(extras, Notification.EXTRA_LARGE_ICON_BIG, seenBitmaps)
			val bigPictureUse =
			computeParcelableUse(extras, Notification.EXTRA_PICTURE, seenBitmaps) +
			computeParcelableUse(extras, Notification.EXTRA_PICTURE_ICON, seenBitmaps)

			// People
			val peopleList = extras.getParcelableArrayList<Person>(Notification.EXTRA_PEOPLE_LIST)
			val peopleUse =
			peopleList?.sumOf { person -> computeIconUse(person.icon, seenBitmaps) } ?: 0

			// Calling
			val callingPersonUse =
			computeParcelableUse(extras, Notification.EXTRA_CALL_PERSON, seenBitmaps)
			val verificationIconUse =
			computeParcelableUse(extras, Notification.EXTRA_VERIFICATION_ICON, seenBitmaps)

			// Messages
			val messages =
			Notification.MessagingStyle.Message.getMessagesFromBundleArray(
			extras.getParcelableArray(Notification.EXTRA_MESSAGES)
			)
			val messagesUse =
			messages.sumOf { msg -> computeIconUse(msg.senderPerson?.icon, seenBitmaps) }
			val historicMessages =
			Notification.MessagingStyle.Message.getMessagesFromBundleArray(
			extras.getParcelableArray(Notification.EXTRA_HISTORIC_MESSAGES)
			)
			val historyicMessagesUse =
			historicMessages.sumOf { msg -> computeIconUse(msg.senderPerson?.icon, seenBitmaps) }

			// Extenders
			val carExtender = extras.getBundle(CAR_EXTENSIONS)
			val carExtenderSize = carExtender?.let { computeBundleSize(it) } ?: 0
			val carExtenderIcon =
			computeParcelableUse(carExtender, CAR_EXTENSIONS_LARGE_ICON, seenBitmaps)

			val tvExtender = extras.getBundle(TV_EXTENSIONS)
			val tvExtenderSize = tvExtender?.let { computeBundleSize(it) } ?: 0

			val wearExtender = extras.getBundle(WEARABLE_EXTENSIONS)
			val wearExtenderSize = wearExtender?.let { computeBundleSize(it) } ?: 0
			val wearExtenderBackground =
			computeParcelableUse(wearExtender, WEARABLE_EXTENSIONS_BACKGROUND, seenBitmaps)

			val style = notification.notificationStyle
			val hasCustomView = notification.contentView != null \|\| notification.bigContentView != null
			val extrasSize = computeBundleSize(extras)

			return NotificationObjectUsage(
			smallIconUse,
			largeIconUse,
			extrasSize,
			style?.simpleName,
			bigPictureIconUse +
			peopleUse +
			callingPersonUse +
			verificationIconUse +
			messagesUse +
			historyicMessagesUse,
			bigPictureUse,
			carExtenderSize +
			carExtenderIcon +
			tvExtenderSize +
			wearExtenderSize +
			wearExtenderBackground,
			hasCustomView
			)
			}

			/**
			* Calculates size of the bundle data (excluding FDs and other shared objects like ashmem
			* bitmaps). Can be slow.
			*/
			private fun computeBundleSize(extras: Bundle): Int {
			val parcel = Parcel.obtain()
			try {
			extras.writeToParcel(parcel, 0)
			return parcel.dataSize()
			} finally {
			parcel.recycle()
			}
			}

			/**
			* Deserializes [Icon], [Bitmap] or [Person] from extras and computes its memory use. Returns 0
			* if the key does not exist in extras.
			*/
			private fun computeParcelableUse(extras: Bundle?, key: String, seenBitmaps: HashSet<Int>): Int {
			return when (val parcelable = extras?.getParcelable<Parcelable>(key)) {
			is Bitmap -> computeBitmapUse(parcelable, seenBitmaps)
			is Icon -> computeIconUse(parcelable, seenBitmaps)
			is Person -> computeIconUse(parcelable.icon, seenBitmaps)
			else -> 0
			}
			}

			/**
			* Calculates the byte size of bitmaps or data in the Icon object. Returns 0 if the icon is
			* defined via Uri or a resource.
			*
			* @return memory usage in bytes or 0 if the icon is Uri/Resource based
			*/
			private fun computeIconUse(icon: Icon?, seenBitmaps: HashSet<Int>) =
			when (icon?.type) {
			Icon.TYPE_BITMAP -> computeBitmapUse(icon.bitmap, seenBitmaps)
			Icon.TYPE_ADAPTIVE_BITMAP -> computeBitmapUse(icon.bitmap, seenBitmaps)
			Icon.TYPE_DATA -> computeDataUse(icon, seenBitmaps)
			else -> 0
			}

			/**
			* Returns the amount of memory a given bitmap is using. If the bitmap reference is part of
			* seenBitmaps set, this method returns 0 to avoid double counting.
			*
			* @return memory usage of the bitmap in bytes
			*/
			private fun computeBitmapUse(bitmap: Bitmap, seenBitmaps: HashSet<Int>? = null): Int {
			val refId = System.identityHashCode(bitmap)
			if (seenBitmaps?.contains(refId) == true) {
			return 0
			}

			seenBitmaps?.add(refId)
			return bitmap.allocationByteCount
			}

			private fun computeDataUse(icon: Icon, seenBitmaps: HashSet<Int>): Int {
			val refId = System.identityHashCode(icon.dataBytes)
			if (seenBitmaps.contains(refId)) {
			return 0
			}

			seenBitmaps.add(refId)
			return icon.dataLength
			}
			}

packages/SystemUI/tests/src/com/android/systemui/statusbar/notification/logging/NotificationMemoryMonitorTest.kt

0 → 100644

+323 −0

File added.

Preview size limit exceeded, changes collapsed.