From e61cc06955d4d76a1781392577d0d1bcc1c60824 Mon Sep 17 00:00:00 2001
From: Faisal Alam <ifaisalalam@gmail.com>
Date: Sat, 23 Apr 2022 15:32:54 +0530
Subject: [PATCH] feat(context): return GIN Context from Value method (#2825)

---
 context.go      | 6 ++++++
 context_test.go | 1 +
 2 files changed, 7 insertions(+)

diff --git a/context.go b/context.go
index 0fe1333..5e53aaf 100644
--- a/context.go
+++ b/context.go
@@ -39,6 +39,9 @@ const (
 // BodyBytesKey indicates a default body bytes key.
 const BodyBytesKey = "_gin-gonic/gin/bodybyteskey"
 
+// ContextKey is the key that a Context returns itself for.
+const ContextKey = "_gin-gonic/gin/contextkey"
+
 // abortIndex represents a typical value used in abort functions.
 const abortIndex int8 = math.MaxInt8 >> 1
 
@@ -1163,6 +1166,9 @@ func (c *Context) Value(key any) any {
 	if key == 0 {
 		return c.Request
 	}
+	if key == ContextKey {
+		return c
+	}
 	if keyAsString, ok := key.(string); ok {
 		if val, exists := c.Get(keyAsString); exists {
 			return val
diff --git a/context_test.go b/context_test.go
index fb46e67..20038e9 100644
--- a/context_test.go
+++ b/context_test.go
@@ -1880,6 +1880,7 @@ func TestContextGolangContext(t *testing.T) {
 	assert.Equal(t, ti, time.Time{})
 	assert.False(t, ok)
 	assert.Equal(t, c.Value(0), c.Request)
+	assert.Equal(t, c.Value(ContextKey), c)
 	assert.Nil(t, c.Value("foo"))
 
 	c.Set("foo", "bar")