blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2026-06-07 04:38:19 +08:00
Code Issues Projects Releases Wiki Activity

feat(context): add ShouldBindBodyWithProtoBuf shortcut

Browse Source
This commit is contained in:
1911860538 2026-05-23 22:42:33 +08:00
parent 5f4f964325
commit cb82095769
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
context.go
View File

@ -967,6 +967,11 @@ func (c *Context) ShouldBindBodyWithPlain(obj any) error {
return c.ShouldBindBodyWith(obj, binding.Plain) return c.ShouldBindBodyWith(obj, binding.Plain)
} }
// ShouldBindBodyWithProtoBuf is a shortcut for c.ShouldBindBodyWith(obj, binding.ProtoBuf).
func (c *Context) ShouldBindBodyWithProtoBuf(obj any) error {
return c.ShouldBindBodyWith(obj, binding.ProtoBuf)
}
// ClientIP implements one best effort algorithm to return the real client IP. // ClientIP implements one best effort algorithm to return the real client IP.
// It calls c.RemoteIP() under the hood, to check if the remote IP is a trusted proxy or not. // It calls c.RemoteIP() under the hood, to check if the remote IP is a trusted proxy or not.
// If it is it will then try to parse the headers defined in Engine.RemoteIPHeaders (defaulting to [X-Forwarded-For, X-Real-IP]). // If it is it will then try to parse the headers defined in Engine.RemoteIPHeaders (defaulting to [X-Forwarded-For, X-Real-IP]).

15
context_test.go
View File

@ -2916,6 +2916,21 @@ func TestContextShouldBindBodyWithPlain(t *testing.T) {
} }
} }
func TestContextShouldBindBodyWithProtoBuf(t *testing.T) {
label := "FOO"
protoBody, err := proto.Marshal(&testdata.Test{Label: &label})
require.NoError(t, err)
w := httptest.NewRecorder()
c, _ := CreateTestContext(w)
c.Request = httptest.NewRequest(http.MethodPost, "/", bytes.NewReader(protoBody))
obj := testdata.Test{}
require.NoError(t, c.ShouldBindBodyWithProtoBuf(&obj))
require.NotNil(t, obj.Label)
assert.Equal(t, "FOO", *obj.Label)
}
func TestContextGolangContext(t *testing.T) { func TestContextGolangContext(t *testing.T) {
c, _ := CreateTestContext(httptest.NewRecorder()) c, _ := CreateTestContext(httptest.NewRecorder())
c.Request, _ = http.NewRequest(http.MethodPost, "/", strings.NewReader(`{"foo":"bar", "bar":"foo"}`)) c.Request, _ = http.NewRequest(http.MethodPost, "/", strings.NewReader(`{"foo":"bar", "bar":"foo"}`))