-
Notifications
You must be signed in to change notification settings - Fork 5
/
util.go
124 lines (114 loc) · 3.24 KB
/
util.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
114
115
116
117
118
119
120
121
122
123
124
package beaconapi_evaluators
import (
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
log "github.com/sirupsen/logrus"
"github.com/theQRL/qrysm/v4/testing/endtoend/params"
)
func doMiddlewareJSONGetRequest(template string, requestPath string, beaconNodeIdx int, dst interface{}, bnType ...string) error {
var port int
if len(bnType) > 0 {
switch bnType[0] {
case "lighthouse":
port = params.TestParams.Ports.LighthouseBeaconNodeHTTPPort
default:
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
} else {
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
basePath := fmt.Sprintf(template, port+beaconNodeIdx)
httpResp, err := http.Get(
basePath + requestPath,
)
if err != nil {
return err
}
if httpResp.StatusCode != http.StatusOK {
var body interface{}
if err := json.NewDecoder(httpResp.Body).Decode(&body); err != nil {
return err
}
return fmt.Errorf("request failed with response code: %d with response body %s", httpResp.StatusCode, body)
}
return json.NewDecoder(httpResp.Body).Decode(&dst)
}
func doMiddlewareSSZGetRequest(template string, requestPath string, beaconNodeIdx int, bnType ...string) ([]byte, error) {
client := &http.Client{}
var port int
if len(bnType) > 0 {
switch bnType[0] {
case "lighthouse":
port = params.TestParams.Ports.LighthouseBeaconNodeHTTPPort
default:
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
} else {
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
basePath := fmt.Sprintf(template, port+beaconNodeIdx)
req, err := http.NewRequest("GET", basePath+requestPath, nil)
if err != nil {
return nil, err
}
req.Header.Set("Accept", "application/octet-stream")
rsp, err := client.Do(req)
if err != nil {
return nil, err
}
if rsp.StatusCode != http.StatusOK {
var body interface{}
if err := json.NewDecoder(rsp.Body).Decode(&body); err != nil {
return nil, err
}
return nil, fmt.Errorf("request failed with response code: %d with response body %s", rsp.StatusCode, body)
}
defer closeBody(rsp.Body)
body, err := io.ReadAll(rsp.Body)
if err != nil {
return nil, err
}
return body, nil
}
func doMiddlewareJSONPostRequest(template string, requestPath string, beaconNodeIdx int, postData, dst interface{}, bnType ...string) error {
var port int
if len(bnType) > 0 {
switch bnType[0] {
case "lighthouse":
port = params.TestParams.Ports.LighthouseBeaconNodeHTTPPort
default:
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
} else {
port = params.TestParams.Ports.PrysmBeaconNodeGatewayPort
}
basePath := fmt.Sprintf(template, port+beaconNodeIdx)
b, err := json.Marshal(postData)
if err != nil {
return err
}
httpResp, err := http.Post(
basePath+requestPath,
"application/json",
bytes.NewBuffer(b),
)
if err != nil {
return err
}
if httpResp.StatusCode != http.StatusOK {
var body interface{}
if err := json.NewDecoder(httpResp.Body).Decode(&body); err != nil {
return err
}
return fmt.Errorf("request failed with response code: %d with response body %s", httpResp.StatusCode, body)
}
return json.NewDecoder(httpResp.Body).Decode(&dst)
}
func closeBody(body io.Closer) {
if err := body.Close(); err != nil {
log.WithError(err).Error("could not close response body")
}
}