mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-26 14:46:42 +00:00
2fd85b3717
* eth, consensus: refactor whitelisting related logs and error handling * core: fix lint * eth: fix tests, check against root error * eth: use ctx correctly while logging * eth: refactor comments and error message
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
package heimdallapp
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/consensus/bor/heimdall/milestone"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
|
|
chTypes "github.com/maticnetwork/heimdall/checkpoint/types"
|
|
hmTypes "github.com/maticnetwork/heimdall/types"
|
|
)
|
|
|
|
func (h *HeimdallAppClient) FetchMilestoneCount(_ context.Context) (int64, error) {
|
|
log.Debug("Fetching milestone count")
|
|
|
|
res := h.hApp.CheckpointKeeper.GetMilestoneCount(h.NewContext())
|
|
|
|
log.Debug("Fetched Milestone Count", "res", int64(res))
|
|
|
|
return int64(res), nil
|
|
}
|
|
|
|
func (h *HeimdallAppClient) FetchMilestone(_ context.Context) (*milestone.Milestone, error) {
|
|
log.Debug("Fetching Latest Milestone")
|
|
|
|
res, err := h.hApp.CheckpointKeeper.GetLastMilestone(h.NewContext())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
milestone := toBorMilestone(res)
|
|
log.Debug("Fetched Latest Milestone", "milestone", milestone)
|
|
|
|
return milestone, nil
|
|
}
|
|
|
|
func (h *HeimdallAppClient) FetchNoAckMilestone(_ context.Context, milestoneID string) error {
|
|
log.Debug("Fetching No Ack Milestone By MilestoneID", "MilestoneID", milestoneID)
|
|
|
|
res := h.hApp.CheckpointKeeper.GetNoAckMilestone(h.NewContext(), milestoneID)
|
|
if res {
|
|
log.Info("Fetched No Ack By MilestoneID", "MilestoneID", milestoneID)
|
|
return nil
|
|
}
|
|
|
|
return fmt.Errorf("still no-ack milestone exist corresponding to milestoneID: %v", milestoneID)
|
|
}
|
|
|
|
func (h *HeimdallAppClient) FetchLastNoAckMilestone(_ context.Context) (string, error) {
|
|
log.Debug("Fetching Latest No Ack Milestone ID")
|
|
|
|
res := h.hApp.CheckpointKeeper.GetLastNoAckMilestone(h.NewContext())
|
|
|
|
log.Debug("Fetched Latest No Ack Milestone ID", "res", res)
|
|
|
|
return res, nil
|
|
}
|
|
|
|
func (h *HeimdallAppClient) FetchMilestoneID(_ context.Context, milestoneID string) error {
|
|
log.Debug("Fetching Milestone ID ", "MilestoneID", milestoneID)
|
|
|
|
res := chTypes.GetMilestoneID()
|
|
|
|
if res == milestoneID {
|
|
return nil
|
|
}
|
|
|
|
return fmt.Errorf("milestone corresponding to milestoneID: %v doesn't exist in heimdall", milestoneID)
|
|
}
|
|
|
|
func toBorMilestone(hdMilestone *hmTypes.Milestone) *milestone.Milestone {
|
|
return &milestone.Milestone{
|
|
Proposer: hdMilestone.Proposer.EthAddress(),
|
|
StartBlock: big.NewInt(int64(hdMilestone.StartBlock)),
|
|
EndBlock: big.NewInt(int64(hdMilestone.EndBlock)),
|
|
Hash: hdMilestone.Hash.EthHash(),
|
|
BorChainID: hdMilestone.BorChainID,
|
|
Timestamp: hdMilestone.TimeStamp,
|
|
}
|
|
}
|