github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-01-23 09:00:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add httpc/Get httpc/Post (#1640)

Browse Source
This commit is contained in:
Kevin Wan 2022-03-13 14:49:14 +08:00 committed by GitHub
parent 3279a7ef0f
commit 85cf662c6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
rest/httpc/requests.go
View File

@ -1,6 +1,7 @@
package httpc
import (
"io"
"net/http"
"github.com/zeromicro/go-zero/core/breaker"
@ -36,6 +37,26 @@ func Do(key string, r *http.Request, opts ...Option) (resp *http.Response, err e
return
}
// Get sends an HTTP GET request to the service assocated with the given key.
func Get(key, url string, opts ...Option) (*http.Response, error) {
r, err := http.NewRequest(http.MethodGet, url, nil)
if err != nil {
return nil, err
}
return Do(key, r, opts...)
}
// Post sends an HTTP POST request to the service assocated with the given key.
func Post(key, url, contentType string, body io.Reader, opts ...Option) (*http.Response, error) {
r, err := http.NewRequest(http.MethodPost, url, body)
if err != nil {
return nil, err
}
return Do(key, r, opts...)
}
func doRequest(key string, r *http.Request, opts ...Option) (resp *http.Response, err error) {
brk := breaker.GetBreaker(key)
err = brk.DoWithAcceptable(func() error {

12
rest/httpc/requests_test.go
View File

@ -11,9 +11,9 @@ import (
func TestDo(t *testing.T) {
svr := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
}))
req, err := http.NewRequest(http.MethodGet, svr.URL, nil)
assert.Nil(t, err)
resp, err := Do("foo", req, func(cli *http.Client) {
_, err := Get("foo", "tcp://bad request")
assert.NotNil(t, err)
resp, err := Get("foo", svr.URL, func(cli *http.Client) {
cli.Transport = http.DefaultTransport
})
assert.Nil(t, err)
@ -22,9 +22,9 @@ func TestDo(t *testing.T) {
func TestDoNotFound(t *testing.T) {
svr := httptest.NewServer(http.NotFoundHandler())
req, err := http.NewRequest(http.MethodGet, svr.URL, nil)
assert.Nil(t, err)
resp, err := Do("foo", req)
_, err := Post("foo", "tcp://bad request", "application/json", nil)
assert.NotNil(t, err)
resp, err := Post("foo", svr.URL, "application/json", nil)
assert.Nil(t, err)
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
}