From 0304ee96edb2f7f863d5e8337073cade0327af4a Mon Sep 17 00:00:00 2001
From: kaiiak <aNxFi37X@outlook.com>
Date: Tue, 1 Sep 2020 09:33:54 +0800
Subject: [PATCH] Add GetUint and GetUint64 method on gin.context (#2487)

---
 context.go      | 16 ++++++++++++++++
 context_test.go | 12 ++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/context.go b/context.go
index 95b1807d72..216cecae34 100644
--- a/context.go
+++ b/context.go
@@ -295,6 +295,22 @@ func (c *Context) GetInt64(key string) (i64 int64) {
 	return
 }
 
+// GetUint returns the value associated with the key as an unsigned integer.
+func (c *Context) GetUint(key string) (ui uint) {
+	if val, ok := c.Get(key); ok && val != nil {
+		ui, _ = val.(uint)
+	}
+	return
+}
+
+// GetUint64 returns the value associated with the key as an unsigned integer.
+func (c *Context) GetUint64(key string) (ui64 uint64) {
+	if val, ok := c.Get(key); ok && val != nil {
+		ui64, _ = val.(uint64)
+	}
+	return
+}
+
 // GetFloat64 returns the value associated with the key as a float64.
 func (c *Context) GetFloat64(key string) (f64 float64) {
 	if val, ok := c.Get(key); ok && val != nil {
diff --git a/context_test.go b/context_test.go
index 1a5a3c5ff5..e2f8de06e4 100644
--- a/context_test.go
+++ b/context_test.go
@@ -261,6 +261,18 @@ func TestContextGetInt64(t *testing.T) {
 	assert.Equal(t, int64(42424242424242), c.GetInt64("int64"))
 }
 
+func TestContextGetUint(t *testing.T) {
+	c, _ := CreateTestContext(httptest.NewRecorder())
+	c.Set("uint", uint(1))
+	assert.Equal(t, uint(1), c.GetUint("uint"))
+}
+
+func TestContextGetUint64(t *testing.T) {
+	c, _ := CreateTestContext(httptest.NewRecorder())
+	c.Set("uint64", uint64(18446744073709551615))
+	assert.Equal(t, uint64(18446744073709551615), c.GetUint64("uint64"))
+}
+
 func TestContextGetFloat64(t *testing.T) {
 	c, _ := CreateTestContext(httptest.NewRecorder())
 	c.Set("float64", 4.2)