Merge pull request #3281 from ioito/feature/qx-secgroup-cache

feature: 支持预cache secgroup到云上

cmd/climc/shell/secgroupcaches.go

@@ -0,0 +1,62 @@
+// Copyright 2019 Yunion
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package shell
+
+import (
+	"yunion.io/x/onecloud/pkg/mcclient"
+	"yunion.io/x/onecloud/pkg/mcclient/modules"
+	"yunion.io/x/onecloud/pkg/mcclient/options"
+)
+
+func init() {
+	type SecGroupCacheListOptions struct {
+		options.BaseListOptions
+		Secgroup string `help:"Secgroup ID or Name"`
+	}
+
+	R(&SecGroupCacheListOptions{}, "secgroup-cache-list", "List security group caches", func(s *mcclient.ClientSession, args *SecGroupCacheListOptions) error {
+		params, err := options.ListStructToParams(args)
+		if err != nil {
+			return err
+		}
+		result, err := modules.SecGroupCaches.List(s, params)
+		if err != nil {
+			return err
+		}
+		printList(result, modules.SecGroupCaches.GetColumns(s))
+		return nil
+	})
+	type SecGroupCacheIdOptions struct {
+		ID string `help:"ID or Name or secgroup cache"`
+	}
+	R(&SecGroupCacheIdOptions{}, "secgroup-cache-show", "Show security group cache", func(s *mcclient.ClientSession, args *SecGroupCacheIdOptions) error {
+		result, err := modules.SecGroupCaches.Get(s, args.ID, nil)
+		if err != nil {
+			return err
+		}
+		printObject(result)
+		return nil
+	})
+
+	R(&SecGroupCacheIdOptions{}, "secgroup-cache-delete", "Delete security group cache", func(s *mcclient.ClientSession, args *SecGroupCacheIdOptions) error {
+		result, err := modules.SecGroupCaches.Delete(s, args.ID, nil)
+		if err != nil {
+			return err
+		}
+		printObject(result)
+		return nil
+	})
+
+}

cmd/climc/shell/secgroups.go

@@ -176,4 +176,37 @@ func init() {
 		printObject(secgroups)
 		return nil
 	})
+
+	type SecurityGroupCacheSecurityGroup struct {
+		ID      string `help:"ID or Name of security group"`
+		VPC     string `help:"ID or Name of vpc"`
+		Classic *bool  `help:"Is classic vpc"`
+	}
+
+	R(&SecurityGroupCacheSecurityGroup{}, "secgroup-cache-secgroup", "Cache security group for special vpc", func(s *mcclient.ClientSession, args *SecurityGroupCacheSecurityGroup) error {
+		params, err := options.StructToParams(args)
+		secgroups, err := modules.SecGroups.PerformAction(s, args.ID, "cache-secgroup", params)
+		if err != nil {
+			return err
+		}
+		printObject(secgroups)
+		return nil
+	})
+
+	type SecurityGroupUncacheSecurityGroup struct {
+		ID    string `help:"ID or Name of security group"`
+		CACHE string `help:"ID of secgroup cache"`
+	}
+
+	R(&SecurityGroupUncacheSecurityGroup{}, "secgroup-uncache-secgroup", "Unache special secgroup cache", func(s *mcclient.ClientSession, args *SecurityGroupUncacheSecurityGroup) error {
+		params := jsonutils.NewDict()
+		params.Add(jsonutils.NewString(args.CACHE), "secgroupcache")
+		secgroups, err := modules.SecGroups.PerformAction(s, args.ID, "uncache-secgroup", params)
+		if err != nil {
+			return err
+		}
+		printObject(secgroups)
+		return nil
+	})
+
 }