kms: add examples for how to use a kms.Client (#13)

* kms: add examples for how to use a `kms.Client` This commit adds examples for how to add and remove cluster nodes, as well as how to create, delete and list enclaves. Signed-off-by: Andreas Auernhammer <[email protected]>
minio · Mar 5, 2024 · 1f846be · 1f846be
1 parent 13e2976
commit 1f846be
Showing 1 changed file with 220 additions and 0 deletions.
diff --git a/kms/client-examples_test.go b/kms/client-examples_test.go
@@ -5,7 +5,10 @@
 package kms_test
 
 import (
+	"context"
 	"crypto/tls"
+	"fmt"
+	"io"
 	"log"
 
 	"github.com/minio/kms-go/kms"
@@ -44,3 +47,220 @@ func ExampleNewClient() {
 
 	// Output:
 }
+
+// ExampleClient_AddNode shows how to add a KMS server to an existing
+// KMS cluster dynamically expanding it. The added KMS server must not
+// be part of an exisiting cluster.
+func ExampleClient_AddNode() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	request := &kms.AddClusterNodeRequest{
+		Host: "10.1.2.3:7373",
+	}
+	if err = client.AddNode(context.TODO(), request); err != nil {
+		log.Fatalf("Failed to add server '%s' to cluster: %v", request.Host, err)
+	}
+}
+
+// ExampleClient_RemoveNode shows how to remove a KMS server from the
+// cluster it is currently part of.
+func ExampleClient_RemoveNode() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	request := &kms.RemoveClusterNodeRequest{
+		Host: "10.1.2.3:7373",
+	}
+	if err = client.RemoveNode(context.TODO(), request); err != nil {
+		log.Fatalf("Failed to remove server '%s' from cluster: %v", request.Host, err)
+	}
+}
+
+// ExampleClient_ClusterStatus shows how to fetch cluster status information
+// from a KMS cluster.
+func ExampleClient_ClusterStatus() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	status, err := client.ClusterStatus(context.TODO(), &kms.ClusterStatusRequest{})
+	if err != nil {
+		log.Fatalf("Failed to fetch cluster status information: %v", err)
+	}
+	log.Printf("Servers: online [%d] - offline [%d]", len(status.NodesUp), len(status.NodesDown))
+}
+
+// ExampleClient_CreateEnclave shows how to create a new enclave.
+func ExampleClient_CreateEnclave() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	request := &kms.CreateEnclaveRequest{
+		Name: "minio-tenant-foo",
+	}
+	if err = client.CreateEnclave(context.TODO(), request); err != nil {
+		log.Fatalf("Failed to create enclave '%s': %v", request.Name, err)
+	}
+}
+
+// ExampleClient_DeleteEnclave shows how to delete an existing enclave.
+func ExampleClient_DeleteEnclave() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	request := &kms.DeleteEnclaveRequest{
+		Name: "minio-tenant-foo",
+	}
+	if err = client.DeleteEnclave(context.TODO(), request); err != nil {
+		log.Fatalf("Failed to delete enclave '%s': %v", request.Name, err)
+	}
+}
+
+// ExampleClient_EnclaveStatus shows how to fetch status information about two enclaves.
+// Fetching information about multiple enclaves requires just a single network request.
+func ExampleClient_EnclaveStatus() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	requests := []*kms.EnclaveStatusRequest{
+		{Name: "minio-tenant-foo"},
+		{Name: "minio-tenant-bar"},
+	}
+	responses, err := client.EnclaveStatus(context.TODO(), requests...)
+	if err != nil {
+		log.Fatalf("Failed to fetch enclave status: %v", err)
+	}
+
+	for _, response := range responses {
+		fmt.Println(response.Name)
+	}
+}
+
+// ExampleClient_EnclaveStatus shows how to fetch status information about two enclaves.
+// Fetching information about multiple enclaves requires just a single network request.
+func ExampleClient_ListEnclaves() {
+	key, err := kms.ParseAPIKey("k1:d7cY_5k8HbBGkZpoy2hGmvkxg83QDBXsA_nFXDfTk2E")
+	if err != nil {
+		log.Fatalf("Failed to parse KMS API key: %v", err)
+	}
+
+	client, err := kms.NewClient(&kms.Config{
+		Endpoints: []string{
+			"127.0.0.1:7373",
+		},
+		APIKey: key,
+		TLS: &tls.Config{
+			RootCAs:            nil,   // Use nil for system root CAs or customize
+			InsecureSkipVerify: false, // Don't skip TLS cert verification in prod
+		},
+	})
+	if err != nil {
+		log.Fatalf("Failed to create KMS client: %v", err)
+	}
+
+	iter := kms.Iter[kms.EnclaveStatusResponse]{
+		NextFn: client.ListEnclaves,
+	}
+	for v, err := iter.Next(context.TODO()); err != io.EOF; v, err = iter.Next(context.TODO()) {
+		if err != nil {
+			log.Fatalf("Failed to list enclaves: %v", err)
+		}
+		fmt.Println(v.Name)
+	}
+}