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
|
package easyhttp
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"log"
"net/http"
"sync"
)
func Get(ctx context.Context, uri, apiKey string) ([]byte, error) {
var (
client = &http.Client{}
bytes []byte
)
req, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
if err != nil {
return bytes, fmt.Errorf("%s: %w", uri, err)
}
req.Header.Set("X-API-KEY", apiKey)
resp, err := client.Do(req)
if err != nil {
return bytes, fmt.Errorf("%s: %w", uri, err)
}
defer resp.Body.Close()
bytes, err = io.ReadAll(resp.Body)
if err != nil {
return bytes, fmt.Errorf("%s: %w", uri, err)
}
return bytes, nil
}
// Get data from JSON
func GetData[T any](ctx context.Context, uri, apiKey string, data *T) error {
bytes, err := Get(ctx, uri, apiKey)
if err != nil {
return err
}
return json.Unmarshal(bytes, data)
}
func Post(ctx context.Context, uri, apiKey string, data []byte) ([]byte, error) {
req, err := http.NewRequestWithContext(ctx, http.MethodPost, uri, bytes.NewBuffer(data))
if err != nil {
return []byte{}, fmt.Errorf("%s: %w", uri, err)
}
req.Header.Set("Content-Type", "application/json")
req.Header.Set("X-API-KEY", apiKey)
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return []byte{}, fmt.Errorf("%s: %w", uri, err)
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return []byte{}, fmt.Errorf("%s: %w", uri, err)
}
switch resp.StatusCode {
case 200:
return body, nil
case 401:
return body, fmt.Errorf("unauthorized, API key configured?")
default:
return body, fmt.Errorf("unexpected HTTP response code %d", resp.StatusCode)
}
}
// Submit structure as JSON to API
func PostData[T any](ctx context.Context, uri, apiKey string, data *T, servers ...string) error {
if len(servers) == 0 {
return fmt.Errorf("no server configured")
}
var errs safErrors
var wg sync.WaitGroup
for _, server := range servers {
log.Println("Submitting data to", server)
wg.Add(1)
go func(server string) {
defer wg.Done()
errs.Append(postData[T](ctx, fmt.Sprintf("%s/%s", server, uri), apiKey, data))
}(server)
}
wg.Wait()
return errs.Join()
}
func postData[T any](ctx context.Context, uri, apiKey string, data *T) error {
jsonData, err := json.Marshal(data)
if err != nil {
return err
}
_, err = Post(ctx, uri, apiKey, jsonData)
return err
}
|
