Merge pull request #32 from mongodb/global-clusters

Global Clusters Service

Author: marinsalinas

mongodbatlas/clusters_test.go

@@ -630,7 +630,7 @@ func TestClusters_UpdateProcessArgs(t *testing.T) {
 
 }
 
-func TestDatabaseUsers_GetProcessArgs(t *testing.T) {
+func TestClusters_GetProcessArgs(t *testing.T) {
 	setup()
 	defer teardown()
 
@@ -676,7 +676,7 @@ func TestClusters_checkClusterNameParam(t *testing.T) {
 	}
 }
 
-func TestDatabaseUsers_Get(t *testing.T) {
+func TestClusters_Get(t *testing.T) {
 	setup()
 	defer teardown()
 

mongodbatlas/global_clusters.go

@@ -0,0 +1,167 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+)
+
+const globalClustersBasePath = "groups/%s/clusters/%s/globalWrites/%s"
+
+//GlobalClustersService is an interface for interfacing with the Global Clusters
+// endpoints of the MongoDB Atlas API.
+//See more: https://docs.atlas.mongodb.com/reference/api/global-clusters/
+type GlobalClustersService interface {
+	Get(context.Context, string, string) (*GlobalCluster, *Response, error)
+	AddManagedNamespace(context.Context, string, string, *ManagedNamespace) (*GlobalCluster, *Response, error)
+	DeleteManagedNamespace(context.Context, string, string, *ManagedNamespace) (*GlobalCluster, *Response, error)
+	AddCustomZoneMappings(context.Context, string, string, *CustomZoneMappingsRequest) (*GlobalCluster, *Response, error)
+	DeleteCustomZoneMappings(context.Context, string, string) (*GlobalCluster, *Response, error)
+}
+
+//GlobalClustersServiceOp handles communication with the GlobalClusters related methos of the
+//MongoDB Atlas API
+type GlobalClustersServiceOp struct {
+	client *Client
+}
+
+var _ GlobalClustersService = &GlobalClustersServiceOp{}
+
+// GlobalCluster represents MongoDB Global Cluster Configuration in your Global Cluster.
+type GlobalCluster struct {
+	CustomZoneMapping map[string]string  `json:"customZoneMapping"`
+	ManagedNamespaces []ManagedNamespace `json:"managedNamespaces"`
+}
+
+//ManagedNamespace represents the information about managed namespace configuration.
+type ManagedNamespace struct {
+	Db             string `json:"db"`
+	Collection     string `json:"collection"`
+	CustomShardKey string `json:"customShardKey,omitempty"`
+}
+
+//CustomZoneMappingsRequest represents the request related to add custom zone mappings to a global cluster.
+type CustomZoneMappingsRequest struct {
+	CustomZoneMappings []CustomZoneMapping `json:"customZoneMappings"`
+}
+
+//CustomZoneMapping represents the custom zone mapping.
+type CustomZoneMapping struct {
+	Location string `json:"location"`
+	Zone     string `json:"zone"`
+}
+
+//Get retrieves all managed namespaces and custom zone mappings associated with the specified Global Cluster.
+//See more: https://docs.atlas.mongodb.com/reference/api/global-clusters-retrieve-namespaces/
+func (s *GlobalClustersServiceOp) Get(ctx context.Context, groupID string, clusterName string) (*GlobalCluster, *Response, error) {
+	if clusterName == "" {
+		return nil, nil, NewArgError("username", "must be set")
+	}
+
+	path := fmt.Sprintf("groups/%s/clusters/%s/globalWrites", groupID, clusterName)
+
+	req, err := s.client.NewRequest(ctx, http.MethodGet, path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(GlobalCluster)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return root, resp, err
+}
+
+//AddManagedNamespace adds a managed namespace to the specified Global Cluster.
+//See more: https://docs.atlas.mongodb.com/reference/api/database-users-create-a-user/
+func (s *GlobalClustersServiceOp) AddManagedNamespace(ctx context.Context, groupID string, clusterName string, createRequest *ManagedNamespace) (*GlobalCluster, *Response, error) {
+	if createRequest == nil {
+		return nil, nil, NewArgError("createRequest", "cannot be nil")
+	}
+
+	path := fmt.Sprintf(globalClustersBasePath, groupID, clusterName, "managedNamespaces")
+
+	req, err := s.client.NewRequest(ctx, http.MethodPost, path, createRequest)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(GlobalCluster)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return root, resp, err
+}
+
+//DeleteManagedNamespace deletes the managed namespace configuration of the global cluster given.
+//See more: https://docs.atlas.mongodb.com/reference/api/global-clusters-delete-namespace/
+func (s *GlobalClustersServiceOp) DeleteManagedNamespace(ctx context.Context, groupID string, clusterName string, deleteRequest *ManagedNamespace) (*GlobalCluster, *Response, error) {
+	if deleteRequest == nil {
+		return nil, nil, NewArgError("createRequest", "cannot be nil")
+	}
+
+	path := fmt.Sprintf(globalClustersBasePath, groupID, clusterName, "managedNamespaces")
+
+	req, err := s.client.NewRequest(ctx, http.MethodDelete, path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	q := req.URL.Query()
+	q.Add("collection", deleteRequest.Collection)
+	q.Add("db", deleteRequest.Db)
+	req.URL.RawQuery = q.Encode()
+
+	root := new(GlobalCluster)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return root, resp, err
+}
+
+//AddCustomZoneMappings adds an entry to the list of custom zone mappings for the specified Global Cluster.
+//See more: https://docs.atlas.mongodb.com/reference/api/global-clusters-add-customzonemapping/
+func (s *GlobalClustersServiceOp) AddCustomZoneMappings(ctx context.Context, groupID string, clusterName string, createRequest *CustomZoneMappingsRequest) (*GlobalCluster, *Response, error) {
+	if createRequest == nil {
+		return nil, nil, NewArgError("createRequest", "cannot be nil")
+	}
+
+	path := fmt.Sprintf(globalClustersBasePath, groupID, clusterName, "customZoneMapping")
+
+	req, err := s.client.NewRequest(ctx, http.MethodPost, path, createRequest)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(GlobalCluster)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return root, resp, err
+}
+
+//DeleteCustomZoneMappings removes all custom zone mappings from the specified Global Cluster.
+//See more: https://docs.atlas.mongodb.com/reference/api/global-clusters-delete-namespace/
+func (s *GlobalClustersServiceOp) DeleteCustomZoneMappings(ctx context.Context, groupID string, clusterName string) (*GlobalCluster, *Response, error) {
+	path := fmt.Sprintf(globalClustersBasePath, groupID, clusterName, "customZoneMapping")
+
+	req, err := s.client.NewRequest(ctx, http.MethodDelete, path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(GlobalCluster)
+	resp, err := s.client.Do(ctx, req, root)
+	if err != nil {
+		return nil, resp, err
+	}
+	return root, resp, err
+}