refactor

weed/mq/broker/broker_grpc_pub_follow.go

@@ -52,7 +52,7 @@ func (b *MessageQueueBroker) PublishFollowMe(stream mq_pb.SeaweedMessaging_Publi
 
 			// TODO: change this to DataMessage
 			// log the message
-			logBuffer.AddToBuffer(dataMessage.Key, dataMessage.Value, dataMessage.TsNs)
+			logBuffer.AddToBuffer(dataMessage)
 
 			// send back the ack
 			if err := stream.Send(&mq_pb.PublishFollowMeResponse{

weed/mq/topic/local_partition.go

@@ -52,7 +52,7 @@ func NewLocalPartition(partition Partition, logFlushFn log_buffer.LogFlushFuncTy
 }
 
 func (p *LocalPartition) Publish(message *mq_pb.DataMessage) error {
-	p.LogBuffer.AddToBuffer(message.Key, message.Value, time.Now().UnixNano())
+	p.LogBuffer.AddToBuffer(message)
 
 	// maybe send to the follower
 	if p.followerStream != nil {