prysm-pulse/shared/slotutil/countdown.go

package slotutil

import (
	"context"
	"fmt"
	"time"

	"github.com/prysmaticlabs/prysm/shared/params"
	"github.com/prysmaticlabs/prysm/shared/roughtime"
	"github.com/sirupsen/logrus"
)

var log = logrus.WithField("prefix", "slotutil")

// CountdownToGenesis starts a ticker at the specified duration
// logging the remaining minutes until the genesis chainstart event
// along with important genesis state metadata such as number
// of genesis validators.
func CountdownToGenesis(ctx context.Context, genesisTime time.Time, genesisValidatorCount uint64) {
	ticker := time.NewTicker(params.BeaconConfig().GenesisCountdownInterval)
	defer func() {
		// Used in anonymous function to make sure that updated (per second) ticker is stopped.
		ticker.Stop()
	}()
	logFields := logrus.Fields{
		"genesisValidators": fmt.Sprintf("%d", genesisValidatorCount),
		"genesisTime":       fmt.Sprintf("%v", genesisTime),
	}
	secondTimerActivated := false
	for {
		currentTime := roughtime.Now()
		if currentTime.After(genesisTime) {
			log.WithFields(logFields).Info("Chain genesis time reached")
			return
		}
		timeRemaining := genesisTime.Sub(currentTime)
		if !secondTimerActivated && timeRemaining <= 2*time.Minute {
			ticker.Stop()
			// Replace ticker with a one having higher granularity.
			ticker = time.NewTicker(time.Second)
			secondTimerActivated = true
		}
		select {
		case <-ticker.C:
			if timeRemaining >= time.Second {
				log.WithFields(logFields).Infof(
					"%s until chain genesis",
					timeRemaining.Truncate(time.Second),
				)
			}
		case <-ctx.Done():
			log.Debug("Context closed, exiting routine")
			return
		}
	}
}