made NumShards a global const

2 files changed, 4 insertions, 4 deletions

diff --git a/main.go b/main.go
index ca8ab44..2dc5b12 100644
--- a/main.go
+++ b/main.go
@@ -9,6 +9,7 @@ import (
 	"strings"
 )
 
+const NumShards = 16
 
 var db RedisDB
 
@@ -41,10 +42,9 @@ var commandRegistry = map[string]handlerFunc{
 }
 
 func main() {
-	numShards := 16
-	shards := make([]*Shard, numShards)
+	shards := make([]*Shard, NumShards)
 
-	for i := 0; i < numShards; i++ {
+	for i := 0; i < NumShards; i++ {
     	shards[i] = &Shard{
         data: make(map[string]Item),
 		id: i,
diff --git a/shards.go b/shards.go
index cba4dd4..c69f920 100644
--- a/shards.go
+++ b/shards.go
@@ -15,7 +15,7 @@ func (db *RedisDB) getShard(key string)*Shard{
 	hash.Write([]byte(key))
 	val := hash.Sum64()
 
-	return db.shards[val%16]
+	return db.shards[val%NumShards]
 }
 
 func (db *RedisDB) Execute(key string, fn func(*Shard) interface{}) interface{} {