zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-10-25 09:40:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

grpc重写插件，首次提交

Browse Source
This commit is contained in:
Liujian
2023-02-15 14:03:26 +08:00
parent 53046df29f
commit c8655ba04d
10 changed files with 158 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
node/grpc-context/context.go
View File

@@ -203,6 +203,15 @@ func (c *Context) Invoke(address string, timeout time.Duration) error {
if err != nil {
return err
}
passHost, targetHost := c.GetUpstreamHostHandler().PassHost()
switch passHost {
case eocontext.PassHost:
case eocontext.NodeHost:
c.proxy.Headers().Set(":authority", address)
case eocontext.ReWriteHost:
c.proxy.Headers().Set(":authority", targetHost)
}
c.proxy.Headers().Set("grpc-timeout", fmt.Sprintf("%dn", timeout))
clientCtx, _ := context.WithCancel(metadata.NewOutgoingContext(c.Context(), c.proxy.Headers().Copy()))
clientStream, err := grpc.NewClientStream(clientCtx, clientStreamDescForProxying, clientConn, c.proxy.FullMethodName())