-
Notifications
You must be signed in to change notification settings - Fork 3
/
close.go
62 lines (55 loc) · 1.14 KB
/
close.go
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package ioKit
import (
"io"
"os"
)
// TryToClose 尝试关闭(如果实现了io.Closer接口的话).
/*
@param writers (1) 可以为nil(即不传参)
(2) 其中可以有nil
@return 发生error的话,返回第一个
*/
func TryToClose(objs ...any) error {
var err error
for _, obj := range objs {
if closer, ok := obj.(io.Closer); ok {
tmpErr := CloseSingle(closer)
if tmpErr != nil && err == nil {
err = tmpErr
}
}
}
return err
}
// Close
/*
PS: 就算循环过程中返回了非nil的error,也要继续向下循环(尽可能多地关闭).
@param closers (1) 可以为nil(即不传参);(2) 其中可以有nil
@return 发生error的话,返回第一个
*/
func Close(closers ...io.Closer) error {
var err error
for _, closer := range closers {
tmpErr := CloseSingle(closer)
if tmpErr != nil && err == nil {
err = tmpErr
}
}
return err
}
func CloseSingle(closer io.Closer) error {
if closer != nil {
switch closer {
case os.Stdin:
fallthrough
case os.Stdout:
fallthrough
case os.Stderr:
// 这三种不关闭
return nil
default:
return closer.Close()
}
}
return nil
}