mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 03:13:22 +08:00
5f2f8bdc8b
* [skip ci](shards): export MAX_WORKERS as configurable parameter Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): skip mishards .env git info Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): support more robust static discovery host configuration Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): update static provider that terminate server if connection to downstream server error during startup Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): add topology.py Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): add connection pool Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): add topology test Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): refactory using topo Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): refactory static discovery using topo Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): refactory kubernetes discovery using topo Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): add more test for connection pool Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): export 19541 and 19542 for all_in_one demo Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): check version on new connection Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): mock connections Signed-off-by: peng.xu <peng.xu@zilliz.com> * [skip ci](shards): update tests Signed-off-by: peng.xu <peng.xu@zilliz.com>
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
import logging
|
|
from jaeger_client import Config
|
|
from grpc_opentracing.grpcext import intercept_server
|
|
from grpc_opentracing import open_tracing_server_interceptor
|
|
from tracer import Tracer
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
PLUGIN_NAME = __file__
|
|
|
|
|
|
class JaegerFactory:
|
|
name = 'jaeger'
|
|
@classmethod
|
|
def Create(cls, plugin_config, **kwargs):
|
|
tracing_config = plugin_config.TRACING_CONFIG
|
|
span_decorator = kwargs.pop('span_decorator', None)
|
|
service_name = plugin_config.TRACING_SERVICE_NAME
|
|
validate = plugin_config.TRACING_VALIDATE
|
|
config = Config(config=tracing_config,
|
|
service_name=service_name,
|
|
validate=validate)
|
|
|
|
tracer = config.initialize_tracer()
|
|
tracer_interceptor = open_tracing_server_interceptor(
|
|
tracer,
|
|
log_payloads=plugin_config.TRACING_LOG_PAYLOAD,
|
|
span_decorator=span_decorator)
|
|
jaeger_logger = logging.getLogger('jaeger_tracing')
|
|
jaeger_logger.setLevel(logging.ERROR)
|
|
|
|
return Tracer(tracer, tracer_interceptor, intercept_server)
|
|
|
|
|
|
def setup(app):
|
|
logger.info('Plugin \'{}\' Installed In Package: {}'.format(PLUGIN_NAME, app.plugin_package_name))
|
|
app.on_plugin_setup(JaegerFactory)
|