新增：etcd的链路追踪入口

farseer-go · Oct 19, 2023 · ef627d5 · ef627d5
1 parent d0683a1
commit ef627d5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/client.go b/client.go
@@ -161,7 +161,9 @@ func (receiver *client) Watch(ctx context.Context, key string, watchFunc func(ev
 					Type: event.Type.String(),
 					Kv:   newValue(event.Kv, &response.Header),
 				}
+				entryWatchKey := receiver.traceManager.EntryWatchKey(key)
 				watchFunc(watchEvent)
+				entryWatchKey.End()
 			}
 		}
 		flog.Info("退出了")
@@ -178,7 +180,9 @@ func (receiver *client) WatchPrefixKey(ctx context.Context, prefixKey string, wa
 					Type: event.Type.String(),
 					Kv:   newValue(event.Kv, &response.Header),
 				}
+				entryWatchKey := receiver.traceManager.EntryWatchKey(prefixKey)
 				watchFunc(watchEvent)
+				entryWatchKey.End()
 			}
 		}
 		flog.Info("退出了")