current node in request option (#205)

go-kratos · Sep 4, 2023 · 084b012 · 084b012
1 parent 39a2e09
commit 084b012
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/client/client.go b/client/client.go
@@ -39,6 +39,7 @@ func (c *client) RoundTrip(req *http.Request) (resp *http.Response, err error) {
 	if err != nil {
 		return nil, err
 	}
+	reqOpt.CurrentNode = n
 
 	addr := n.Address()
 	reqOpt.Backends = append(reqOpt.Backends, addr)

diff --git a/middleware/request.go b/middleware/request.go
@@ -17,6 +17,7 @@ type RequestOptions struct {
 	Metadata             map[string]string
 	UpstreamStatusCode   []int
 	UpstreamResponseTime []float64
+	CurrentNode          selector.Node
 	DoneFunc             selector.DoneFunc
 	LastAttempt          bool
 	Values               RequestValues