@@ -19,6 +19,7 @@ import (
1919 "fmt"
2020
2121 atlas "github.com/mongodb/go-client-mongodb-atlas/mongodbatlas"
22+ om "github.com/mongodb/go-client-mongodb-ops-manager/opsmngr"
2223 "github.com/mongodb/mongocli/internal/config"
2324)
2425
@@ -46,6 +47,9 @@ func (s *Store) Alert(projectID, alertID string) (*atlas.Alert, error) {
4647 case config .CloudService :
4748 result , _ , err := s .client .(* atlas.Client ).Alerts .Get (context .Background (), projectID , alertID )
4849 return result , err
50+ case config .OpsManagerService , config .CloudManagerService :
51+ result , _ , err := s .client .(* om.Client ).Alerts .Get (context .Background (), projectID , alertID )
52+ return result , err
4953 default :
5054 return nil , fmt .Errorf ("unsupported service: %s" , s .service )
5155 }
@@ -57,6 +61,9 @@ func (s *Store) Alerts(projectID string, opts *atlas.ListOptions) ([]atlas.Alert
5761 case config .CloudService :
5862 result , _ , err := s .client .(* atlas.Client ).Alerts .List (context .Background (), projectID , opts )
5963 return result , err
64+ case config .OpsManagerService , config .CloudManagerService :
65+ result , _ , err := s .client .(* om.Client ).Alerts .List (context .Background (), projectID , opts )
66+ return result , err
6067 default :
6168 return nil , fmt .Errorf ("unsupported service: %s" , s .service )
6269 }
@@ -68,6 +75,9 @@ func (s *Store) AcknowledgeAlert(projectID, alertID string, body *atlas.Acknowle
6875 case config .CloudService :
6976 result , _ , err := s .client .(* atlas.Client ).Alerts .Acknowledge (context .Background (), projectID , alertID , body )
7077 return result , err
78+ case config .OpsManagerService , config .CloudManagerService :
79+ result , _ , err := s .client .(* om.Client ).Alerts .Acknowledge (context .Background (), projectID , alertID , body )
80+ return result , err
7181 default :
7282 return nil , fmt .Errorf ("unsupported service: %s" , s .service )
7383 }
0 commit comments