CLOUDP-58633: Add Restore Jobs support to atlas client

mongodbatlas/continuous_backup_restore_job.go

@@ -0,0 +1,165 @@
+package mongodbatlas
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+)
+
+const continuousBackupRestoreJobsPath = "/groups/%s/clusters/%s/restoreJobs"
+
+type ContinuousBackupRestoreJobsService interface {
+	List(context.Context, string, string, *ListOptions) (*ContinuousBackupJobs, *Response, error)
+	Get(context.Context, string, string, string) (*ContinuousBackupJob, *Response, error)
+	Create(context.Context, string, string, *JobRequest) (*ContinuousBackupJob, *Response, error)
+}
+
+// ContinuousBackupRestoreJobsServiceOp handles communication with the Continuous Backup Restore Jobs related methods
+// of the MongoDB Atlas API
+type ContinuousBackupRestoreJobsServiceOp struct {
+	Client RequestDoer
+}
+
+var _ ContinuousBackupRestoreJobsService = &ContinuousBackupRestoreJobsServiceOp{}
+
+type ContinuousBackupJob struct {
+	BatchID           string            `json:"batchId,omitempty"`
+	ClusterID         string            `json:"clusterId,omitempty"`
+	Created           string            `json:"created"`
+	ClusterName       string            `json:"clusterName,omitempty"`
+	Delivery          *Delivery         `json:"delivery,omitempty"`
+	EncryptionEnabled bool              `json:"encryptionEnabled"`
+	GroupID           string            `json:"groupId"`
+	Hashes            []*Hash           `json:"hashes,omitempty"`
+	ID                string            `json:"id"`
+	Links             []*Link           `json:"links,omitempty"`
+	MasterKeyUUID     string            `json:"masterKeyUUID,omitempty"`
+	SnapshotID        string            `json:"snapshotId"`
+	StatusName        string            `json:"statusName"`
+	PointInTime       *bool             `json:"pointInTime,omitempty"`
+	Timestamp         SnapshotTimestamp `json:"timestamp"`
+}
+
+type ContinuousBackupJobs struct {
+	Results    []*ContinuousBackupJob `json:"results,omitempty"`
+	Links      []*Link                `json:"links,omitempty"`
+	TotalCount int64                  `json:"totalCount,omitempty"`
+}
+
+type SnapshotTimestamp struct {
+	Date      string `json:"date"`
+	Increment int64  `json:"increment"`
+}
+
+type Delivery struct {
+	Expires           string `json:"expires,omitempty"`
+	ExpirationHours   int64  `json:"expirationHours,omitempty"`
+	MaxDownloads      int64  `json:"maxDownloads,omitempty"`
+	MethodName        string `json:"methodName"`
+	StatusName        string `json:"statusName,omitempty"`
+	URL               string `json:"url,omitempty"`
+	TargetClusterID   string `json:"targetClusterId,omitempty"`
+	TargetClusterName string `json:"targetClusterName,omitempty"`
+	TargetGroupID     string `json:"targetGroupId,omitempty"`
+}
+
+type Hash struct {
+	TypeName string `json:"typeName"`
+	FileName string `json:"fileName"`
+	Hash     string `json:"hash"`
+}
+
+type JobRequest struct {
+	CheckPointID         string   `json:"checkPointId,omitempty"`
+	Delivery             Delivery `json:"delivery"`
+	OplogTs              string   `json:"oplogTs,omitempty"`
+	OplogInc             string   `json:"oplogInc,omitempty"`
+	PointInTimeUTCMillis float64  `json:"pointInTimeUTCMillis,omitempty"`
+	SnapshotID           string   `json:"snapshotId,omitempty"`
+}
+
+// List list all continuous backup jobs in Atlas
+// See more: https://docs.atlas.mongodb.com/reference/api/restore-jobs-get-all/
+func (s *ContinuousBackupRestoreJobsServiceOp) List(ctx context.Context, groupID, clusterID string, opts *ListOptions) (*ContinuousBackupJobs, *Response, error) {
+
+	if clusterID == "" {
+		return nil, nil, NewArgError("clusterID", "must be set")
+	}
+	if groupID == "" {
+		return nil, nil, NewArgError("groupID", "must be set")
+	}
+
+	path := fmt.Sprintf(continuousBackupRestoreJobsPath, groupID, clusterID)
+
+	path, err := setListOptions(path, opts)
+	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(ContinuousBackupJobs)
+	resp, err := s.Client.Do(ctx, req, root)
+
+	return root, resp, err
+}
+
+// Get gets a continuous backup job in Atlas
+// See more: https://docs.atlas.mongodb.com/reference/api/restore-jobs-get-one/
+func (s *ContinuousBackupRestoreJobsServiceOp) Get(ctx context.Context, groupID, clusterID, jobID string) (*ContinuousBackupJob, *Response, error) {
+
+	if clusterID == "" {
+		return nil, nil, NewArgError("clusterID", "must be set")
+	}
+	if groupID == "" {
+		return nil, nil, NewArgError("groupID", "must be set")
+	}
+	if jobID == "" {
+		return nil, nil, NewArgError("jobID", "must be set")
+	}
+	defaultPath := fmt.Sprintf(continuousBackupRestoreJobsPath, groupID, clusterID)
+
+	path := fmt.Sprintf("%s/%s", defaultPath, jobID)
+
+	req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
+
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(ContinuousBackupJob)
+	resp, err := s.Client.Do(ctx, req, root)
+
+	return root, resp, err
+}
+
+// Create creates a continuous backup job in Atlas
+// See more: https://docs.atlas.mongodb.com/reference/api/restore-jobs-create-one/
+func (s *ContinuousBackupRestoreJobsServiceOp) Create(ctx context.Context, groupID, clusterID string, request *JobRequest) (*ContinuousBackupJob, *Response, error) {
+
+	if request == nil {
+		return nil, nil, NewArgError("request", "must be set")
+	}
+	if clusterID == "" {
+		return nil, nil, NewArgError("clusterID", "must be set")
+	}
+	if groupID == "" {
+		return nil, nil, NewArgError("groupID", "must be set")
+	}
+
+	path := fmt.Sprintf(continuousBackupRestoreJobsPath, groupID, clusterID)
+
+	req, err := s.Client.NewRequest(ctx, http.MethodPost, path, request)
+
+	if err != nil {
+		return nil, nil, err
+	}
+
+	root := new(ContinuousBackupJob)
+	resp, err := s.Client.Do(ctx, req, root)
+
+	return root, resp, err
+}