feat: some more alias to atlas client types (#217)

mongodb · Nov 7, 2023 · c3bc956 · c3bc956
1 parent c7d192c
commit c3bc956
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/opsmngr/live_data_migration.go b/opsmngr/live_data_migration.go
@@ -24,12 +24,14 @@ import (
 
 const liveMigrationBasePath = "api/public/v1.0/orgs/%s/liveExport/migrationLink"
 
+type LinkToken = atlas.LinkToken
+
 // LiveDataMigrationService is an interface for interfacing with the Live Migration
 // endpoints of the MongoDB Ops Manager API.
 //
 // See more: https://docs.opsmanager.mongodb.com/current/reference/api/cloud-migration/
 type LiveDataMigrationService interface {
-	ConnectOrganizations(context.Context, string, *atlas.LinkToken) (*ConnectionStatus, *Response, error)
+	ConnectOrganizations(context.Context, string, *LinkToken) (*ConnectionStatus, *Response, error)
 	DeleteConnection(context.Context, string) (*Response, error)
 	ConnectionStatus(context.Context, string) (*ConnectionStatus, *Response, error)
 }
@@ -69,7 +71,7 @@ func (s *LiveDataMigrationServiceOp) ConnectionStatus(ctx context.Context, orgID
 // ConnectOrganizations connects the source Ops Manager organization with a target MongoDB Atlas organization.
 //
 // See more: https://docs.opsmanager.mongodb.com/current/reference/api/cloud-migration/link-the-organization-with-atlas/
-func (s *LiveDataMigrationServiceOp) ConnectOrganizations(ctx context.Context, orgID string, linkToken *atlas.LinkToken) (*ConnectionStatus, *Response, error) {
+func (s *LiveDataMigrationServiceOp) ConnectOrganizations(ctx context.Context, orgID string, linkToken *LinkToken) (*ConnectionStatus, *Response, error) {
 	if orgID == "" {
 		return nil, nil, atlas.NewArgError("orgID", "must be set")
 	}

diff --git a/opsmngr/measurements.go b/opsmngr/measurements.go
@@ -25,6 +25,7 @@ type (
 	ProcessDiskMeasurements       = atlas.ProcessDiskMeasurements
 	ProcessDatabaseMeasurements   = atlas.ProcessDatabaseMeasurements
 	ProcessMeasurementListOptions = atlas.ProcessMeasurementListOptions
+	Measurements                  = atlas.Measurements
 )
 
 // MeasurementsService provides access to the measurement related functions in the Ops Manager API.

diff --git a/opsmngr/opsmngr.go b/opsmngr/opsmngr.go
@@ -52,6 +52,8 @@ type (
 	Checkpoint                = atlas.Checkpoint
 	Checkpoints               = atlas.Checkpoints
 	SnapshotTimestamp         = atlas.SnapshotTimestamp
+	IndexOptions              = atlas.IndexOptions
+	CollationOptions          = atlas.CollationOptions
 )
 
 type HTTPClient interface {