feat: support multi unit in trace config (#1233)

Author: AnnaYue

consumer/interceptor.go

@@ -20,6 +20,7 @@ package consumer
 import (
 	"context"
 	"fmt"
+	"reflect"
 	"time"
 
 	"github.com/apache/rocketmq-client-go/v2/internal"
@@ -40,7 +41,7 @@ func WithTrace(traceCfg *primitive.TraceConfig) Option {
 }
 
 func newTraceInterceptor(dispatcher internal.TraceDispatcher) primitive.Interceptor {
-	if dispatcher != nil {
+	if dispatcher != nil && !reflect.ValueOf(dispatcher).IsNil() {
 		dispatcher.Start()
 	}
 

internal/trace.go

@@ -275,6 +275,7 @@ func NewTraceDispatcher(traceCfg *primitive.TraceConfig) *traceDispatcher {
 
 	cliOp := DefaultClientOptions()
 	cliOp.GroupName = traceCfg.GroupName
+	cliOp.UnitName = traceCfg.UnitName
 	cliOp.NameServerAddrs = traceCfg.NamesrvAddrs
 	cliOp.InstanceName = "INNER_TRACE_CLIENT_DEFAULT"
 	cliOp.RetryTimes = 0

primitive/trace.go

@@ -21,6 +21,7 @@ package primitive
 type TraceConfig struct {
 	TraceTopic   string
 	GroupName    string
+	UnitName     string
 	Access       AccessChannel
 	NamesrvAddrs []string
 	Resolver     NsResolver