fix compilation

This commit is contained in:
chrislu 2023-09-19 16:22:41 -07:00
parent 440ebd7b6d
commit 411bdda08d
2 changed files with 4 additions and 4 deletions

View file

@ -2,13 +2,13 @@ package pub_client
import (
"fmt"
"github.com/seaweedfs/seaweedfs/weed/mq/broker"
"github.com/seaweedfs/seaweedfs/weed/mq/balancer"
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
"github.com/seaweedfs/seaweedfs/weed/util"
)
func (p *TopicPublisher) Publish(key, value []byte) error {
hashKey := util.HashToInt32(key) % broker.MaxPartitionCount
hashKey := util.HashToInt32(key) % balancer.MaxPartitionCount
if hashKey < 0 {
hashKey = -hashKey
}

View file

@ -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()
}