forked from cloudfoundry-attic/bosh-init
-
Notifications
You must be signed in to change notification settings - Fork 0
/
agent_request.go
62 lines (50 loc) · 1.47 KB
/
agent_request.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
package http
import (
"encoding/json"
"io/ioutil"
"net/http"
bihttpclient "github.com/cloudfoundry/bosh-init/deployment/httpclient"
bosherr "github.com/cloudfoundry/bosh-utils/errors"
)
type AgentRequestMessage struct {
Method string `json:"method"`
Arguments []interface{} `json:"arguments"`
ReplyTo string `json:"reply_to"`
}
type agentRequest struct {
directorID string
endpoint string
httpClient bihttpclient.HTTPClient
}
func (r agentRequest) Send(method string, arguments []interface{}, response Response) error {
postBody := AgentRequestMessage{
Method: method,
Arguments: arguments,
ReplyTo: r.directorID,
}
agentRequestJSON, err := json.Marshal(postBody)
if err != nil {
return bosherr.WrapError(err, "Marshaling agent request")
}
httpResponse, err := r.httpClient.Post(r.endpoint, agentRequestJSON)
if err != nil {
return bosherr.WrapErrorf(err, "Performing request to agent endpoint '%s'", r.endpoint)
}
defer httpResponse.Body.Close()
if httpResponse.StatusCode != http.StatusOK {
return bosherr.Errorf("Agent responded with non-successful status code: %d", httpResponse.StatusCode)
}
responseBody, err := ioutil.ReadAll(httpResponse.Body)
if err != nil {
return bosherr.WrapError(err, "Reading agent response")
}
err = response.Unmarshal(responseBody)
if err != nil {
return bosherr.WrapError(err, "Unmarshaling agent response")
}
err = response.ServerError()
if err != nil {
return err
}
return nil
}