aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main.go6
-rw-r--r--shards.go2
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{} {