Merge "Add methods to simplify UptcMap usage." (65287598) · Commits · e / os / android_frameworks_base

services/core/java/com/android/server/utils/quota/UptcMap.java

+40 −0

Original line number	Diff line number	Diff line
		@@ -22,6 +22,7 @@ import android.util.ArrayMap;
		import android.util.SparseArrayMap;

		import java.util.function.Consumer;
		import java.util.function.Function;

		/**
		* A SparseArrayMap of ArrayMaps, which is suitable for holding userId-packageName-tag combination
		@@ -94,6 +95,23 @@ class UptcMap<T> {
		return data != null ? data.get(tag) : null;
		}

		/**
		* Returns the saved object for the given UPTC. If there was no saved object, it will create a
		* new object using creator, insert it, and return it.
		*/
		@Nullable
		public T getOrCreate(int userId, @NonNull String packageName, @Nullable String tag,
		Function<Void, T> creator) {
		final ArrayMap<String, T> data = mData.get(userId, packageName);
		if (data == null \|\| !data.containsKey(tag)) {
		// We've never inserted data for this combination before. Create a new object.
		final T val = creator.apply(null);
		add(userId, packageName, tag, val);
		return val;
		}
		return data.get(tag);
		}

		/**
		* Returns the index for which {@link #getUserIdAtIndex(int)} would return the specified userId,
		* or a negative number if the specified userId is not mapped.
		@@ -160,4 +178,26 @@ class UptcMap<T> {
		}
		});
		}

		public void forEach(UptcDataConsumer<T> consumer) {
		final int uCount = userCount();
		for (int u = 0; u < uCount; ++u) {
		final int userId = getUserIdAtIndex(u);

		final int pkgCount = packageCountForUser(userId);
		for (int p = 0; p < pkgCount; ++p) {
		final String pkgName = getPackageNameAtIndex(u, p);

		final int tagCount = tagCountForUserAndPackage(userId, pkgName);
		for (int t = 0; t < tagCount; ++t) {
		final String tag = getTagAtIndex(u, p, t);
		consumer.accept(userId, pkgName, tag, get(userId, pkgName, tag));
		}
		}
		}
		}

		interface UptcDataConsumer<D> {
		void accept(int userId, @NonNull String packageName, @Nullable String tag, @Nullable D obj);
		}
		}