diff options
Diffstat (limited to 'core/hash.go')
| -rw-r--r-- | core/hash.go | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/core/hash.go b/core/hash.go new file mode 100644 index 0000000..af15546 --- /dev/null +++ b/core/hash.go | |||
| @@ -0,0 +1,19 @@ | |||
| 1 | package core | ||
| 2 | |||
| 3 | import "hash/fnv" | ||
| 4 | |||
| 5 | const DefaultMaxParallelism = 128 | ||
| 6 | |||
| 7 | func AssignToKeyGroup(key string, maxParallelism int) int { | ||
| 8 | |||
| 9 | if maxParallelism <= 0 { | ||
| 10 | maxParallelism = DefaultMaxParallelism | ||
| 11 | } | ||
| 12 | |||
| 13 | // hash | ||
| 14 | h := fnv.New32a() | ||
| 15 | h.Write([]byte(key)) | ||
| 16 | hash := h.Sum32() | ||
| 17 | |||
| 18 | return int(hash % uint32(maxParallelism)) | ||
| 19 | } | ||
