1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
# -*- coding: utf-8 -*-
"""This is not part of the test suite.
"""
try:
import ketama
except Exception:
ketama = None
from tempfile import NamedTemporaryFile
from time import time
from uhashring import HashRing
num = 1000000
print("running {} key generation comparison".format(num))
# ketama C binding
if ketama:
with NamedTemporaryFile(prefix="benchmark_") as ketama_config_file:
ketama_config_file.write("127.0.0.1:11211\t600\n")
ketama_config_file.write("127.0.0.1:11212\t400\n")
ketama_config_file.flush()
kt = ketama.Continuum(ketama_config_file.name)
pt = time()
for i in range(num):
key = "myval-{}".format(i)
kt.get_server(key)
print("ketama took {} s".format(time() - pt))
# pure python implementation
ring = HashRing(
nodes={"127.0.0.1:11211": 600, "127.0.0.1:11212": 400},
replicas=4,
vnodes=40,
compat=True,
)
pt = time()
for i in range(num):
key = "myval-{}".format(i)
ring.get_server(key)
print("HashRing took {} s".format(time() - pt))
|
