conversationID

pkg/common/kafka/producer.go

@@ -51,7 +51,7 @@ func NewKafkaProducer(addr []string, topic string) *Producer {
 	return &p
 }
 func GetMQHeaderWithContext(ctx context.Context) ([]sarama.RecordHeader, error) {
-	operationID, opUserID, platform, connID, err := mcontext.GetMustCtxInfo(ctx)
+	operationID, opUserID, platform, connID, err := mcontext.GetCtxInfos(ctx)
 	if err != nil {
 		return nil, err
 	}
@@ -69,7 +69,7 @@ func GetContextWithMQHeader(header []*sarama.RecordHeader) context.Context {
 	return mcontext.WithMustInfoCtx(values) // TODO
 }
 func (p *Producer) SendMessage(ctx context.Context, key string, m proto.Message) (int32, int64, error) {
-	log.ZDebug(ctx, "SendMessage", "key ", key, "msg", m)
+	log.ZDebug(ctx, "SendMessage", "key ", key, "msg", m, "topic", p.topic)
 	kMsg := &sarama.ProducerMessage{}
 	kMsg.Topic = p.topic
 	kMsg.Key = sarama.StringEncoder(key)

pkg/common/mcontext/ctx.go

@@ -2,6 +2,7 @@ package mcontext
 
 import (
 	"context"
+
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant"
 	"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
 )
@@ -109,8 +110,20 @@ func GetMustCtxInfo(ctx context.Context) (operationID, opUserID, platform, connI
 	}
 	connID, _ = ctx.Value(constant.ConnID).(string)
 	return
-
 }
+
+func GetCtxInfos(ctx context.Context) (operationID, opUserID, platform, connID string, err error) {
+	operationID, ok := ctx.Value(constant.OperationID).(string)
+	if !ok {
+		err = errs.ErrArgs.Wrap("ctx missing operationID")
+		return
+	}
+	opUserID, _ = ctx.Value(constant.OpUserID).(string)
+	platform, _ = ctx.Value(constant.OpUserPlatform).(string)
+	connID, _ = ctx.Value(constant.ConnID).(string)
+	return
+}
+
 func WithMustInfoCtx(values []string) context.Context {
 	ctx := context.Background()
 	for i, v := range values {