-
Notifications
You must be signed in to change notification settings - Fork 1k
/
shim_io_file.go
80 lines (65 loc) · 1.4 KB
/
shim_io_file.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
// Copyright (c) 2022 Ant Group
//
// SPDX-License-Identifier: Apache-2.0
//
package containerdshim
import (
"context"
"io"
"net/url"
"os"
"path/filepath"
cioutil "github.com/containerd/containerd/pkg/ioutil"
)
var (
_ IO = &fileIO{}
)
// fileIO only support write both stdout/stderr to the same file
type fileIO struct {
outw io.WriteCloser
errw io.WriteCloser
path string
}
// openLogFile opens/creates a container log file with its directory.
func openLogFile(path string) (*os.File, error) {
if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
return nil, err
}
return os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0640)
}
func newFileIO(ctx context.Context, stdio *stdio, uri *url.URL) (*fileIO, error) {
var outw, errw, f io.WriteCloser
var err error
logFile := uri.Path
if f, err = openLogFile(logFile); err != nil {
return nil, err
}
if stdio.Stdout != "" {
outw = cioutil.NewSerialWriteCloser(f)
}
if !stdio.Console && stdio.Stderr != "" {
errw = cioutil.NewSerialWriteCloser(f)
}
return &fileIO{
path: logFile,
outw: outw,
errw: errw,
}, nil
}
func (fi *fileIO) Close() error {
if fi.outw != nil {
return wc(fi.outw)
} else if fi.errw != nil {
return wc(fi.errw)
}
return nil
}
func (fi *fileIO) Stdin() io.ReadCloser {
return nil
}
func (fi *fileIO) Stdout() io.Writer {
return fi.outw
}
func (fi *fileIO) Stderr() io.Writer {
return fi.errw
}