Browse Source

rename

pull/1318/head
Chris Lu 5 years ago
parent
commit
25257acd51
  1. 18
      weed/messaging/broker/topic_manager.go

18
weed/messaging/broker/topic_manager.go

@ -36,14 +36,14 @@ type TopicCursor struct {
type TopicManager struct { type TopicManager struct {
sync.Mutex sync.Mutex
topicControls map[TopicPartition]*TopicCursor
broker *MessageBroker
topicCursors map[TopicPartition]*TopicCursor
broker *MessageBroker
} }
func NewTopicManager(messageBroker *MessageBroker) *TopicManager { func NewTopicManager(messageBroker *MessageBroker) *TopicManager {
return &TopicManager{ return &TopicManager{
topicControls: make(map[TopicPartition]*TopicCursor),
broker: messageBroker,
topicCursors: make(map[TopicPartition]*TopicCursor),
broker: messageBroker,
} }
} }
@ -79,10 +79,10 @@ func (tm *TopicManager) RequestLock(partition TopicPartition, topicConfig *messa
tm.Lock() tm.Lock()
defer tm.Unlock() defer tm.Unlock()
lock, found := tm.topicControls[partition]
lock, found := tm.topicCursors[partition]
if !found { if !found {
lock = &TopicCursor{} lock = &TopicCursor{}
tm.topicControls[partition] = lock
tm.topicCursors[partition] = lock
lock.subscriptions = NewTopicPartitionSubscriptions() lock.subscriptions = NewTopicPartitionSubscriptions()
lock.logBuffer = tm.buildLogBuffer(lock, partition, topicConfig) lock.logBuffer = tm.buildLogBuffer(lock, partition, topicConfig)
} }
@ -98,7 +98,7 @@ func (tm *TopicManager) ReleaseLock(partition TopicPartition, isPublisher bool)
tm.Lock() tm.Lock()
defer tm.Unlock() defer tm.Unlock()
lock, found := tm.topicControls[partition]
lock, found := tm.topicCursors[partition]
if !found { if !found {
return return
} }
@ -108,7 +108,7 @@ func (tm *TopicManager) ReleaseLock(partition TopicPartition, isPublisher bool)
lock.subscriberCount-- lock.subscriberCount--
} }
if lock.subscriberCount <= 0 && lock.publisherCount <= 0 { if lock.subscriberCount <= 0 && lock.publisherCount <= 0 {
delete(tm.topicControls, partition)
delete(tm.topicCursors, partition)
lock.logBuffer.Shutdown() lock.logBuffer.Shutdown()
} }
} }
@ -117,7 +117,7 @@ func (tm *TopicManager) ListTopicPartitions() (tps []TopicPartition) {
tm.Lock() tm.Lock()
defer tm.Unlock() defer tm.Unlock()
for k := range tm.topicControls {
for k := range tm.topicCursors {
tps = append(tps, k) tps = append(tps, k)
} }
return return

Loading…
Cancel
Save