forked from pingcap/tidb
/
flags.go
35 lines (30 loc) · 918 Bytes
/
flags.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
// Copyright 2020 PingCAP, Inc. Licensed under Apache-2.0.
package storage
import (
"github.com/pingcap/errors"
"github.com/spf13/pflag"
)
// DefineFlags adds flags to the flag set corresponding to all backend options.
func DefineFlags(flags *pflag.FlagSet) {
defineS3Flags(flags)
defineGCSFlags(flags)
defineAzblobFlags(flags)
}
// HiddenFlagsForStream hidden flags for stream cmd.
func HiddenFlagsForStream(flags *pflag.FlagSet) {
hiddenGCSFlags(flags)
hiddenAzblobFlags(flags)
}
// ParseFromFlags obtains the backend options from the flag set.
func (options *BackendOptions) ParseFromFlags(flags *pflag.FlagSet) error {
if err := options.S3.parseFromFlags(flags); err != nil {
return errors.Trace(err)
}
if err := options.GCS.parseFromFlags(flags); err != nil {
return errors.Trace(err)
}
if err := options.Azblob.parseFromFlags(flags); err != nil {
return errors.Trace(err)
}
return nil
}