diff --git a/cmd/vspd/main.go b/cmd/vspd/main.go index b22894c..94357c1 100644 --- a/cmd/vspd/main.go +++ b/cmd/vspd/main.go @@ -14,6 +14,7 @@ import ( "github.com/decred/dcrd/wire" "github.com/decred/vspd/database" "github.com/decred/vspd/internal/config" + "github.com/decred/vspd/internal/signal" "github.com/decred/vspd/internal/version" "github.com/decred/vspd/internal/vspd" "github.com/decred/vspd/internal/webapi" @@ -108,7 +109,7 @@ func run() int { // Create a context that is canceled when a shutdown request is received // through an interrupt signal such as SIGINT (Ctrl+C). - ctx := shutdownListener(log) + ctx := signal.ShutdownListener(log) // Start the webapi server. shutdownWg.Add(1) diff --git a/cmd/vspd/signal.go b/internal/signal/signal.go similarity index 89% rename from cmd/vspd/signal.go rename to internal/signal/signal.go index 851810a..6705f26 100644 --- a/cmd/vspd/signal.go +++ b/internal/signal/signal.go @@ -3,7 +3,7 @@ // Use of this source code is governed by an ISC // license that can be found in the LICENSE file. -package main +package signal import ( "context" @@ -17,10 +17,10 @@ import ( // shutdown. This may be modified during init depending on the platform. var interruptSignals = []os.Signal{os.Interrupt} -// shutdownListener listens for OS Signals such as SIGINT (Ctrl+C) and shutdown +// ShutdownListener listens for OS Signals such as SIGINT (Ctrl+C) and shutdown // requests from requestShutdown. It returns a context that is canceled when // either signal is received. -func shutdownListener(log slog.Logger) context.Context { +func ShutdownListener(log slog.Logger) context.Context { ctx, cancel := context.WithCancel(context.Background()) go func() { interruptChannel := make(chan os.Signal, 1) diff --git a/cmd/vspd/signal_syscall.go b/internal/signal/signal_syscall.go similarity index 96% rename from cmd/vspd/signal_syscall.go rename to internal/signal/signal_syscall.go index c40d3ba..32c3806 100644 --- a/cmd/vspd/signal_syscall.go +++ b/internal/signal/signal_syscall.go @@ -5,7 +5,7 @@ //go:build windows || aix || android || darwin || dragonfly || freebsd || hurd || illumos || ios || linux || netbsd || openbsd || solaris -package main +package signal import ( "syscall"