From e8fd63cb1072a482c4db53185db7dc0e2ed43fd7 Mon Sep 17 00:00:00 2001
From: rutherford-jasper <jrutherford@featurebase.com>
Date: Mon, 10 Apr 2023 10:50:09 -0500
Subject: [PATCH] Fixed function rename :)

---
 dax/controller/balancer.go   | 6 +++---
 dax/controller/controller.go | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

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