From 1055df5fcedce7e63d32aa4063305a4aa68603a8 Mon Sep 17 00:00:00 2001 From: Zhengyi Lai Date: Fri, 12 Jun 2020 15:12:38 +0800 Subject: [PATCH] Add option to enable network policy or not --- pkg/simple/client/network/options.go | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) 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.") }