/
agents.go
113 lines (96 loc) · 3.55 KB
/
agents.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// Copyright 2020 MongoDB Inc
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package opsmngr
import (
"context"
"fmt"
"net/http"
atlas "go.mongodb.org/atlas/mongodbatlas"
)
const (
agentsBasePath = "groups/%s/agents"
)
// AgentsService provides access to the agent related functions in the Ops Manager API.
//
// See more: https://docs.opsmanager.mongodb.com/current/reference/api/agents/
type AgentsService interface {
ListAgentLinks(context.Context, string) (*Agents, *atlas.Response, error)
ListAgentsByType(context.Context, string, string) (*Agents, *atlas.Response, error)
CreateAgentAPIKey(context.Context, string, *AgentAPIKeysRequest) (*AgentAPIKey, *atlas.Response, error)
ListAgentAPIKeys(context.Context, string) ([]*AgentAPIKey, *atlas.Response, error)
DeleteAgentAPIKey(context.Context, string, string) (*atlas.Response, error)
}
// AgentsServiceOp provides an implementation of the AgentsService interface
type AgentsServiceOp service
var _ AgentsService = new(AgentsServiceOp)
// Agent represents an Ops Manager agent
type Agent struct {
TypeName string `json:"typeName"`
Hostname string `json:"hostname"`
ConfCount int64 `json:"confCount"`
LastConf string `json:"lastConf"`
StateName string `json:"stateName"`
PingCount int64 `json:"pingCount"`
IsManaged bool `json:"isManaged"`
LastPing string `json:"lastPing"`
Tag *string `json:"tag"`
}
// Agents is a paginated collection of Agent
type Agents struct {
Links []*atlas.Link `json:"links"`
Results []*Agent `json:"results"`
TotalCount int `json:"totalCount"`
}
// ListAgentLinks gets links to monitoring, backup, and automation agent resources for a project.
//
// See more: https://docs.opsmanager.mongodb.com/current/reference/api/agents-get-all/
func (s *AgentsServiceOp) ListAgentLinks(ctx context.Context, groupID string) (*Agents, *atlas.Response, error) {
if groupID == "" {
return nil, nil, atlas.NewArgError("groupID", "must be set")
}
path := fmt.Sprintf(agentsBasePath, groupID)
req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
if err != nil {
return nil, nil, err
}
root := new(Agents)
resp, err := s.Client.Do(ctx, req, root)
if err != nil {
return nil, resp, err
}
return root, resp, err
}
// List gets agents of a specified type (i.e. Monitoring, Backup, or Automation) for a project.
//
// See more: https://docs.opsmanager.mongodb.com/current/reference/api/agents-get-by-type/
func (s *AgentsServiceOp) ListAgentsByType(ctx context.Context, groupID, agentType string) (*Agents, *atlas.Response, error) {
if groupID == "" {
return nil, nil, atlas.NewArgError("groupID", "must be set")
}
if agentType == "" {
return nil, nil, atlas.NewArgError("agentType", "must be set")
}
basePath := fmt.Sprintf(agentsBasePath, groupID)
path := fmt.Sprintf("%s/%s", basePath, agentType)
req, err := s.Client.NewRequest(ctx, http.MethodGet, path, nil)
if err != nil {
return nil, nil, err
}
root := new(Agents)
resp, err := s.Client.Do(ctx, req, root)
if err != nil {
return nil, resp, err
}
return root, resp, err
}