volume snapshot

pkg/simple/client/k8s/kubernetes.go

@@ -1,6 +1,7 @@
 package k8s
 
 import (
+	snapshotclient "github.com/kubernetes-csi/external-snapshotter/v2/pkg/client/clientset/versioned"
 	applicationclientset "github.com/kubernetes-sigs/application/pkg/client/clientset/versioned"
 	istioclient "istio.io/client-go/pkg/clientset/versioned"
 	"k8s.io/client-go/discovery"
@@ -16,6 +17,7 @@ type Client interface {
 	KubeSphere() kubesphere.Interface
 	Istio() istioclient.Interface
 	Application() applicationclientset.Interface
+	Snapshot() snapshotclient.Interface
 	Discovery() discovery.DiscoveryInterface
 	Master() string
 	Config() *rest.Config
@@ -35,6 +37,8 @@ type kubernetesClient struct {
 
 	istio istioclient.Interface
 
+	snapshot snapshotclient.Interface
+
 	master string
 
 	config *rest.Config
@@ -56,6 +60,7 @@ func NewKubernetesClientOrDie(options *KubernetesOptions) Client {
 		ks:              kubesphere.NewForConfigOrDie(config),
 		istio:           istioclient.NewForConfigOrDie(config),
 		application:     applicationclientset.NewForConfigOrDie(config),
+		snapshot:        snapshotclient.NewForConfigOrDie(config),
 		master:          config.Host,
 		config:          config,
 	}
@@ -104,6 +109,11 @@ func NewKubernetesClient(options *KubernetesOptions) (Client, error) {
 		return nil, err
 	}
 
+	k.snapshot, err = snapshotclient.NewForConfig(config)
+	if err != nil {
+		return nil, err
+	}
+
 	k.master = options.Master
 	k.config = config
 
@@ -130,6 +140,10 @@ func (k *kubernetesClient) Istio() istioclient.Interface {
 	return k.istio
 }
 
+func (k *kubernetesClient) Snapshot() snapshotclient.Interface {
+	return k.snapshot
+}
+
 // master address used to generate kubeconfig for downloading
 func (k *kubernetesClient) Master() string {
 	return k.master

pkg/simple/client/k8s/null_client.go

@@ -1,6 +1,7 @@
 package k8s
 
 import (
+	snapshotclient "github.com/kubernetes-csi/external-snapshotter/v2/pkg/client/clientset/versioned"
 	application "github.com/kubernetes-sigs/application/pkg/client/clientset/versioned"
 	istio "istio.io/client-go/pkg/clientset/versioned"
 	"k8s.io/client-go/discovery"
@@ -32,6 +33,10 @@ func (n nullClient) Application() application.Interface {
 	return nil
 }
 
+func (n nullClient) Snapshot() snapshotclient.Interface {
+	return nil
+}
+
 func (n nullClient) Discovery() discovery.DiscoveryInterface {
 	return nil
 }