DataHoarder
30623dbd55
All checks were successful
continuous-integration/drone/push Build is passing
130 lines
2.5 KiB
Go
130 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"git.gammaspectra.live/P2Pool/p2pool-observer/utils"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
)
|
|
|
|
var apiUrl string
|
|
|
|
func getTypeFromAPI[T any](method string) *T {
|
|
uri, _ := url.Parse(apiUrl + method)
|
|
if response, err := http.DefaultClient.Do(&http.Request{
|
|
Method: "GET",
|
|
URL: uri,
|
|
}); err != nil {
|
|
return nil
|
|
} else {
|
|
defer response.Body.Close()
|
|
defer io.ReadAll(response.Body)
|
|
if response.StatusCode == http.StatusOK {
|
|
var result T
|
|
decoder := utils.NewJSONDecoder(response.Body)
|
|
if decoder.Decode(&result) != nil {
|
|
return nil
|
|
} else {
|
|
return &result
|
|
}
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func getSliceFromAPI[T any](method string) []T {
|
|
|
|
uri, _ := url.Parse(apiUrl + method)
|
|
if response, err := http.DefaultClient.Do(&http.Request{
|
|
Method: "GET",
|
|
URL: uri,
|
|
}); err != nil {
|
|
return nil
|
|
} else {
|
|
defer response.Body.Close()
|
|
defer io.ReadAll(response.Body)
|
|
if response.StatusCode == http.StatusOK {
|
|
var result []T
|
|
decoder := utils.NewJSONDecoder(response.Body)
|
|
if decoder.Decode(&result) != nil {
|
|
return nil
|
|
} else {
|
|
return result
|
|
}
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func getStreamFromAPI[T any](method string) <-chan T {
|
|
result := make(chan T, 1)
|
|
|
|
go func() {
|
|
defer close(result)
|
|
uri, _ := url.Parse(apiUrl + method)
|
|
if response, err := http.DefaultClient.Do(&http.Request{
|
|
Method: "GET",
|
|
URL: uri,
|
|
}); err != nil {
|
|
return
|
|
} else {
|
|
defer response.Body.Close()
|
|
defer io.ReadAll(response.Body)
|
|
|
|
if response.StatusCode == http.StatusOK {
|
|
|
|
var err error
|
|
|
|
// Read opening
|
|
var b [1]byte
|
|
for {
|
|
if _, err = response.Body.Read(b[:]); err != nil {
|
|
return
|
|
}
|
|
if b[0] == '[' {
|
|
break
|
|
} else if b[0] != ' ' && b[0] != 0xa {
|
|
return
|
|
}
|
|
}
|
|
|
|
decoder := utils.NewJSONDecoder(response.Body)
|
|
for decoder.More() {
|
|
var item T
|
|
if err := decoder.Decode(&item); err != nil {
|
|
return
|
|
} else {
|
|
result <- item
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
return result
|
|
}
|
|
|
|
func getFromAPIRaw(method string) []byte {
|
|
uri, _ := url.Parse(apiUrl + method)
|
|
if response, err := http.DefaultClient.Do(&http.Request{
|
|
Method: "GET",
|
|
URL: uri,
|
|
}); err != nil {
|
|
return nil
|
|
} else {
|
|
defer response.Body.Close()
|
|
defer io.ReadAll(response.Body)
|
|
if response.StatusCode == http.StatusOK {
|
|
if data, err := io.ReadAll(response.Body); err != nil {
|
|
return nil
|
|
} else {
|
|
return data
|
|
}
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
}
|