新增：事件驱动和链路追踪

farseer-go · Jan 16, 2024 · e70f466 · e70f466
1 parent c927539
commit e70f466
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 5 deletions.
diff --git a/trace/emptyManager.go b/trace/emptyManager.go
@@ -10,11 +10,12 @@ func (*EmptyManager) EntryWebApi(domain string, path string, method string, cont
 func (*EmptyManager) EntryMqConsumer(server string, queueName string, routingKey string) ITraceContext {
 	return &emptyTraceContext{}
 }
-
 func (*EmptyManager) EntryQueueConsumer(queueName, subscribeName string) ITraceContext {
 	return &emptyTraceContext{}
 }
-
+func (*EmptyManager) EntryEventConsumer(eventName, subscribeName string) ITraceContext {
+	return &emptyTraceContext{}
+}
 func (*EmptyManager) EntryTask(taskName string) ITraceContext {
 	return &emptyTraceContext{}
 }
@@ -41,7 +42,8 @@ func (*EmptyManager) TraceElasticsearch(method string, IndexName string, Aliases
 func (*EmptyManager) TraceEtcd(method string, key string, leaseID int64) ITraceDetail {
 	return &emptyTraceDetail{}
 }
-func (*EmptyManager) TraceHand(name string) ITraceDetail { return &emptyTraceDetail{} }
+func (*EmptyManager) TraceHand(name string) ITraceDetail              { return &emptyTraceDetail{} }
+func (*EmptyManager) TraceEventPublish(eventName string) ITraceDetail { return &emptyTraceDetail{} }
 func (*EmptyManager) TraceMqSend(method string, server string, exchange string, routingKey string) ITraceDetail {
 	return &emptyTraceDetail{}
 }

diff --git a/trace/eumCallType/enum.go b/trace/eumCallType/enum.go
@@ -11,6 +11,7 @@ const (
 	Elasticsearch             // Elasticsearch
 	Etcd                      // Etcd
 	Hand                      // Hand
+	EventPublish              // EventPublish
 )
 
 func (receiver Enum) ToString() string {
@@ -31,6 +32,8 @@ func (receiver Enum) ToString() string {
 		return "Hand"
 	case Etcd:
 		return "Etcd"
+	case EventPublish:
+		return "EventPublish"
 	}
 	return ""
 }
diff --git a/trace/iManager.go b/trace/iManager.go
@@ -15,6 +15,8 @@ type IManager interface {
 	TraceEtcd(method string, key string, leaseID int64) ITraceDetail
 	// TraceHand 手动埋点
 	TraceHand(name string) ITraceDetail
+	// TraceEventPublish 事件发布
+	TraceEventPublish(eventName string) ITraceDetail
 	// TraceMqSend send埋点
 	TraceMqSend(method string, server string, exchange string, routingKey string) ITraceDetail
 	// TraceMq open、create埋点
@@ -23,6 +25,8 @@ type IManager interface {
 	EntryMqConsumer(server string, queueName string, routingKey string) ITraceContext
 	// EntryQueueConsumer 创建Queue消费入口
 	EntryQueueConsumer(queueName, subscribeName string) ITraceContext
+	// EntryEventConsumer 创建Event消费入口
+	EntryEventConsumer(eventName, subscribeName string) ITraceContext
 	// TraceRedis Redis埋点
 	TraceRedis(method string, key string, field string) ITraceDetail
 	// TraceHttp http埋点

diff --git a/trace/trace_test.go b/trace/trace_test.go
@@ -14,15 +14,15 @@ func TestTrace(t *testing.T) {
 	emptyManager.TraceElasticsearch("", "", "")
 	emptyManager.TraceHand("")
 	emptyManager.TraceMq("", "", "")
-	emptyManager.EntryFSchedule("", 0, 0)
+	emptyManager.EntryFSchedule("", 0, nil)
 	emptyManager.EntryMqConsumer("", "", "")
 	emptyManager.EntryQueueConsumer("", "")
+	emptyManager.EntryEventConsumer("", "")
 	emptyManager.EntryTask("")
 	emptyManager.EntryWatchKey("")
 	emptyManager.EntryWebApi("", "", "", "", nil, "", "")
 	emptyManager.TraceDatabaseOpen("", "")
 	emptyManager.TraceHttp("", "")
-	emptyManager.TraceKeyLocation("")
 	emptyManager.TraceMqSend("", "", "", "")
 	emptyManager.TraceRedis("", "", "")