Skip to content
Snippets Groups Projects
Select Git revision
  • 324cf7858dfe23970a9f4f59e882832522215b52
  • develop default protected
  • congestioncontrol
  • merge-v-data-collection-spammer-0.8.2
  • WIP-merge-v-data-collection-spammer-0.8.2
  • merge-v-data-collection-spammer-0.7.7
  • tmp
  • test-masterpow-fixing
  • test-masterpow
  • test-echo
  • v-data-collection
  • v-data-collection-spammer
  • tmp-dump-spam-info
  • dump-msg-info-0.3.1
  • test-dump-message-info
  • spammer-exprandom
  • extra/tutorial
  • without_tipselection
  • hacking-docker-network
  • hacking-docker-network-0.2.3
  • master
  • v0.2.3
22 results

docker-compose.yml

  • 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
    }