Skip to content
Snippets Groups Projects
Select Git revision
  • af182caa8c569d11af0d176ed5663b3ac2791fd6
  • without_tipselection default
  • develop protected
  • fix/grafana-local-dashboard
  • wasp
  • fix/dashboard-explorer-freeze
  • master
  • feat/timerqueue
  • test/sync_debug_and_650
  • feat/sync_revamp_inv
  • wip/sync
  • tool/db-recovery
  • portcheck/fix
  • fix/synchronization
  • feat/new-dashboard-analysis
  • feat/refactored-analysis-dashboard
  • feat/new-analysis-dashboard
  • test/demo-prometheus-fpc
  • prometheus_metrics
  • wip/analysis-server
  • merge/fpc-test-value-transfer
  • v0.2.2
  • v0.2.1
  • v0.2.0
  • v0.1.3
  • v0.1.2
  • v0.1.1
  • v0.1.0
28 results

plugin.go

Blame
  • plugin.go 2.18 KiB
    package gracefulshutdown
    
    import (
    	"os"
    	"os/signal"
    	"sort"
    	"strings"
    	"sync"
    	"syscall"
    	"time"
    
    	"github.com/iotaledger/goshimmer/plugins/config"
    	"github.com/iotaledger/hive.go/daemon"
    	"github.com/iotaledger/hive.go/logger"
    	"github.com/iotaledger/hive.go/node"
    )
    
    // PluginName is the name of the graceful shutdown plugin.
    const PluginName = "Graceful Shutdown"
    
    var (
    	// plugin is the plugin instance of the graceful shutdown plugin.
    	plugin                  *node.Plugin
    	once                    sync.Once
    	log                     *logger.Logger
    	gracefulStop            chan os.Signal
    	waitToKillTimeInSeconds int
    )
    
    func configure(*node.Plugin) {
    	waitToKillTimeInSeconds = config.Node().GetInt(CfgWaitToKillTimeInSeconds)
    
    	log = logger.NewLogger(PluginName)
    	gracefulStop = make(chan os.Signal)
    
    	signal.Notify(gracefulStop, syscall.SIGTERM)
    	signal.Notify(gracefulStop, syscall.SIGINT)
    
    	go func() {
    		<-gracefulStop
    
    		log.Warnf("Received shutdown request - waiting (max %d) to finish processing ...", waitToKillTimeInSeconds)
    
    		go func() {
    			ticker := time.NewTicker(1 * time.Second)
    			defer ticker.Stop()
    
    			start := time.Now()
    			for x := range ticker.C {
    				secondsSinceStart := x.Sub(start).Seconds()
    
    				if secondsSinceStart <= float64(waitToKillTimeInSeconds) {
    					processList := ""
    					runningBackgroundWorkers := daemon.GetRunningBackgroundWorkers()
    					if len(runningBackgroundWorkers) >= 1 {
    						sort.Strings(runningBackgroundWorkers)
    						processList = "(" + strings.Join(runningBackgroundWorkers, ", ") + ") "
    					}
    					log.Warnf("Received shutdown request - waiting (max %d seconds) to finish processing %s...", waitToKillTimeInSeconds-int(secondsSinceStart), processList)
    				} else {
    					log.Error("Background processes did not terminate in time! Forcing shutdown ...")
    					os.Exit(1)
    				}
    			}
    		}()
    
    		daemon.Shutdown()
    	}()
    }
    
    // Plugin gets the plugin instance.
    func Plugin() *node.Plugin {
    	once.Do(func() {
    		plugin = node.NewPlugin(PluginName, node.Enabled, configure)
    	})
    	return plugin
    }
    
    // ShutdownWithError prints out an error message and shuts down the default daemon instance.
    func ShutdownWithError(err error) {
    	log.Error(err)
    	gracefulStop <- syscall.SIGINT
    }