weed/mq/client/agent_client/agent_subscribe.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

package agent_client

import (
	"github.com/seaweedfs/seaweedfs/weed/pb/schema_pb"
)

func (a *SubscribeSession) SubscribeMessageRecord(
	onEachMessageFn func(key []byte, record *schema_pb.RecordValue),
	onCompletionFn func()) error {
	for {
		resp, err := a.stream.Recv()
		if err != nil {
			return err
		}
		onEachMessageFn(resp.Key, resp.Value)
	}
}