Refactor (f6f64620) · Commits · e / os / android_frameworks_base

core/java/com/android/internal/protolog/PerfettoProtoLogImpl.java

+92 −86

Original line number	Diff line number	Diff line
		@@ -76,7 +76,6 @@ import java.util.ArrayList;
		import java.util.Arrays;
		import java.util.List;
		import java.util.Map;
		import java.util.Optional;
		import java.util.Set;
		import java.util.TreeMap;
		import java.util.UUID;
		@@ -159,27 +158,18 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		*/
		@VisibleForTesting
		@Override
		public void log(LogLevel level, IProtoLogGroup group, long messageHash, int paramsMask,
		public void log(LogLevel logLevel, IProtoLogGroup group, long messageHash, int paramsMask,
		@Nullable Object[] args) {
		if (isProtoEnabled()) {
		long tsNanos = SystemClock.elapsedRealtimeNanos();
		final String stacktrace;
		if (mCollectStackTraceGroupCounts.getOrDefault(group, 0) > 0) {
		stacktrace = collectStackTrace();
		} else {
		stacktrace = null;
		}
		mBackgroundLoggingService.execute(() ->
		logToProto(level, group, messageHash, paramsMask, args, tsNanos,
		stacktrace));
		}
		if (group.isLogToLogcat()) {
		logToLogcat(group.getTag(), level, messageHash, args);
		}
		log(logLevel, group, new Message(messageHash, paramsMask), args);
		}

		@Override
		public void log(LogLevel logLevel, IProtoLogGroup group, String messageString, Object... args) {
		log(logLevel, group, new Message(messageString), args);
		}

		private void log(LogLevel logLevel, IProtoLogGroup group, Message message,
		@Nullable Object[] args) {
		if (isProtoEnabled()) {
		long tsNanos = SystemClock.elapsedRealtimeNanos();
		final String stacktrace;
		@@ -188,12 +178,17 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		} else {
		stacktrace = null;
		}
		mBackgroundLoggingService.execute(
		() -> logStringMessageToProto(logLevel, group, messageString, args,
		tsNanos, stacktrace));
		try {
		mBackgroundServiceLock.lock();
		mBackgroundLoggingService.execute(() ->
		logToProto(logLevel, group, message, args, tsNanos,
		stacktrace));
		} finally {
		mBackgroundServiceLock.unlock();
		}
		}
		if (group.isLogToLogcat()) {
		logToLogcat(group.getTag(), logLevel, messageString, args);
		logToLogcat(group.getTag(), logLevel, message, args);
		}
		}

		@@ -317,25 +312,23 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		os.end(outMessagesToken);
		}

		private void logToLogcat(String tag, LogLevel level, long messageHash,
		private void logToLogcat(String tag, LogLevel level, Message message,
		@Nullable Object[] args) {
		Trace.traceBegin(Trace.TRACE_TAG_WINDOW_MANAGER, "logToLogcat");
		try {
		String messageString = mViewerConfigReader.getViewerString(messageHash);
		String messageString = message.getMessage(mViewerConfigReader);

		if (messageString == null) {
		StringBuilder builder = new StringBuilder("UNKNOWN MESSAGE");
		for (Object o : args) {
		builder.append(" ").append(o);
		if (args != null) {
		builder.append(" args = (");
		builder.append(String.join(", ", Arrays.stream(args).map(
		Object::toString).toList()));
		builder.append(")");
		}
		messageString = builder.toString();
		args = new Object[0];
		}

		doLogToLogcat(tag, level, messageString, args);
		} finally {
		Trace.traceEnd(Trace.TRACE_TAG_WINDOW_MANAGER);
		}
		logToLogcat(tag, level, messageString, args);
		}

		private void logToLogcat(String tag, LogLevel level, String message, @Nullable Object[] args) {
		@@ -351,7 +344,13 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		@Nullable Object[] args) {
		String message;
		if (args != null) {
		try {
		message = TextUtils.formatSimple(messageString, args);
		} catch (IllegalArgumentException e) {
		message = "FORMAT_ERROR \"" + messageString + "\", args=("
		+ String.join(
		", ", Arrays.stream(args).map(Object::toString).toList()) + ")";
		}
		} else {
		message = messageString;
		}
		@@ -385,52 +384,18 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		}
		}

		private void logToProto(LogLevel level, IProtoLogGroup logGroup, long messageHash,
		int paramsMask, Object[] args, long tsNanos, @Nullable String stacktrace) {
		private void logToProto(LogLevel level, IProtoLogGroup logGroup, Message message, Object[] args,
		long tsNanos, @Nullable String stacktrace) {
		Trace.traceBegin(Trace.TRACE_TAG_WINDOW_MANAGER, "logToProto");
		try {
		doLogToProto(level, logGroup, new Message(messageHash), paramsMask, args, tsNanos,
		stacktrace);
		} finally {
		Trace.traceEnd(Trace.TRACE_TAG_WINDOW_MANAGER);
		}
		}

		private void logStringMessageToProto(LogLevel logLevel, IProtoLogGroup group,
		String messageString, Object[] args, long tsNanos, @Nullable String stacktrace) {
		Trace.traceBegin(Trace.TRACE_TAG_WINDOW_MANAGER, "logStringMessageToProto");
		try {
		doLogToProto(logLevel, group, messageString, args, tsNanos, stacktrace);
		doLogToProto(level, logGroup, message, args, tsNanos, stacktrace);
		} finally {
		Trace.traceEnd(Trace.TRACE_TAG_WINDOW_MANAGER);
		}
		}

		private void doLogToProto(LogLevel level, IProtoLogGroup logGroup, String messageString,
		Object[] args, long tsNanos, @Nullable String stacktrace) {
		final List<Integer> argTypes = LogDataType.parseFormatString(messageString);
		final int typeMask = LogDataType.logDataTypesToBitMask(argTypes);
		doLogToProto(level, logGroup, new Message(messageString), typeMask, args, tsNanos,
		stacktrace);
		}

		private static class Message {
		private final Optional<Long> mMessageHash;
		private final Optional<String> mMessageString;

		private Message(Long messageHash) {
		this.mMessageHash = Optional.of(messageHash);
		this.mMessageString = Optional.empty();
		}

		private Message(String messageString) {
		this.mMessageHash = Optional.empty();
		this.mMessageString = Optional.of(messageString);
		}
		}

		private void doLogToProto(LogLevel level, IProtoLogGroup logGroup, Message message,
		int paramsMask, Object[] args, long tsNanos, @Nullable String stacktrace) {
		Object[] args, long tsNanos, @Nullable String stacktrace) {
		mDataSource.trace(ctx -> {
		final ProtoLogDataSource.TlsState tlsState = ctx.getCustomTlsState();
		final LogLevel logFrom = tlsState.getLogFromLevel(logGroup.name());
		@@ -445,7 +410,7 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		// trace processing easier.
		int argIndex = 0;
		for (Object o : args) {
		int type = LogDataType.bitmaskToLogDataType(paramsMask, argIndex);
		int type = LogDataType.bitmaskToLogDataType(message.getMessageMask(), argIndex);
		if (type == LogDataType.STRING) {
		internStringArg(ctx, o.toString());
		}
		@@ -464,13 +429,13 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		boolean needsIncrementalState = false;

		long messageHash = 0;
		if (message.mMessageHash.isPresent()) {
		messageHash = message.mMessageHash.get();
		if (message.mMessageHash != null) {
		messageHash = message.mMessageHash;
		}
		if (message.mMessageString.isPresent()) {
		if (message.mMessageString != null) {
		needsIncrementalState = true;
		messageHash =
		internProtoMessage(ctx, level, logGroup, message.mMessageString.get());
		internProtoMessage(ctx, level, logGroup, message.mMessageString);
		}

		final ProtoOutputStream os = ctx.newTracePacket();
		@@ -486,7 +451,7 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		ArrayList<Double> doubleParams = new ArrayList<>();
		ArrayList<Boolean> booleanParams = new ArrayList<>();
		for (Object o : args) {
		int type = LogDataType.bitmaskToLogDataType(paramsMask, argIndex);
		int type = LogDataType.bitmaskToLogDataType(message.getMessageMask(), argIndex);
		try {
		switch (type) {
		case LogDataType.STRING:
		@@ -628,7 +593,7 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		ProtoLogDataSource.IncrementalState> ctx,
		Map<String, Integer> internMap,
		long fieldId,
		String string
		@NonNull String string
		) {
		final ProtoLogDataSource.IncrementalState incrementalState = ctx.getIncrementalState();

		@@ -774,8 +739,6 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		}

		private synchronized void onTracingInstanceStart(ProtoLogDataSource.ProtoLogConfig config) {
		this.mTracingInstances.incrementAndGet();

		final LogLevel defaultLogFrom = config.getDefaultGroupConfig().logFrom;
		for (int i = defaultLogFrom.ordinal(); i < LogLevel.values().length; i++) {
		mDefaultLogLevelCounts[i]++;
		@@ -786,12 +749,22 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		for (String overriddenGroupTag : overriddenGroupTags) {
		IProtoLogGroup group = mLogGroups.get(overriddenGroupTag);

		if (group == null) {
		throw new IllegalArgumentException("Trying to set config for \""
		+ overriddenGroupTag + "\" that isn't registered");
		}

		mLogLevelCounts.putIfAbsent(group, new int[LogLevel.values().length]);
		final int[] logLevelsCountsForGroup = mLogLevelCounts.get(group);

		final LogLevel logFromLevel = config.getConfigFor(overriddenGroupTag).logFrom;
		for (int i = defaultLogFrom.ordinal(); i < LogLevel.values().length; i++) {
		logLevelsCountsForGroup[logFromLevel.ordinal()]++;
		for (int i = logFromLevel.ordinal(); i < LogLevel.values().length; i++) {
		logLevelsCountsForGroup[i]++;
		}

		if (config.getConfigFor(overriddenGroupTag).collectStackTrace) {
		mCollectStackTraceGroupCounts.put(group,
		mCollectStackTraceGroupCounts.getOrDefault(group, 0) + 1);
		}

		if (config.getConfigFor(overriddenGroupTag).collectStackTrace) {
		@@ -801,6 +774,8 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		}

		mCacheUpdater.run();

		this.mTracingInstances.incrementAndGet();
		}

		private synchronized void onTracingInstanceStop(ProtoLogDataSource.ProtoLogConfig config) {
		@@ -846,5 +821,36 @@ public class PerfettoProtoLogImpl implements IProtoLog {
		pw.flush();
		}
		}

		private static class Message {
		private final Long mMessageHash;
		private final Integer mMessageMask;
		private final String mMessageString;

		private Message(Long messageHash, int messageMask) {
		this.mMessageHash = messageHash;
		this.mMessageMask = messageMask;
		this.mMessageString = null;
		}

		private Message(String messageString) {
		this.mMessageHash = null;
		final List<Integer> argTypes = LogDataType.parseFormatString(messageString);
		this.mMessageMask = LogDataType.logDataTypesToBitMask(argTypes);
		this.mMessageString = messageString;
		}

		private int getMessageMask() {
		return mMessageMask;
		}

		private String getMessage(ProtoLogViewerConfigReader viewerConfigReader) {
		if (mMessageString != null) {
		return mMessageString;
		}

		return viewerConfigReader.getViewerString(mMessageHash);
		}
		}
		}