mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
961a012502
* define ComponentRunner * update ETH1 node * extrace BootNode * add logger * remove refs to t * gazelle * update BeaconNode * parametrize tests * update log formatting * update ValidatorNode * remove redundant test id from params * decrease delta * move BootNode * update BootNode * revert params * update SlasherNode * mask unused param * update helpers * update test runner * go mod tidy * remove unused fields * re-arrange tests * extract sync into its own method * gazelle * BeaconNode * types/types.go * BootNode * gofmt * remove unused argument * remove redundant comment * add deprecation comment * types comment * remove deprecated method * BeaconNodes -> BeaconNodeSet * make sure that slasher is required component * update StartBeaconNodeSet() and StartValidatorNodeSet() * update SlasherNodeSet * gazelle
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package components
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/bazelbuild/rules_go/go/tools/bazel"
|
|
"github.com/prysmaticlabs/prysm/endtoend/helpers"
|
|
e2e "github.com/prysmaticlabs/prysm/endtoend/params"
|
|
e2etypes "github.com/prysmaticlabs/prysm/endtoend/types"
|
|
)
|
|
|
|
var _ e2etypes.ComponentRunner = (*BootNode)(nil)
|
|
|
|
// BootNode represents boot node.
|
|
type BootNode struct {
|
|
e2etypes.ComponentRunner
|
|
started chan struct{}
|
|
enr string
|
|
}
|
|
|
|
// NewBootNode creates and returns boot node.
|
|
func NewBootNode() *BootNode {
|
|
return &BootNode{
|
|
started: make(chan struct{}, 1),
|
|
}
|
|
}
|
|
|
|
// ENR exposes node's ENR.
|
|
func (node *BootNode) ENR() string {
|
|
return node.enr
|
|
}
|
|
|
|
// StartBootnode starts a bootnode blocks up until ctx is cancelled.
|
|
func (node *BootNode) Start(ctx context.Context) error {
|
|
binaryPath, found := bazel.FindBinary("tools/bootnode", "bootnode")
|
|
if !found {
|
|
log.Info(binaryPath)
|
|
return errors.New("boot node binary not found")
|
|
}
|
|
|
|
stdOutFile, err := helpers.DeleteAndCreateFile(e2e.TestParams.LogPath, e2e.BootNodeLogFileName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
args := []string{
|
|
fmt.Sprintf("--log-file=%s", stdOutFile.Name()),
|
|
fmt.Sprintf("--discv5-port=%d", e2e.TestParams.BootNodePort),
|
|
fmt.Sprintf("--metrics-port=%d", e2e.TestParams.BootNodePort+20),
|
|
"--debug",
|
|
}
|
|
|
|
cmd := exec.CommandContext(ctx, binaryPath, args...)
|
|
cmd.Stdout = stdOutFile
|
|
cmd.Stderr = stdOutFile
|
|
log.Infof("Starting boot node with flags: %s", strings.Join(args[1:], " "))
|
|
if err = cmd.Start(); err != nil {
|
|
return fmt.Errorf("failed to start beacon node: %w", err)
|
|
}
|
|
|
|
if err = helpers.WaitForTextInFile(stdOutFile, "Running bootnode"); err != nil {
|
|
return fmt.Errorf("could not find enr for bootnode, this means the bootnode had issues starting: %w", err)
|
|
}
|
|
|
|
node.enr, err = enrFromLogFile(stdOutFile.Name())
|
|
if err != nil {
|
|
return fmt.Errorf("could not get enr for bootnode: %w", err)
|
|
}
|
|
|
|
// Mark node as ready.
|
|
close(node.started)
|
|
|
|
return cmd.Wait()
|
|
}
|
|
|
|
// Started checks whether a boot node is started and ready to be queried.
|
|
func (node *BootNode) Started() <-chan struct{} {
|
|
return node.started
|
|
}
|
|
|
|
func enrFromLogFile(name string) (string, error) {
|
|
byteContent, err := ioutil.ReadFile(name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
contents := string(byteContent)
|
|
|
|
searchText := "Running bootnode: "
|
|
startIdx := strings.Index(contents, searchText)
|
|
if startIdx == -1 {
|
|
return "", fmt.Errorf("did not find ENR text in %s", contents)
|
|
}
|
|
startIdx += len(searchText)
|
|
endIdx := strings.Index(contents[startIdx:], " prefix=bootnode")
|
|
if endIdx == -1 {
|
|
return "", fmt.Errorf("did not find ENR text in %s", contents)
|
|
}
|
|
return contents[startIdx : startIdx+endIdx-1], nil
|
|
}
|