/
config.go
106 lines (92 loc) · 2.11 KB
/
config.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*
* Copyright (c) 2020. Ant Group. All rights reserved.
*
* SPDX-License-Identifier: Apache-2.0
*/
package stargz
import (
"os"
"github.com/containerd/nydus-snapshotter/config"
"github.com/containerd/nydus-snapshotter/pkg/filesystem/meta"
"github.com/containerd/nydus-snapshotter/pkg/process"
"github.com/pkg/errors"
)
func WithMeta(root string) NewFSOpt {
return func(d *filesystem) error {
if root == "" {
return errors.New("rootDir is required")
}
d.FileSystemMeta = meta.FileSystemMeta{
RootDir: root,
}
return nil
}
}
func WithNydusdBinaryPath(p string) NewFSOpt {
return func(d *filesystem) error {
if p == "" {
return errors.New("nydusd binary path is required")
}
d.nydusdBinaryPath = p
return nil
}
}
func WithProcessManager(pm *process.Manager) NewFSOpt {
return func(d *filesystem) error {
if pm == nil {
return errors.New("process manager cannot be nil")
}
d.manager = pm
return nil
}
}
func WithNydusImageBinaryPath(p string) NewFSOpt {
return func(d *filesystem) error {
if p == "" {
return errors.New("nydus image binary path is required")
}
d.nydusdImageBinaryPath = p
return nil
}
}
func WithDaemonConfig(cfg config.DaemonConfig) NewFSOpt {
return func(d *filesystem) error {
if (config.DaemonConfig{}) == cfg {
return errors.New("daemon config is empty")
}
d.daemonCfg = cfg
return nil
}
}
func WithLogLevel(logLevel string) NewFSOpt {
return func(d *filesystem) error {
if logLevel == "" {
d.logLevel = config.DefaultLogLevel
} else {
d.logLevel = logLevel
}
return nil
}
}
func WithLogDir(dir string) NewFSOpt {
return func(d *filesystem) error {
if err := os.MkdirAll(dir, 0755); err != nil {
return errors.Errorf("failed to create logDir %s: %v", dir, err)
}
d.logDir = dir
return nil
}
}
func WithLogToStdout(logToStdout bool) NewFSOpt {
return func(d *filesystem) error {
d.logToStdout = logToStdout
return nil
}
}
type NewFSOpt func(d *filesystem) error
func WithNydusdThreadNum(nydusdThreadNum int) NewFSOpt {
return func(d *filesystem) error {
d.nydusdThreadNum = nydusdThreadNum
return nil
}
}