优化生产数据的数据结构

define.go

@@ -13,7 +13,7 @@ package kafka
 //
 // Date : 4:13 下午 2021/9/21
 type ProducerData struct {
-	Data interface{} // 发送的数据
+	Data map[interface{}]interface{} // 发送的数据
 	Key  string                      // 分区key
 }
 

producer.go

@@ -141,14 +141,10 @@ func (p *Producer) buildMessage(data ProducerData) *sarama.ProducerMessage {
 		Key:       sarama.StringEncoder(data.Key),
 		Timestamp: time.Now(),
 	}
-	switch v := data.Data.(type) {
-	case string:
-		mes.Value = sarama.StringEncoder(v)
-	case []byte:
-		mes.Value = sarama.ByteEncoder(v)
-	default:
+	if nil == data.Data {
+		data.Data = make(map[interface{}]interface{})
+	}
 	byteData, _ := json.Marshal(data.Data)
 	mes.Value = sarama.ByteEncoder(byteData)
-	}
 	return mes
 }