You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.3 KiB
43 lines
1.3 KiB
3 years ago
|
import time
|
||
|
|
|
||
|
|
import jury
|
||
|
|
import redis
|
||
|
|
|
||
|
|
|
||
|
|
class Redis:
|
||
|
|
def __init__(self, args=None):
|
||
|
|
self.args = args
|
||
|
|
if args is None:
|
||
|
|
self.args = []
|
||
|
|
|
||
|
|
self.redis = None
|
||
|
|
self.host = "127.0.0.1"
|
||
|
|
self.port = jury.get_free_port()
|
||
|
|
self.dir = f"{jury.get_case_dir()}/redis_{self.port}"
|
||
|
|
self.server = jury.Launcher(args=["redis-server", "--port", str(self.port)] + self.args, work_dir=self.dir)
|
||
|
|
self.__wait_start()
|
||
|
|
jury.log_yellow(f"Redis started(pid={self.server.get_pid()}). redis-cli -p {self.port}")
|
||
|
|
|
||
|
|
def get_address(self):
|
||
|
|
return f"127.0.0.1:{self.port}"
|
||
|
|
|
||
|
|
def __wait_start(self, timeout=5):
|
||
|
|
timer = jury.Timer()
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
self.__create_client()
|
||
|
|
self.client.ping()
|
||
|
|
break
|
||
|
|
except redis.exceptions.ConnectionError:
|
||
|
|
time.sleep(0.01)
|
||
|
|
except redis.exceptions.ResponseError as e:
|
||
|
|
if str(e) == "LOADING Redis is loading the dataset in memory":
|
||
|
|
time.sleep(0.01)
|
||
|
|
else:
|
||
|
|
raise e
|
||
|
|
if timer.elapsed_time() > timeout:
|
||
|
|
raise Exception(f"tair start timeout, {self.dir}")
|
||
|
|
|
||
|
|
def __create_client(self):
|
||
|
|
self.client = redis.Redis(port=self.port, single_connection_client=True)
|