summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-iotools/close.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/codeberg.org/gruf/go-iotools/close.go')
-rw-r--r--vendor/codeberg.org/gruf/go-iotools/close.go35
1 files changed, 35 insertions, 0 deletions
diff --git a/vendor/codeberg.org/gruf/go-iotools/close.go b/vendor/codeberg.org/gruf/go-iotools/close.go
new file mode 100644
index 000000000..fbed7f33c
--- /dev/null
+++ b/vendor/codeberg.org/gruf/go-iotools/close.go
@@ -0,0 +1,35 @@
+package iotools
+
+import "io"
+
+// CloserFunc is a function signature which allows
+// a function to implement the io.Closer type.
+type CloserFunc func() error
+
+func (c CloserFunc) Close() error {
+ return c()
+}
+
+func CloserCallback(c io.Closer, cb func()) io.Closer {
+ return CloserFunc(func() error {
+ defer cb()
+ return c.Close()
+ })
+}
+
+// CloseOnce wraps an io.Closer to ensure it only performs the close logic once.
+func CloseOnce(c io.Closer) io.Closer {
+ return CloserFunc(func() error {
+ if c == nil {
+ // already run.
+ return nil
+ }
+
+ // Acquire.
+ cptr := c
+ c = nil
+
+ // Call the closer.
+ return cptr.Close()
+ })
+}