github.com/koko1123/flow-go-1@v0.29.6/engine/execution/ingestion/stop_control.go

package ingestion

import (
	"context"
	"fmt"
	"sync"

	"github.com/rs/zerolog"

	"github.com/koko1123/flow-go-1/engine/execution/state"
	"github.com/koko1123/flow-go-1/model/flow"
)

// StopControl is a specialized component used by ingestion.Engine to encapsulate
// control of pausing/stopping blocks execution.
// It is intended to work tightly with the Engine, not as a general mechanism or interface.
// StopControl follows states described in StopState
type StopControl struct {
	sync.RWMutex
	// desired stop height, the first value new version should be used, so this height WON'T
	// be executed
	height uint64

	// if the node should crash or just pause after reaching stop height
	crash              bool
	stopAfterExecuting flow.Identifier

	log   zerolog.Logger
	state StopControlState

	// used to prevent setting stop height to block which has already been executed
	highestExecutingHeight uint64
}

type StopControlState byte

const (
	// StopControlOff default state, envisioned to be used most of the time. Stopping module is simply off,
	// blocks will be processed "as usual".
	StopControlOff StopControlState = iota

	// StopControlSet means stop height is set but not reached yet, and nothing related to stopping happened yet.
	// We could still go back to StopControlOff or progress to StopControlCommenced.
	StopControlSet

	// StopControlCommenced indicates that stopping process has commenced and no parameters can be changed anymore.
	// For example, blocks at or above stop height has been received, but finalization didn't reach stop height yet.
	// It can only progress to StopControlPaused
	StopControlCommenced

	// StopControlPaused means EN has stopped processing blocks. It can happen by reaching the set stopping `height`, or
	// if the node was started in pause mode.
	// It is a final state and cannot be changed
	StopControlPaused
)

// NewStopControl creates new empty NewStopControl
func NewStopControl(log zerolog.Logger, paused bool, lastExecutedHeight uint64) *StopControl {
	state := StopControlOff
	if paused {
		state = StopControlPaused
	}
	log.Debug().Msgf("created StopControl module with paused = %t", paused)
	return &StopControl{
		log:                    log,
		state:                  state,
		highestExecutingHeight: lastExecutedHeight,
	}
}

// GetState returns current state of StopControl module
func (s *StopControl) GetState() StopControlState {
	s.RLock()
	defer s.RUnlock()
	return s.state
}

// IsPaused returns true is block execution has been paused
func (s *StopControl) IsPaused() bool {
	s.RLock()
	defer s.RUnlock()
	return s.state == StopControlPaused
}

// SetStopHeight sets new stop height and crash mode, and return old values:
//   - height
//   - crash
//
// Returns error if the stopping process has already commenced, new values will be rejected.
func (s *StopControl) SetStopHeight(height uint64, crash bool) (uint64, bool, error) {
	s.Lock()
	defer s.Unlock()

	oldHeight := s.height
	oldCrash := s.crash

	if s.state == StopControlCommenced {
		return oldHeight, oldCrash, fmt.Errorf("cannot update stop height, stopping commenced for height %d with crash=%t", oldHeight, oldCrash)
	}

	if s.state == StopControlPaused {
		return oldHeight, oldCrash, fmt.Errorf("cannot update stop height, already paused")
	}

	// +1 because we track last executing height, so +1 is the lowest possible block to stop
	if height <= s.highestExecutingHeight+1 {
		return oldHeight, oldCrash, fmt.Errorf("cannot update stop height, given height %d at or below last executed %d", height, s.highestExecutingHeight)
	}

	s.log.Info().
		Int8("previous_state", int8(s.state)).Int8("new_state", int8(StopControlSet)).
		Uint64("height", height).Bool("crash", crash).
		Uint64("old_height", oldHeight).Bool("old_crash", oldCrash).Msg("new stop height set")

	s.state = StopControlSet

	s.height = height
	s.crash = crash
	s.stopAfterExecuting = flow.ZeroID

	return oldHeight, oldCrash, nil
}

