blackteay/gf
1
0
Fork 0
mirror of https://github.com/gogf/gf.git synced 2025-04-05 11:18:50 +08:00
Code Issues Projects Releases Wiki Activity

add chaining function Discovery to disable/enable discovery feature for package gclient; fix issue #2737 (#2738)

Browse Source
This commit is contained in:
John Guo 2023-07-04 20:21:12 +08:00 committed by GitHub
parent 740dfa58a6
commit ba2a7e4417
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
contrib/registry/file/file_z_http_test.go
View File

@ -12,6 +12,8 @@ import (
"time"
"github.com/gogf/gf/contrib/registry/file/v2"
"github.com/gogf/gf/v2/errors/gcode"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/net/ghttp"
"github.com/gogf/gf/v2/net/gsvc"
@ -48,3 +50,38 @@ func Test_HTTP_Registry(t *testing.T) {
t.Assert(client.GetContent(ctx, "/http-registry"), svcName)
})
}
func Test_HTTP_Discovery_Disable(t *testing.T) {
var (
svcName = guid.S()
dirPath = gfile.Temp(svcName)
)
defer gfile.Remove(dirPath)
gsvc.SetRegistry(file.New(dirPath))
s := g.Server()
s.BindHandler("/http-registry", func(r *ghttp.Request) {
r.Response.Write(svcName)
})
s.SetDumpRouterMap(false)
s.Start()
defer s.Shutdown()
time.Sleep(100 * time.Millisecond)
gtest.C(t, func(t *gtest.T) {
client := g.Client()
client.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))
result, err := client.Get(ctx, "/http-registry")
defer result.Close()
t.Assert(gerror.Code(err), gcode.CodeNotFound)
})
gtest.C(t, func(t *gtest.T) {
client := g.Client()
client.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))
result, err := client.Discovery(nil).Get(ctx, "/http-registry")
defer result.Close()
t.AssertNil(err)
t.Assert(result.ReadAllString(), svcName)
})
}

10
net/gclient/gclient_chain.go
View File

@ -8,6 +8,8 @@ package gclient
import (
"time"
"github.com/gogf/gf/v2/net/gsvc"
)
// Prefix is a chaining function,
@ -37,6 +39,14 @@ func (c *Client) HeaderRaw(headers string) *Client {
return newClient
}
// Discovery is a chaining function, which sets the discovery for client.
// You can use `Discovery(nil)` to disable discovery feature for current client.
func (c *Client) Discovery(discovery gsvc.Discovery) *Client {
newClient := c.Clone()
newClient.SetDiscovery(discovery)
return newClient
}
// Cookie is a chaining function,
// which sets cookie items with map for next request.
func (c *Client) Cookie(m map[string]string) *Client {