mirror of https://github.com/milvus-io/milvus.git
update for better test
parent
43bc2cc60c
commit
3ddd181dd2
|
@ -20,7 +20,7 @@ def app(request):
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def started_app(app):
|
def started_app(app):
|
||||||
app.on_pre_run()
|
app.on_pre_run()
|
||||||
app.start(app.port)
|
app.start(settings.SERVER_TEST_PORT)
|
||||||
|
|
||||||
yield app
|
yield app
|
||||||
|
|
||||||
|
|
|
@ -91,7 +91,7 @@ class Server:
|
||||||
handler_class(tracer=self.tracer,
|
handler_class(tracer=self.tracer,
|
||||||
router=self.router), self.server_impl)
|
router=self.router), self.server_impl)
|
||||||
self.server_impl.add_insecure_port("[::]:{}".format(
|
self.server_impl.add_insecure_port("[::]:{}".format(
|
||||||
str(port or self._port)))
|
str(port or self.port)))
|
||||||
self.server_impl.start()
|
self.server_impl.start()
|
||||||
|
|
||||||
def run(self, port):
|
def run(self, port):
|
||||||
|
|
|
@ -25,6 +25,7 @@ TIMEOUT = env.int('TIMEOUT', 60)
|
||||||
MAX_RETRY = env.int('MAX_RETRY', 3)
|
MAX_RETRY = env.int('MAX_RETRY', 3)
|
||||||
|
|
||||||
SERVER_PORT = env.int('SERVER_PORT', 19530)
|
SERVER_PORT = env.int('SERVER_PORT', 19530)
|
||||||
|
SERVER_TEST_PORT = env.int('SERVER_TEST_PORT', 19530)
|
||||||
WOSERVER = env.str('WOSERVER')
|
WOSERVER = env.str('WOSERVER')
|
||||||
|
|
||||||
SD_PROVIDER_SETTINGS = None
|
SD_PROVIDER_SETTINGS = None
|
||||||
|
@ -36,11 +37,13 @@ if SD_PROVIDER == 'Kubernetes':
|
||||||
in_cluster=env.bool('SD_IN_CLUSTER', False),
|
in_cluster=env.bool('SD_IN_CLUSTER', False),
|
||||||
poll_interval=env.int('SD_POLL_INTERVAL', 5),
|
poll_interval=env.int('SD_POLL_INTERVAL', 5),
|
||||||
pod_patt=env.str('SD_ROSERVER_POD_PATT', ''),
|
pod_patt=env.str('SD_ROSERVER_POD_PATT', ''),
|
||||||
label_selector=env.str('SD_LABEL_SELECTOR', ''))
|
label_selector=env.str('SD_LABEL_SELECTOR', ''),
|
||||||
|
port=env.int('SD_PORT', 19530))
|
||||||
elif SD_PROVIDER == 'Static':
|
elif SD_PROVIDER == 'Static':
|
||||||
from sd.static_provider import StaticProviderSettings
|
from sd.static_provider import StaticProviderSettings
|
||||||
SD_PROVIDER_SETTINGS = StaticProviderSettings(
|
SD_PROVIDER_SETTINGS = StaticProviderSettings(
|
||||||
hosts=env.list('SD_STATIC_HOSTS', []))
|
hosts=env.list('SD_STATIC_HOSTS', []),
|
||||||
|
port=env.int('SD_STATIC_PORT', 19530))
|
||||||
|
|
||||||
# TESTING_WOSERVER = env.str('TESTING_WOSERVER', 'tcp://127.0.0.1:19530')
|
# TESTING_WOSERVER = env.str('TESTING_WOSERVER', 'tcp://127.0.0.1:19530')
|
||||||
|
|
||||||
|
|
|
@ -226,12 +226,13 @@ class EventHandler(threading.Thread):
|
||||||
|
|
||||||
class KubernetesProviderSettings:
|
class KubernetesProviderSettings:
|
||||||
def __init__(self, namespace, pod_patt, label_selector, in_cluster,
|
def __init__(self, namespace, pod_patt, label_selector, in_cluster,
|
||||||
poll_interval, **kwargs):
|
poll_interval, port=None, **kwargs):
|
||||||
self.namespace = namespace
|
self.namespace = namespace
|
||||||
self.pod_patt = pod_patt
|
self.pod_patt = pod_patt
|
||||||
self.label_selector = label_selector
|
self.label_selector = label_selector
|
||||||
self.in_cluster = in_cluster
|
self.in_cluster = in_cluster
|
||||||
self.poll_interval = poll_interval
|
self.poll_interval = poll_interval
|
||||||
|
self.port = int(port) if port else 19530
|
||||||
|
|
||||||
|
|
||||||
@singleton
|
@singleton
|
||||||
|
@ -245,6 +246,7 @@ class KubernetesProvider(object):
|
||||||
self.label_selector = settings.label_selector
|
self.label_selector = settings.label_selector
|
||||||
self.in_cluster = settings.in_cluster
|
self.in_cluster = settings.in_cluster
|
||||||
self.poll_interval = settings.poll_interval
|
self.poll_interval = settings.poll_interval
|
||||||
|
self.port = settings.port
|
||||||
self.kwargs = kwargs
|
self.kwargs = kwargs
|
||||||
self.queue = queue.Queue()
|
self.queue = queue.Queue()
|
||||||
|
|
||||||
|
@ -279,7 +281,7 @@ class KubernetesProvider(object):
|
||||||
**kwargs)
|
**kwargs)
|
||||||
|
|
||||||
def add_pod(self, name, ip):
|
def add_pod(self, name, ip):
|
||||||
self.conn_mgr.register(name, 'tcp://{}:19530'.format(ip))
|
self.conn_mgr.register(name, 'tcp://{}:{}'.format(ip, self.port))
|
||||||
|
|
||||||
def delete_pod(self, name):
|
def delete_pod(self, name):
|
||||||
self.conn_mgr.unregister(name)
|
self.conn_mgr.unregister(name)
|
||||||
|
|
|
@ -9,8 +9,9 @@ from sd import ProviderManager
|
||||||
|
|
||||||
|
|
||||||
class StaticProviderSettings:
|
class StaticProviderSettings:
|
||||||
def __init__(self, hosts):
|
def __init__(self, hosts, port=None):
|
||||||
self.hosts = hosts
|
self.hosts = hosts
|
||||||
|
self.port = int(port) if port else 19530
|
||||||
|
|
||||||
|
|
||||||
@singleton
|
@singleton
|
||||||
|
@ -21,6 +22,7 @@ class KubernetesProvider(object):
|
||||||
def __init__(self, settings, conn_mgr, **kwargs):
|
def __init__(self, settings, conn_mgr, **kwargs):
|
||||||
self.conn_mgr = conn_mgr
|
self.conn_mgr = conn_mgr
|
||||||
self.hosts = [socket.gethostbyname(host) for host in settings.hosts]
|
self.hosts = [socket.gethostbyname(host) for host in settings.hosts]
|
||||||
|
self.port = settings.port
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
for host in self.hosts:
|
for host in self.hosts:
|
||||||
|
@ -31,7 +33,7 @@ class KubernetesProvider(object):
|
||||||
self.delete_pod(host)
|
self.delete_pod(host)
|
||||||
|
|
||||||
def add_pod(self, name, ip):
|
def add_pod(self, name, ip):
|
||||||
self.conn_mgr.register(name, 'tcp://{}:19530'.format(ip))
|
self.conn_mgr.register(name, 'tcp://{}:{}'.format(ip, self.port))
|
||||||
|
|
||||||
def delete_pod(self, name):
|
def delete_pod(self, name):
|
||||||
self.conn_mgr.unregister(name)
|
self.conn_mgr.unregister(name)
|
||||||
|
|
Loading…
Reference in New Issue