From 51b062f6cdfb59ef82d0a6fa2ca79cb00dd953d2 Mon Sep 17 00:00:00 2001
From: Dean Deng <deandeng@google.com>
Date: Mon, 2 Nov 2020 19:13:16 -0800
Subject: Skip log.Sprintfs when leak check logging is not enabled.

PiperOrigin-RevId: 340361998
---
 pkg/refsvfs2/refs_map.go | 29 ++++++++++++++++++++---------
 1 file changed, 20 insertions(+), 9 deletions(-)

(limited to 'pkg')

diff --git a/pkg/refsvfs2/refs_map.go b/pkg/refsvfs2/refs_map.go
index faf191f39..9fbc5466f 100644
--- a/pkg/refsvfs2/refs_map.go
+++ b/pkg/refsvfs2/refs_map.go
@@ -62,7 +62,9 @@ func Register(obj CheckedObject) {
 		}
 		liveObjects[obj] = struct{}{}
 		liveObjectsMu.Unlock()
-		logEvent(obj, "registered")
+		if leakCheckEnabled() && obj.LogRefs() {
+			logEvent(obj, "registered")
+		}
 	}
 }
 
@@ -75,31 +77,40 @@ func Unregister(obj CheckedObject) {
 			panic(fmt.Sprintf("Expected to find entry in leak checking map for reference %p", obj))
 		}
 		delete(liveObjects, obj)
-		logEvent(obj, "unregistered")
+		if leakCheckEnabled() && obj.LogRefs() {
+			logEvent(obj, "unregistered")
+		}
 	}
 }
 
 // LogIncRef logs a reference increment.
 func LogIncRef(obj CheckedObject, refs int64) {
-	logEvent(obj, fmt.Sprintf("IncRef to %d", refs))
+	if leakCheckEnabled() && obj.LogRefs() {
+		logEvent(obj, fmt.Sprintf("IncRef to %d", refs))
+	}
 }
 
 // LogTryIncRef logs a successful TryIncRef call.
 func LogTryIncRef(obj CheckedObject, refs int64) {
-	logEvent(obj, fmt.Sprintf("TryIncRef to %d", refs))
+	if leakCheckEnabled() && obj.LogRefs() {
+		logEvent(obj, fmt.Sprintf("TryIncRef to %d", refs))
+	}
 }
 
 // LogDecRef logs a reference decrement.
 func LogDecRef(obj CheckedObject, refs int64) {
-	logEvent(obj, fmt.Sprintf("DecRef to %d", refs))
+	if leakCheckEnabled() && obj.LogRefs() {
+		logEvent(obj, fmt.Sprintf("DecRef to %d", refs))
+	}
 }
 
 // logEvent logs a message for the given reference-counted object.
+//
+// obj.LogRefs() should be checked before calling logEvent, in order to avoid
+// calling any text processing needed to evaluate msg.
 func logEvent(obj CheckedObject, msg string) {
-	if obj.LogRefs() {
-		log.Infof("[%s %p] %s:", obj.RefType(), obj, msg)
-		log.Infof(refs_vfs1.FormatStack(refs_vfs1.RecordStack()))
-	}
+	log.Infof("[%s %p] %s:", obj.RefType(), obj, msg)
+	log.Infof(refs_vfs1.FormatStack(refs_vfs1.RecordStack()))
 }
 
 // DoLeakCheck iterates through the live object map and logs a message for each
-- 
cgit v1.2.3