zishuo/rpcx
1
0
Fork 0
mirror of https://github.com/smallnest/rpcx.git synced 2025-10-26 17:40:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

#232 support opentracing

Browse Source
This commit is contained in:
smallnest
2019-01-25 17:03:25 +08:00
parent 6f029cdf1f
commit e67be6a4de
8 changed files with 263 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
server/server.go
View File

@@ -53,6 +53,8 @@ var (
StartRequestContextKey = &contextKey{"start-parse-request"}
// StartSendRequestContextKey records the start time
StartSendRequestContextKey = &contextKey{"start-send-request"}
// TagContextKey is used to record extra info in handling services. Its value is a map[string]interface{}
TagContextKey = &contextKey{"service-tag"}
)
// Server is rpcx server that use TCP or UDP.
@@ -132,7 +134,7 @@ func (s *Server) ActiveClientConn() []net.Conn {
//
// servicePath, serviceMethod, metadata can be set to zero values.
func (s *Server) SendMessage(conn net.Conn, servicePath, serviceMethod string, metadata map[string]string, data []byte) error {
ctx := context.WithValue(context.Background(), StartSendRequestContextKey, time.Now().UnixNano())
ctx := share.WithValue(context.Background(), StartSendRequestContextKey, time.Now().UnixNano())
s.Plugins.DoPreWriteRequest(ctx)
req := protocol.GetPooledMsg()
@@ -345,7 +347,8 @@ func (s *Server) serveConn(conn net.Conn) {
conn.SetReadDeadline(t0.Add(s.readTimeout))
}
ctx := context.WithValue(context.Background(), RemoteConnContextKey, conn)
ctx := share.WithValue(context.Background(), RemoteConnContextKey, conn)
req, err := s.readRequest(ctx, r)
if err != nil {
if err == io.EOF {
@@ -362,7 +365,7 @@ func (s *Server) serveConn(conn net.Conn) {
conn.SetWriteDeadline(t0.Add(s.writeTimeout))
}
ctx = context.WithValue(ctx, StartRequestContextKey, time.Now().UnixNano())
ctx = share.WithLocalValue(ctx, StartRequestContextKey, time.Now().UnixNano())
if !req.IsHeartbeat() {
err = s.auth(ctx, req)
}
@@ -400,9 +403,11 @@ func (s *Server) serveConn(conn net.Conn) {
}
resMetadata := make(map[string]string)
newCtx := context.WithValue(context.WithValue(ctx, share.ReqMetaDataKey, req.Metadata),
newCtx := share.WithLocalValue(share.WithLocalValue(ctx, share.ReqMetaDataKey, req.Metadata),
share.ResMetaDataKey, resMetadata)
s.Plugins.DoPreHandleRequest(newCtx, req)
res, err := s.handleRequest(newCtx, req)
if err != nil {