// GetStopHeight returns:
//   - height
//   - crash
//
// Values are undefined if they were not previously set
func (s *StopControl) GetStopHeight() (uint64, bool) {
	s.RLock()
	defer s.RUnlock()

	return s.height, s.crash
}

// blockProcessable should be called when new block is processable.
// It returns boolean indicating if the block should be processed.
func (s *StopControl) blockProcessable(b *flow.Header) bool {

	s.Lock()
	defer s.Unlock()

	if s.state == StopControlOff {
		return true
	}

	if s.state == StopControlPaused {
		return false
	}

	// skips blocks at or above requested stop height
	if b.Height >= s.height {
		s.log.Warn().Int8("previous_state", int8(s.state)).Int8("new_state", int8(StopControlCommenced)).Msgf("Skipping execution of %s at height %d because stop has been requested at height %d", b.ID(), b.Height, s.height)
		s.state = StopControlCommenced // if block was skipped, move into commenced state
		return false
	}

	return true
}

// blockFinalized should be called when a block is marked as finalized
func (s *StopControl) blockFinalized(ctx context.Context, execState state.ReadOnlyExecutionState, h *flow.Header) {

	s.Lock()
	defer s.Unlock()

	if s.state == StopControlOff || s.state == StopControlPaused {
		return
	}

	// Once finalization reached stop height we can be sure no other fork will be valid at this height,
	// if this block's parent has been executed, we are safe to stop or crash.
	// This will happen during normal execution, where blocks are executed before they are finalized.
	// However, it is possible that EN block computation progress can fall behind. In this case,
	// we want to crash only after the execution reached the stop height.
	if h.Height == s.height {

		executed, err := state.IsBlockExecuted(ctx, execState, h.ParentID)
		if err != nil {
			// any error here would indicate unexpected storage error, so we crash the node
			s.log.Fatal().Err(err).Str("block_id", h.ID().String()).Msg("failed to check if the block has been executed")
			return
		}

		if executed {
			s.stopExecution()
		} else {
			s.stopAfterExecuting = h.ParentID
			s.log.Info().Msgf("Node scheduled to stop executing after executing block %s at height %d", s.stopAfterExecuting.String(), h.Height-1)
		}

	}

}

// blockExecuted should be called after a block has finished execution
func (s *StopControl) blockExecuted(h *flow.Header) {
	s.Lock()
	defer s.Unlock()

	if s.state == StopControlPaused || s.state == StopControlOff {
		return
	}

	if s.stopAfterExecuting == h.ID() {
		// double check. Even if requested stop height has been changed multiple times,
		// as long as it matches this block we are safe to terminate

		if h.Height == s.height-1 {
			s.stopExecution()
		} else {
			s.log.Warn().Msgf("Inconsistent stopping state. Scheduled to stop after executing block ID %s and height %d, but this block has a height %d. ",
				h.ID().String(), s.height-1, h.Height)
		}
	}
}

func (s *StopControl) stopExecution() {
	if s.crash {
		s.log.Fatal().Msgf("Crashing as finalization reached requested stop height %d and the highest executed block is (%d - 1)", s.height, s.height)
	} else {
		s.log.Debug().Int8("previous_state", int8(s.state)).Int8("new_state", int8(StopControlPaused)).Msg("StopControl state transition")
		s.state = StopControlPaused
		s.log.Warn().Msgf("Pausing execution as finalization reached requested stop height %d", s.height)
	}
}

// executingBlockHeight should be called while execution of height starts, used for internal tracking of the minimum
// possible value of height
func (s *StopControl) executingBlockHeight(height uint64) {
	if s.state == StopControlPaused {
		return
	}

	// updating the highest executing height, which will be used to reject setting stop height that
	// is too low.
	if height > s.highestExecutingHeight {
		s.highestExecutingHeight = height
	}
}