diff --git a/dax/controller/balancer.go b/dax/controller/balancer.go index 09d9ba76f..54fb5ea9c 100644 --- a/dax/controller/balancer.go +++ b/dax/controller/balancer.go @@ -47,8 +47,8 @@ type Balancer interface { // Nodes returns all nodes known by the Balancer. Nodes(tx dax.Transaction) ([]*dax.Node, error) - // GetDatabaseWorkerCount returns the number of workers - GetDatabaseWorkerCount(tx dax.Transaction, qdbid dax.QualifiedDatabaseID) (int, error) + // WorkerCount returns the number of workers + WorkerCount(tx dax.Transaction, qdbid dax.QualifiedDatabaseID) (int, error) } // Ensure type implements interface. @@ -97,6 +97,6 @@ func (b *NopBalancer) ReadNode(tx dax.Transaction, addr dax.Address) (*dax.Node, func (b *NopBalancer) Nodes(tx dax.Transaction) ([]*dax.Node, error) { return []*dax.Node{}, nil } -func (b *NopBalancer) GetDatabaseWorkerCount(tx dax.Transaction, qdbid dax.QualifiedDatabaseID) (int, error) { +func (b *NopBalancer) WorkerCount(tx dax.Transaction, qdbid dax.QualifiedDatabaseID) (int, error) { return 0, nil } diff --git a/dax/controller/controller.go b/dax/controller/controller.go index 6381e0cf2..de5f97ebb 100644 --- a/dax/controller/controller.go +++ b/dax/controller/controller.go @@ -793,7 +793,7 @@ func (c *Controller) GetDatabaseWorkerCount(ctx context.Context, qdbid dax.Quali } defer tx.Rollback() - return c.Balancer.GetDatabaseWorkerCount(tx, qdbid) + return c.Balancer.WorkerCount(tx, qdbid) } // CreateTable adds a table to the schemar, and then sends directives to all