mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package broker
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/messaging_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
)
|
|
|
|
func (broker *MessageBroker) Subscribe(stream messaging_pb.SeaweedMessaging_SubscribeServer) error {
|
|
|
|
// process initial request
|
|
in, err := stream.Recv()
|
|
if err == io.EOF {
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
subscriberId := in.Init.SubscriberId
|
|
println("+ subscriber:", subscriberId)
|
|
defer println("- subscriber:", subscriberId)
|
|
|
|
// TODO look it up
|
|
topicConfig := &messaging_pb.TopicConfiguration{
|
|
}
|
|
|
|
// get lock
|
|
tp := TopicPartition{
|
|
Namespace: in.Init.Namespace,
|
|
Topic: in.Init.Topic,
|
|
Partition: in.Init.Partition,
|
|
}
|
|
lock := broker.topicLocks.RequestLock(tp, topicConfig, false)
|
|
defer broker.topicLocks.ReleaseLock(tp, false)
|
|
|
|
lastReadTime := time.Now()
|
|
switch in.Init.StartPosition {
|
|
case messaging_pb.SubscriberMessage_InitMessage_TIMESTAMP:
|
|
lastReadTime = time.Unix(0, in.Init.TimestampNs)
|
|
case messaging_pb.SubscriberMessage_InitMessage_LATEST:
|
|
case messaging_pb.SubscriberMessage_InitMessage_EARLIEST:
|
|
}
|
|
|
|
// how to process each message
|
|
// an error returned will end the subscription
|
|
eachMessageFn := func(m *messaging_pb.Message) error {
|
|
err := stream.Send(&messaging_pb.BrokerMessage{
|
|
Data: m,
|
|
})
|
|
if err != nil {
|
|
glog.V(0).Infof("=> subscriber %v: %+v", subscriberId, err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// loop through all messages
|
|
for {
|
|
|
|
_, buf := lock.logBuffer.ReadFromBuffer(lastReadTime)
|
|
|
|
for pos := 0; pos+4 < len(buf); {
|
|
|
|
size := util.BytesToUint32(buf[pos : pos+4])
|
|
entryData := buf[pos+4 : pos+4+int(size)]
|
|
|
|
logEntry := &filer_pb.LogEntry{}
|
|
if err = proto.Unmarshal(entryData, logEntry); err != nil {
|
|
glog.Errorf("unexpected unmarshal messaging_pb.Message: %v", err)
|
|
pos += 4 + int(size)
|
|
continue
|
|
}
|
|
|
|
m := &messaging_pb.Message{}
|
|
if err = proto.Unmarshal(logEntry.Data, m); err != nil {
|
|
glog.Errorf("unexpected unmarshal messaging_pb.Message: %v", err)
|
|
pos += 4 + int(size)
|
|
continue
|
|
}
|
|
|
|
// fmt.Printf("sending : %d : %s\n", len(m.Value), string(m.Value))
|
|
if err = eachMessageFn(m); err != nil {
|
|
return err
|
|
}
|
|
|
|
lastReadTime = time.Unix(0, m.Timestamp)
|
|
pos += 4 + int(size)
|
|
}
|
|
|
|
lock.Mutex.Lock()
|
|
lock.cond.Wait()
|
|
lock.Mutex.Unlock()
|
|
}
|
|
|
|
}
|