swap.Chain()

tada-team · Mar 14, 2021 · 011b60f · 011b60f
1 parent 82d046c
commit 011b60f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -19,4 +19,13 @@ func TestWithSwap(t *testing.T) {
 	defer swap.Int(&myConfig.Bar, 42)()
 	// ...test cases...
 }
+
+// more sugar
+func TestWithSwapChain(t *testing.T) {
+    defer swap.Chain(
+    	swap.Bool(&myConfig.Foo, "test value"),
+        swap.Int(&myConfig.Bar, 42),
+    )
+    // ...test cases...
+}
 ```
diff --git a/swap.go b/swap.go
@@ -2,6 +2,14 @@ package swap
 
 import "time"
 
+func Chain(fns ...func()) func() {
+	return func() {
+		for _, fn := range fns {
+			fn()
+		}
+	}
+}
+
 func Bool(p *bool, v bool) func() {
 	old := *p
 	*p = v
@@ -26,6 +34,12 @@ func Int(p *int, v int) func() {
 	return func() { *p = old }
 }
 
+func Int64(p *int64, v int64) func() {
+	old := *p
+	*p = v
+	return func() { *p = old }
+}
+
 func Duration(p *time.Duration, v time.Duration) func() {
 	old := *p
 	*p = v