blob: 3f0ee778003036cf3be023fd3faa45640d9198c4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
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()
})
}
func CloserAfterCallback(c io.Closer, cb func()) io.Closer {
return CloserFunc(func() (err error) {
defer func() { err = c.Close() }()
cb()
return
})
}
// 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()
})
}
|
