feat: introduce BaseContext, pass shutdown context to f()

app/app.go

@@ -120,7 +120,10 @@ func Run(f func(ctx context.Context, lg *zap.Logger, m *Telemetry) error, op ...
 	if ctx, err = autologs.Setup(ctx, m.LoggerProvider(), opts.zapTee); err != nil {
 		panic(fmt.Sprintf("failed to setup logs: %v", err))
 	}
+
 	shutdownCtx = zctx.Base(shutdownCtx, zctx.From(ctx))
+	m.shutdownContext = shutdownCtx
+	m.baseContext = ctx
 
 	{
 		// Automatically setting GOMAXPROCS.
@@ -155,7 +158,8 @@ func Run(f func(ctx context.Context, lg *zap.Logger, m *Telemetry) error, op ...
 				rerr = fmt.Errorf("shutting down (panic): %v", ec)
 			}
 		}()
-		if err := f(ctx, zctx.From(ctx), m); err != nil {
+		m.baseContext = ctx
+		if err := f(m.shutdownContext, zctx.From(ctx), m); err != nil {
 			if errors.Is(err, ctx.Err()) {
 				// Parent context got cancelled, error is expected.
 				// TODO(ernado): check for shutdownCtx instead.

app/telemetry.go

@@ -53,22 +53,28 @@ type Telemetry struct {
 	meterProvider   metric.MeterProvider
 	loggerProvider  log.LoggerProvider
 	shutdownContext context.Context
+	baseContext     context.Context
 
-	resource   *resource.Resource
-	propagator propagation.TextMapPropagator
+	resource *resource.Resource
 
-	shutdowns []shutdown
+	propagator propagation.TextMapPropagator
+	shutdowns  []shutdown
 }
 
 // ShutdownContext is context for triggering graceful shutdown.
 // It is cancelled on SIGINT.
 //
-// Base context can be used during shutdown to finish pending operations, it will be cancelled later
+// Base context [Telemetry.BaseContext] can be used during shutdown to finish pending operations, it will be cancelled later
 // on timeout.
 func (m *Telemetry) ShutdownContext() context.Context {
 	return m.shutdownContext
 }
 
+// BaseContext is base context for the application.
+func (m *Telemetry) BaseContext() context.Context {
+	return m.baseContext
+}
+
 func (m *Telemetry) registerShutdown(name string, fn func(ctx context.Context) error) {
 	m.shutdowns = append(m.shutdowns, shutdown{name: name, fn: fn})
 }
@@ -214,6 +220,7 @@ func newTelemetry(
 		resource: res,
 
 		shutdownContext: shutdownCtx,
+		baseContext:     baseCtx,
 	}
 	ctx := baseCtx
 	{