seaweedfs/weed/mq/agent/agent_grpc_publish.go

package agent

import (
	"fmt"
	"github.com/seaweedfs/seaweedfs/weed/pb/mq_agent_pb"
	"golang.org/x/exp/slog"
	"time"
)

func (a *MessageQueueAgent) PublishRecordRequest(stream mq_agent_pb.SeaweedMessagingAgent_PublishRecordServer) error {
	m, err := stream.Recv()
	if err != nil {
		return err
	}
	a.publishersLock.RLock()
	publisherEntry, found := a.publishers[SessionId(m.SessionId)]
	a.publishersLock.RUnlock()
	if !found {
		return fmt.Errorf("session id %d not found", m.SessionId)
	}
	defer func() {
		if finishErr := publisherEntry.publisher.FinishPublish(); finishErr != nil {
			slog.Warn("failed to finish publish", "error", finishErr)
		}
		publisherEntry.lastActiveTsNs = time.Now().UnixNano()
	}()
	publisherEntry.lastActiveTsNs = 0

	if m.Value != nil {
		if err := publisherEntry.publisher.PublishRecord(m.Key, m.Value); err != nil {
			return err
		}
	}

	for {
		m, err := stream.Recv()
		if err != nil {
			return err
		}
		if m.Value == nil {
			continue
		}
		if err := publisherEntry.publisher.PublishRecord(m.Key, m.Value); err != nil {
			return err
		}
	}
}