v1.0.0

internal/app/context.go

@@ -0,0 +1,18 @@
+// Package app provides application lifecycle primitives.
+package app
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+)
+
+// NewContext returns a context for the whole application lifecycle.
+//
+// The context is cancelled when the parent is cancelled or when the process
+// receives an interrupt or a termination signal. Call the returned function
+// during shutdown to unregister signal notifications and release resources.
+func NewContext(parent context.Context) (context.Context, context.CancelFunc) {
+	return signal.NotifyContext(parent, os.Interrupt, syscall.SIGTERM)
+}

internal/app/context_test.go

@@ -0,0 +1,21 @@
+package app
+
+import (
+	"context"
+	"testing"
+	"time"
+)
+
+func TestNewContextCancelsWithParent(t *testing.T) {
+	parent, cancelParent := context.WithCancel(context.Background())
+	ctx, stop := NewContext(parent)
+	t.Cleanup(stop)
+
+	cancelParent()
+
+	select {
+	case <-ctx.Done():
+	case <-time.After(time.Second):
+		t.Fatal("application context was not cancelled")
+	}
+}