Simplify Defer: do not wrap first deferred func

Faster call to the deferred function for the most common case (a single Defer call).
frankban · Jun 10, 2023 · ee6dcaa · ee6dcaa
1 parent d07c1d3
commit ee6dcaa
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/quicktest.go b/quicktest.go
@@ -129,11 +129,13 @@ func (c *C) Defer(f func()) {
 	}
 
 	oldDeferred := c.deferred
-	c.deferred = func() {
-		if oldDeferred != nil {
+	if oldDeferred != nil {
+		c.deferred = func() {
 			defer oldDeferred()
+			f()
 		}
-		f()
+	} else {
+		c.deferred = f
 	}
 }