consensus/cmd/apitocache/api.go
DataHoarder 30623dbd55
All checks were successful
continuous-integration/drone/push Build is passing
Added cmd/apitocache tool
2023-11-04 22:34:42 +01:00

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
}
}
}