mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-05-31 15:51:39 +02:00
28 lines
752 B
Go
28 lines
752 B
Go
package broker
|
|
|
|
import (
|
|
"context"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
)
|
|
|
|
func (b *MessageQueueBroker) BalanceTopics(ctx context.Context, request *mq_pb.BalanceTopicsRequest) (resp *mq_pb.BalanceTopicsResponse, err error) {
|
|
if !b.isLockOwner() {
|
|
proxyErr := b.withBrokerClient(false, pb.ServerAddress(b.lockAsBalancer.LockOwner()), func(client mq_pb.SeaweedMessagingClient) error {
|
|
resp, err = client.BalanceTopics(ctx, request)
|
|
return nil
|
|
})
|
|
if proxyErr != nil {
|
|
return nil, proxyErr
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
ret := &mq_pb.BalanceTopicsResponse{}
|
|
|
|
actions := b.Balancer.BalancePublishers()
|
|
err = b.Balancer.ExecuteBalanceAction(actions, b.grpcDialOption)
|
|
|
|
return ret, err
|
|
}
|