|
import logging |
|
import time |
|
import random |
|
from os.path import expanduser, abspath |
|
from petrel_client.client_base import ClientBase |
|
from petrel_client.common.io_profile import profile |
|
from petrel_client.common.exception import ObjectNotFoundError |
|
from petrel_client.common.config import Config |
|
from petrel_client.common.log import init_log |
|
from petrel_client.common.io_profile import Profiler |
|
|
|
|
|
LOG = logging.getLogger(__name__) |
|
|
|
|
|
class Client(ClientBase): |
|
def __init__(self, conf_path, name, count_disp): |
|
conf_path = abspath(expanduser(conf_path)) |
|
config = Config(conf_path) |
|
self._default_config = config.default() |
|
|
|
init_log(self._default_config) |
|
LOG.info('init io_profile_test.Client, conf_path %s', conf_path) |
|
|
|
Profiler.set_default_conf(self._default_config) |
|
super(Client, self).__init__(name=name, count_disp=count_disp) |
|
|
|
@profile('get') |
|
def get(self, key): |
|
|
|
def not_found(): |
|
raise ObjectNotFoundError(key) |
|
|
|
def error(): |
|
raise Exception(key) |
|
|
|
def found(): |
|
return 'content' |
|
|
|
action = random.choice([found, not_found, error]) |
|
time.sleep(0.001) |
|
return action() |
|
|
|
@profile('put') |
|
def put(self, key, content): |
|
def normal(): |
|
return len(content) |
|
|
|
def error(): |
|
raise Exception(key) |
|
|
|
action = random.choice([normal, error]) |
|
return action() |
|
|
|
|
|
c = Client(conf_path='~/petreloss.conf', name='cluster1', count_disp=50) |
|
|
|
|
|
for _ in range(100): |
|
try: |
|
c.get('key') |
|
except Exception: |
|
pass |
|
|
