handle_additional_operations.go

package actions

import (
	"os"
	"os/signal"
	"sync"
	"syscall"

	"github.com/rarimo/bdjuno/modules/actions/handlers"
	actionstypes "github.com/rarimo/bdjuno/modules/actions/types"
)

var (
	waitGroup sync.WaitGroup
)

func (m *Module) RunAdditionalOperations() error {
	// Build the worker
	context := actionstypes.NewContext(m.node, m.sources)
	worker := actionstypes.NewActionsWorker(context)

	// Register the endpoints

	// -- Bank --
	worker.RegisterHandler("/account_balance", handlers.AccountBalanceHandler)

	// -- Distribution --
	worker.RegisterHandler("/delegation_reward", handlers.DelegationRewardHandler)
	worker.RegisterHandler("/delegator_withdraw_address", handlers.DelegatorWithdrawAddressHandler)
	worker.RegisterHandler("/validator_commission_amount", handlers.ValidatorCommissionAmountHandler)

	// -- Staking Delegator --
	worker.RegisterHandler("/delegation", handlers.DelegationHandler)
	worker.RegisterHandler("/delegation_total", handlers.TotalDelegationAmountHandler)
	worker.RegisterHandler("/unbonding_delegation", handlers.UnbondingDelegationsHandler)
	worker.RegisterHandler("/unbonding_delegation_total", handlers.UnbondingDelegationsTotal)
	worker.RegisterHandler("/redelegation", handlers.RedelegationHandler)

	// -- Staking Validator --
	worker.RegisterHandler("/validator_delegations", handlers.ValidatorDelegation)
	worker.RegisterHandler("/validator_redelegations_from", handlers.ValidatorRedelegationsFromHandler)
	worker.RegisterHandler("/validator_unbonding_delegations", handlers.ValidatorUnbondingDelegationsHandler)

	// Listen for and trap any OS signal to gracefully shutdown and exit
	m.trapSignal()

	// Start the worker
	waitGroup.Add(1)
	go worker.Start(m.cfg.Port)

	// Block main process (signal capture will call WaitGroup's Done)
	waitGroup.Wait()
	return nil
}

// trapSignal will listen for any OS signal and invoke Done on the main
// WaitGroup allowing the main process to gracefully exit.
func (m *Module) trapSignal() {
	var sigCh = make(chan os.Signal, 1)

	signal.Notify(sigCh, syscall.SIGTERM)
	signal.Notify(sigCh, syscall.SIGINT)

	go func() {
		defer m.node.Stop()
		defer waitGroup.Done()
	}()
}