dev day #79

internal/middleware/db_context.go

@@ -0,0 +1,37 @@
+package middleware
+
+import (
+	"context"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+// DBContext adds a context with timeout to all database operations
+// This prevents queries from hanging indefinitely and exhausting connections
+func DBContext() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		// Create a context with timeout for this request's database operations
+		// 15 seconds is generous for most queries while preventing indefinite hangs
+		ctx, cancel := context.WithTimeout(c.Request.Context(), 15*time.Second)
+		defer cancel()
+		
+		// Store the context so controllers can use it with db.WithContext(ctx)
+		c.Set("dbCtx", ctx)
+		
+		c.Next()
+	}
+}
+
+// GetDBContext retrieves the database context from gin.Context
+// Returns a background context with timeout if not found
+func GetDBContext(c *gin.Context) context.Context {
+	if ctx, exists := c.Get("dbCtx"); exists {
+		if dbCtx, ok := ctx.(context.Context); ok {
+			return dbCtx
+		}
+	}
+	// Fallback with timeout
+	ctx, _ := context.WithTimeout(context.Background(), 15*time.Second)
+	return ctx
+}