-
Notifications
You must be signed in to change notification settings - Fork 16
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(cannon): Add Blockprint Block Classification Deriving
- Loading branch information
Showing
28 changed files
with
3,097 additions
and
1,800 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| package blockprint | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
| "net/http" | ||
| ) | ||
|
|
||
| type Client struct { | ||
| endpoint string | ||
| httpClient *http.Client | ||
| headers map[string]string | ||
| } | ||
|
|
||
| func NewClient(endpoint string, headers map[string]string) *Client { | ||
| return &Client{ | ||
| endpoint: endpoint, | ||
| httpClient: http.DefaultClient, | ||
| headers: headers, | ||
| } | ||
| } | ||
|
|
||
| func (c *Client) get(ctx context.Context, path string) (json.RawMessage, error) { | ||
| req, err := http.NewRequestWithContext(ctx, "GET", c.endpoint+path, http.NoBody) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Set headers from c.headers | ||
| for k, v := range c.headers { | ||
| req.Header.Set(k, v) | ||
| } | ||
|
|
||
| rsp, err := c.httpClient.Do(req) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| defer rsp.Body.Close() | ||
|
|
||
| if rsp.StatusCode != http.StatusOK { | ||
| return nil, fmt.Errorf("status code: %d", rsp.StatusCode) | ||
| } | ||
|
|
||
| data, err := io.ReadAll(rsp.Body) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| data = bytes.TrimPrefix(data, []byte("\xef\xbb\xbf")) | ||
|
|
||
| resp := new(json.RawMessage) | ||
| if err := json.Unmarshal(data, &resp); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return *resp, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,52 @@ | ||
| package blockprint | ||
|
|
||
| type ProbabilityMap map[ClientName]float64 | ||
|
|
||
| type ClientName string | ||
|
|
||
| var ( | ||
| ClientNameUnknown = ClientName("Unknown") | ||
| ClientNameUncertain = ClientName("Uncertain") | ||
| ClientNamePrysm = ClientName("Prysm") | ||
| ClientNameLighthouse = ClientName("Lighthouse") | ||
| ClientNameLodestar = ClientName("Lodestar") | ||
| ClientNameNimbus = ClientName("Nimbus") | ||
| ClientNameTeku = ClientName("Teku") | ||
| ClientNameGrandine = ClientName("Teku") | ||
| ) | ||
|
|
||
| func (p ProbabilityMap) safeGet(name ClientName) float64 { | ||
| if p[name] == 0 { | ||
| return 0 | ||
| } | ||
|
|
||
| return p[name] | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Prysm() float64 { | ||
| return p.safeGet(ClientNamePrysm) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Lighthouse() float64 { | ||
| return p.safeGet(ClientNameLighthouse) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Lodestar() float64 { | ||
| return p.safeGet(ClientNameLodestar) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Nimbus() float64 { | ||
| return p.safeGet(ClientNameNimbus) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Teku() float64 { | ||
| return p.safeGet(ClientNameTeku) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Grandine() float64 { | ||
| return p.safeGet(ClientNameGrandine) | ||
| } | ||
|
|
||
| func (p ProbabilityMap) Uncertain() float64 { | ||
| return p.safeGet(ClientNameUncertain) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,44 @@ | ||
| package blockprint | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
|
|
||
| "github.com/attestantio/go-eth2-client/spec/phase0" | ||
| ) | ||
|
|
||
| type ProposersBlocksResponse struct { | ||
| //nolint:tagliatelle // Defined by API. | ||
| ProposerIndex uint64 `json:"proposer_index"` | ||
| Slot uint64 `json:"slot"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| BestGuessSingle ClientName `json:"best_guess_single"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| BestGuessMulti string `json:"best_guess_multi"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| ProbabilityMap *ProbabilityMap `json:"probability_map"` | ||
| } | ||
|
|
||
| func (c *Client) BlocksRange(ctx context.Context, startSlot phase0.Slot, endSlot ...phase0.Slot) ([]*ProposersBlocksResponse, error) { | ||
| var path string | ||
| if len(endSlot) > 0 { | ||
| path = fmt.Sprintf("/blocks/%d/%d", startSlot, endSlot[0]) | ||
| } else { | ||
| path = fmt.Sprintf("/blocks/%d", startSlot) | ||
| } | ||
|
|
||
| data, err := c.get(ctx, path) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| var result []*ProposersBlocksResponse | ||
|
|
||
| err = json.Unmarshal(data, &result) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return result, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| package blockprint | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "net/http" | ||
| ) | ||
|
|
||
| type BlocksPerClientResponse struct { | ||
| //nolint:tagliatelle // Defined by API. | ||
| Uncertain uint64 `json:"Uncertain"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Lighthouse uint64 `json:"Lighthouse"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Lodestar uint64 `json:"Lodestar"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Nimbus uint64 `json:"Nimbus"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Other uint64 `json:"Other"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Prysm uint64 `json:"Prysm"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Teku uint64 `json:"Teku"` | ||
| //nolint:tagliatelle // Defined by API. | ||
| Grandine uint64 `json:"Grandine"` | ||
| } | ||
|
|
||
| type SyncStatusResponse struct { | ||
| //nolint:tagliatelle // Defined by API. | ||
| GreatestBlockSlot uint64 `json:"greatest_block_slot"` | ||
| Synced bool `json:"synced"` | ||
| } | ||
|
|
||
| func (c *Client) BlocksPerClient(ctx context.Context, startEpoch, endEpoch string) (*BlocksPerClientResponse, error) { | ||
| req, err := http.NewRequestWithContext(ctx, "GET", c.endpoint+"/blocks_per_client/"+startEpoch+"/"+endEpoch, http.NoBody) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| resp, err := c.httpClient.Do(req) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| defer resp.Body.Close() | ||
|
|
||
| var result BlocksPerClientResponse | ||
|
|
||
| err = json.NewDecoder(resp.Body).Decode(&result) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return &result, nil | ||
| } | ||
|
|
||
| func (c *Client) SyncStatus(ctx context.Context) (*SyncStatusResponse, error) { | ||
| data, err := c.get(ctx, "/sync/status") | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| var result SyncStatusResponse | ||
|
|
||
| err = json.Unmarshal(data, &result) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return &result, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.