370 lines
13 KiB
Go
370 lines
13 KiB
Go
|
package mongodbatlas
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const clustersPath = "groups/%s/clusters"
|
|||
|
|
|
|||
|
|
// ClustersService is an interface for interfacing with the Clusters
|
|||
|
|
// endpoints of the MongoDB Atlas API.
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters/
|
|||
|
|
type ClustersService interface {
|
|||
|
|
List(context.Context, string, *ListOptions) ([]Cluster, *Response, error)
|
|||
|
|
Get(context.Context, string, string) (*Cluster, *Response, error)
|
|||
|
|
Create(context.Context, string, *Cluster) (*Cluster, *Response, error)
|
|||
|
|
Update(context.Context, string, string, *Cluster) (*Cluster, *Response, error)
|
|||
|
|
Delete(context.Context, string, string) (*Response, error)
|
|||
|
|
UpdateProcessArgs(context.Context, string, string, *ProcessArgs) (*ProcessArgs, *Response, error)
|
|||
|
|
GetProcessArgs(context.Context, string, string) (*ProcessArgs, *Response, error)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ClustersServiceOp handles communication with the Cluster related methods
|
|||
|
|
// of the MongoDB Atlas API
|
|||
|
|
type ClustersServiceOp service
|
|||
|
|
|
|||
|
|
var _ ClustersService = &ClustersServiceOp{}
|
|||
|
|
|
|||
|
|
// AutoScaling configures your cluster to automatically scale its storage
|
|||
|
|
type AutoScaling struct {
|
|||
|
|
DiskGBEnabled *bool `json:"diskGBEnabled,omitempty"`
|
|||
|
|
Compute *Compute `json:"compute,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Compute Specifies whether the cluster automatically scales its cluster tier and whether the cluster can scale down.
|
|||
|
|
type Compute struct {
|
|||
|
|
Enabled *bool `json:"enabled,omitempty"`
|
|||
|
|
ScaleDownEnabled *bool `json:"scaleDownEnabled,omitempty"`
|
|||
|
|
MinInstanceSize string `json:"minInstanceSize,omitempty"`
|
|||
|
|
MaxInstanceSize string `json:"maxInstanceSize,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BiConnector specifies BI Connector for Atlas configuration on this cluster
|
|||
|
|
type BiConnector struct {
|
|||
|
|
Enabled *bool `json:"enabled,omitempty"`
|
|||
|
|
ReadPreference string `json:"readPreference,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ProviderSettings configuration for the provisioned servers on which MongoDB runs. The available options are specific to the cloud service provider.
|
|||
|
|
type ProviderSettings struct {
|
|||
|
|
BackingProviderName string `json:"backingProviderName,omitempty"`
|
|||
|
|
DiskIOPS *int64 `json:"diskIOPS,omitempty"`
|
|||
|
|
DiskTypeName string `json:"diskTypeName,omitempty"`
|
|||
|
|
EncryptEBSVolume *bool `json:"encryptEBSVolume,omitempty"`
|
|||
|
|
InstanceSizeName string `json:"instanceSizeName,omitempty"`
|
|||
|
|
ProviderName string `json:"providerName,omitempty"`
|
|||
|
|
RegionName string `json:"regionName,omitempty"`
|
|||
|
|
VolumeType string `json:"volumeType,omitempty"`
|
|||
|
|
AutoScaling *AutoScaling `json:"autoScaling,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RegionsConfig describes the region’s priority in elections and the number and type of MongoDB nodes Atlas deploys to the region.
|
|||
|
|
type RegionsConfig struct {
|
|||
|
|
AnalyticsNodes *int64 `json:"analyticsNodes,omitempty"`
|
|||
|
|
ElectableNodes *int64 `json:"electableNodes,omitempty"`
|
|||
|
|
Priority *int64 `json:"priority,omitempty"`
|
|||
|
|
ReadOnlyNodes *int64 `json:"readOnlyNodes,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ReplicationSpec represents a configuration for cluster regions
|
|||
|
|
type ReplicationSpec struct {
|
|||
|
|
ID string `json:"id,omitempty"`
|
|||
|
|
NumShards *int64 `json:"numShards,omitempty"`
|
|||
|
|
ZoneName string `json:"zoneName,omitempty"`
|
|||
|
|
RegionsConfig map[string]RegionsConfig `json:"regionsConfig,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ConnectionStrings configuration for applications use to connect to this cluster
|
|||
|
|
type ConnectionStrings struct {
|
|||
|
|
Standard string `json:"standard,omitempty"`
|
|||
|
|
StandardSrv string `json:"standardSrv,omitempty"`
|
|||
|
|
AwsPrivateLink map[string]string `json:"awsPrivateLink,omitempty"`
|
|||
|
|
AwsPrivateLinkSrv map[string]string `json:"awsPrivateLinkSrv,omitempty"`
|
|||
|
|
Private string `json:"private,omitempty"`
|
|||
|
|
PrivateSrv string `json:"privateSrv,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Cluster represents MongoDB cluster.
|
|||
|
|
type Cluster struct {
|
|||
|
|
AutoScaling *AutoScaling `json:"autoScaling,omitempty"`
|
|||
|
|
BackupEnabled *bool `json:"backupEnabled,omitempty"`
|
|||
|
|
BiConnector *BiConnector `json:"biConnector,omitempty"`
|
|||
|
|
ClusterType string `json:"clusterType,omitempty"`
|
|||
|
|
DiskSizeGB *float64 `json:"diskSizeGB,omitempty"`
|
|||
|
|
EncryptionAtRestProvider string `json:"encryptionAtRestProvider,omitempty"`
|
|||
|
|
Labels []Label `json:"labels,omitempty"`
|
|||
|
|
ID string `json:"id,omitempty"`
|
|||
|
|
GroupID string `json:"groupId,omitempty"`
|
|||
|
|
MongoDBVersion string `json:"mongoDBVersion,omitempty"`
|
|||
|
|
MongoDBMajorVersion string `json:"mongoDBMajorVersion,omitempty"`
|
|||
|
|
MongoURI string `json:"mongoURI,omitempty"`
|
|||
|
|
MongoURIUpdated string `json:"mongoURIUpdated,omitempty"`
|
|||
|
|
MongoURIWithOptions string `json:"mongoURIWithOptions,omitempty"`
|
|||
|
|
Name string `json:"name,omitempty"`
|
|||
|
|
NumShards *int64 `json:"numShards,omitempty"`
|
|||
|
|
Paused *bool `json:"paused,omitempty"`
|
|||
|
|
PitEnabled *bool `json:"pitEnabled,omitempty"`
|
|||
|
|
ProviderBackupEnabled *bool `json:"providerBackupEnabled,omitempty"`
|
|||
|
|
ProviderSettings *ProviderSettings `json:"providerSettings,omitempty"`
|
|||
|
|
ReplicationFactor *int64 `json:"replicationFactor,omitempty"`
|
|||
|
|
ReplicationSpec map[string]RegionsConfig `json:"replicationSpec,omitempty"`
|
|||
|
|
ReplicationSpecs []ReplicationSpec `json:"replicationSpecs,omitempty"`
|
|||
|
|
SrvAddress string `json:"srvAddress,omitempty"`
|
|||
|
|
StateName string `json:"stateName,omitempty"`
|
|||
|
|
ConnectionStrings *ConnectionStrings `json:"connectionStrings,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ProcessArgs represents the advanced configuration options for the cluster
|
|||
|
|
type ProcessArgs struct {
|
|||
|
|
FailIndexKeyTooLong *bool `json:"failIndexKeyTooLong,omitempty"`
|
|||
|
|
JavascriptEnabled *bool `json:"javascriptEnabled,omitempty"`
|
|||
|
|
MinimumEnabledTLSProtocol string `json:"minimumEnabledTlsProtocol,omitempty"`
|
|||
|
|
NoTableScan *bool `json:"noTableScan,omitempty"`
|
|||
|
|
OplogSizeMB *int64 `json:"oplogSizeMB,omitempty"`
|
|||
|
|
SampleSizeBIConnector *int64 `json:"sampleSizeBIConnector,omitempty"`
|
|||
|
|
SampleRefreshIntervalBIConnector *int64 `json:"sampleRefreshIntervalBIConnector,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// clustersResponse is the response from the ClustersService.List.
|
|||
|
|
type clustersResponse struct {
|
|||
|
|
Links []*Link `json:"links,omitempty"`
|
|||
|
|
Results []Cluster `json:"results,omitempty"`
|
|||
|
|
TotalCount int `json:"totalCount,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DefaultDiskSizeGB represents the Tier and the default disk size for each one
|
|||
|
|
// it can be use like: DefaultDiskSizeGB["AWS"]["M10"]
|
|||
|
|
var DefaultDiskSizeGB = map[string]map[string]float64{
|
|||
|
|
"TENANT": {
|
|||
|
|
"M2": 2,
|
|||
|
|
"M5": 5,
|
|||
|
|
},
|
|||
|
|
"AWS": {
|
|||
|
|
"M10": 10,
|
|||
|
|
"M20": 20,
|
|||
|
|
"M30": 40,
|
|||
|
|
"M40": 80,
|
|||
|
|
"R40": 80,
|
|||
|
|
"M40_NVME": 380,
|
|||
|
|
"M50": 160,
|
|||
|
|
"R50": 160,
|
|||
|
|
"M50_NVME": 760,
|
|||
|
|
"M60": 320,
|
|||
|
|
"R60": 320,
|
|||
|
|
"M60_NVME": 1600,
|
|||
|
|
"M80": 750,
|
|||
|
|
"R80": 750,
|
|||
|
|
"M80_NVME": 1600,
|
|||
|
|
"M140": 1000,
|
|||
|
|
"M200": 1500,
|
|||
|
|
"R200": 1500,
|
|||
|
|
"M200_NVME": 3100,
|
|||
|
|
"M300": 2000,
|
|||
|
|
"R300": 2000,
|
|||
|
|
"R400": 3000,
|
|||
|
|
"M400_NVME": 4000,
|
|||
|
|
},
|
|||
|
|
"GCP": {
|
|||
|
|
"M10": 10,
|
|||
|
|
"M20": 20,
|
|||
|
|
"M30": 40,
|
|||
|
|
"M40": 80,
|
|||
|
|
"M50": 160,
|
|||
|
|
"M60": 320,
|
|||
|
|
"M80": 750,
|
|||
|
|
"M200": 1500,
|
|||
|
|
"M300": 2200,
|
|||
|
|
},
|
|||
|
|
"AZURE": {
|
|||
|
|
"M10": 32,
|
|||
|
|
"M20": 32,
|
|||
|
|
"M30": 32,
|
|||
|
|
"M40": 128,
|
|||
|
|
"M50": 128,
|
|||
|
|
"M60": 128,
|
|||
|
|
"M80": 256,
|
|||
|
|
"M200": 256,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// List all clusters in the project associated to {GROUP-ID}.
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-get-all/
|
|||
|
|
func (s *ClustersServiceOp) List(ctx context.Context, groupID string, listOptions *ListOptions) ([]Cluster, *Response, error) {
|
|||
|
|
path := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
|
|||
|
|
// Add query params from listOptions
|
|||
|
|
path, err := setListOptions(path, listOptions)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(clustersResponse)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if l := root.Links; l != nil {
|
|||
|
|
resp.Links = l
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root.Results, resp, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Get gets the cluster specified to {ClUSTER-NAME} from the project associated to {GROUP-ID}.
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-get-one/
|
|||
|
|
func (s *ClustersServiceOp) Get(ctx context.Context, groupID, clusterName string) (*Cluster, *Response, error) {
|
|||
|
|
if err := checkClusterNameParam(clusterName); err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
escapedEntry := url.PathEscape(clusterName)
|
|||
|
|
path := fmt.Sprintf("%s/%s", basePath, escapedEntry)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(Cluster)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Create adds a cluster to the project associated to {GROUP-ID}.
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-create-one/
|
|||
|
|
func (s *ClustersServiceOp) Create(ctx context.Context, groupID string, createRequest *Cluster) (*Cluster, *Response, error) {
|
|||
|
|
if createRequest == nil {
|
|||
|
|
return nil, nil, NewArgError("createRequest", "cannot be nil")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
path := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodPost, path, createRequest)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(Cluster)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Update a cluster in the project associated to {GROUP-ID}
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-modify-one/
|
|||
|
|
func (s *ClustersServiceOp) Update(ctx context.Context, groupID, clusterName string, updateRequest *Cluster) (*Cluster, *Response, error) {
|
|||
|
|
if updateRequest == nil {
|
|||
|
|
return nil, nil, NewArgError("updateRequest", "cannot be nil")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
path := fmt.Sprintf("%s/%s", basePath, clusterName)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodPatch, path, updateRequest)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(Cluster)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Delete the cluster specified to {CLUSTER-NAME} from the project associated to {GROUP-ID}.
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-delete-one/
|
|||
|
|
func (s *ClustersServiceOp) Delete(ctx context.Context, groupID, clusterName string) (*Response, error) {
|
|||
|
|
if clusterName == "" {
|
|||
|
|
return nil, NewArgError("clusterName", "must be set")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
escapedEntry := url.PathEscape(clusterName)
|
|||
|
|
path := fmt.Sprintf("%s/%s", basePath, escapedEntry)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodDelete, path, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp, err := s.Client.Do(ctx, req, nil)
|
|||
|
|
|
|||
|
|
return resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateProcessArgs Modifies Advanced Configuration Options for One Cluster
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-modify-advanced-configuration-options/
|
|||
|
|
func (s *ClustersServiceOp) UpdateProcessArgs(ctx context.Context, groupID, clusterName string, updateRequest *ProcessArgs) (*ProcessArgs, *Response, error) {
|
|||
|
|
if updateRequest == nil {
|
|||
|
|
return nil, nil, NewArgError("updateRequest", "cannot be nil")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
path := fmt.Sprintf("%s/%s/processArgs", basePath, clusterName)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodPatch, path, updateRequest)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(ProcessArgs)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetProcessArgs gets the Advanced Configuration Options for One Cluster
|
|||
|
|
// See more: https://docs.atlas.mongodb.com/reference/api/clusters-get-advanced-configuration-options/#get-advanced-configuration-options-for-one-cluster
|
|||
|
|
func (s *ClustersServiceOp) GetProcessArgs(ctx context.Context, groupID, clusterName string) (*ProcessArgs, *Response, error) {
|
|||
|
|
if err := checkClusterNameParam(clusterName); err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
basePath := fmt.Sprintf(clustersPath, groupID)
|
|||
|
|
escapedEntry := url.PathEscape(clusterName)
|
|||
|
|
path := fmt.Sprintf("%s/%s/processArgs", basePath, escapedEntry)
|
|||
|
|
|
|||
|
|
req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
root := new(ProcessArgs)
|
|||
|
|
resp, err := s.Client.Do(ctx, req, root)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return root, resp, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func checkClusterNameParam(clusterName string) error {
|
|||
|
|
if clusterName == "" {
|
|||
|
|
return NewArgError("name", "must be set")
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|