gitlab-com-seaweedfs/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2024-01-19 02:48:24 +00:00
Code Issues Projects Releases Wiki Activity

rename

Browse source
This commit is contained in:
Chris Lu 2020-05-12 22:55:55 -07:00
parent 25257acd51
commit ca4017dd87
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
weed/messaging/broker/topic_manager.go
View file

@ -25,7 +25,7 @@ func (tp *TopicPartition) String() string {
return fmt.Sprintf(TopicPartitionFmt, tp.Namespace, tp.Topic, tp.Partition)
}
type TopicCursor struct {
type TopicControl struct {
sync.Mutex
cond *sync.Cond
subscriberCount int
@ -36,18 +36,18 @@ type TopicCursor struct {
type TopicManager struct {
sync.Mutex
topicCursors map[TopicPartition]*TopicCursor
topicCursors map[TopicPartition]*TopicControl
broker *MessageBroker
}
func NewTopicManager(messageBroker *MessageBroker) *TopicManager {
return &TopicManager{
topicCursors: make(map[TopicPartition]*TopicCursor),
topicCursors: make(map[TopicPartition]*TopicControl),
broker: messageBroker,
}
}
func (tm *TopicManager) buildLogBuffer(tl *TopicCursor, tp TopicPartition, topicConfig *messaging_pb.TopicConfiguration) *log_buffer.LogBuffer {
func (tm *TopicManager) buildLogBuffer(tc *TopicControl, tp TopicPartition, topicConfig *messaging_pb.TopicConfiguration) *log_buffer.LogBuffer {
flushFn := func(startTime, stopTime time.Time, buf []byte) {
@ -69,29 +69,29 @@ func (tm *TopicManager) buildLogBuffer(tl *TopicCursor, tp TopicPartition, topic
}
}
logBuffer := log_buffer.NewLogBuffer(time.Minute, flushFn, func() {
tl.subscriptions.NotifyAll()
tc.subscriptions.NotifyAll()
})
return logBuffer
}
func (tm *TopicManager) RequestLock(partition TopicPartition, topicConfig *messaging_pb.TopicConfiguration, isPublisher bool) *TopicCursor {
func (tm *TopicManager) RequestLock(partition TopicPartition, topicConfig *messaging_pb.TopicConfiguration, isPublisher bool) *TopicControl {
tm.Lock()
defer tm.Unlock()
lock, found := tm.topicCursors[partition]
tc, found := tm.topicCursors[partition]
if !found {
lock = &TopicCursor{}
tm.topicCursors[partition] = lock
lock.subscriptions = NewTopicPartitionSubscriptions()
lock.logBuffer = tm.buildLogBuffer(lock, partition, topicConfig)
tc = &TopicControl{}
tm.topicCursors[partition] = tc
tc.subscriptions = NewTopicPartitionSubscriptions()
tc.logBuffer = tm.buildLogBuffer(tc, partition, topicConfig)
}
if isPublisher {
lock.publisherCount++
tc.publisherCount++
} else {
lock.subscriberCount++
tc.subscriberCount++
}
return lock
return tc
}
func (tm *TopicManager) ReleaseLock(partition TopicPartition, isPublisher bool) {