diff --git a/pkg/simple/client/network/options.go b/pkg/simple/client/network/options.go index 3bc42e1d6..0323e6dba 100644 --- a/pkg/simple/client/network/options.go +++ b/pkg/simple/client/network/options.go @@ -6,12 +6,15 @@ type Options struct { // weave scope service host WeaveScopeHost string `json:"weaveScopeHost,omitempty" yaml:"weaveScopeHost"` + + EnableNetworkPolicy bool `json:"enableNetworkPolicy,omitempty" yaml:"enableNetworkPolicy"` } // NewNetworkOptions returns a `zero` instance func NewNetworkOptions() *Options { return &Options{ - WeaveScopeHost: "weave-scope-app.weave.svc", + WeaveScopeHost: "weave-scope-app.weave.svc", + EnableNetworkPolicy: false, } } @@ -24,9 +27,11 @@ func (s *Options) ApplyTo(options *Options) { if s.WeaveScopeHost != "" { options.WeaveScopeHost = s.WeaveScopeHost } + options.EnableNetworkPolicy = s.EnableNetworkPolicy } func (s *Options) AddFlags(fs *pflag.FlagSet, c *Options) { - fs.StringVar(&s.WeaveScopeHost, "weave-scope-host", c.WeaveScopeHost, ""+ - "weave scope service host") + fs.StringVar(&s.WeaveScopeHost, "weave-scope-host", c.WeaveScopeHost, "weave scope service host") + fs.BoolVar(&s.EnableNetworkPolicy, "enable-network-policy", c.EnableNetworkPolicy, + "This field instructs KubeSphere to enable network policy or not.") }