aboutsummaryrefslogtreecommitdiff
path: root/weed/mq/client/pub_client/publisher.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/mq/client/pub_client/publisher.go')
-rw-r--r--weed/mq/client/pub_client/publisher.go4
1 files changed, 2 insertions, 2 deletions
diff --git a/weed/mq/client/pub_client/publisher.go b/weed/mq/client/pub_client/publisher.go
index 7073457f3..bf1711e38 100644
--- a/weed/mq/client/pub_client/publisher.go
+++ b/weed/mq/client/pub_client/publisher.go
@@ -2,7 +2,7 @@ package pub_client
import (
"github.com/rdleal/intervalst/interval"
- "github.com/seaweedfs/seaweedfs/weed/mq/broker"
+ "github.com/seaweedfs/seaweedfs/weed/mq/balancer"
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
@@ -46,7 +46,7 @@ func (p *TopicPublisher) Connect(bootstrapBroker string) error {
func (p *TopicPublisher) Shutdown() error {
- if clients, found := p.partition2Broker.AllIntersections(0, broker.MaxPartitionCount); found {
+ if clients, found := p.partition2Broker.AllIntersections(0, balancer.MaxPartitionCount); found {
for _, client := range clients {
client.CloseSend()
}