| # |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| import inspect |
| import os |
| import uuid |
| from typing import TYPE_CHECKING |
| |
| if TYPE_CHECKING: |
| from typing import List |
| |
| service_name = os.getenv('SW_AGENT_NAME') or 'Python Service Name' # type: str |
| service_instance = os.getenv('SW_AGENT_INSTANCE') or str(uuid.uuid1()).replace('-', '') # type: str |
| collector_address = os.getenv('SW_AGENT_COLLECTOR_BACKEND_SERVICES') or '127.0.0.1:11800' # type: str |
| protocol = (os.getenv('SW_AGENT_PROTOCOL') or 'grpc').lower() # type: str |
| authentication = os.getenv('SW_AGENT_AUTHENTICATION') # type: str |
| logging_level = os.getenv('SW_AGENT_LOGGING_LEVEL') or 'INFO' # type: str |
| disable_plugins = (os.getenv('SW_AGENT_DISABLE_PLUGINS') or '').split(',') # type: List[str] |
| mysql_trace_sql_parameters = True if os.getenv('SW_MYSQL_TRACE_SQL_PARAMETERS') and \ |
| os.getenv('SW_MYSQL_TRACE_SQL_PARAMETERS') == 'True' else False # type: bool |
| mysql_sql_parameters_max_length = int(os.getenv('SW_MYSQL_SQL_PARAMETERS_MAX_LENGTH') or '512') # type: int |
| pymongo_trace_parameters = True if os.getenv('SW_PYMONGO_TRACE_PARAMETERS') and \ |
| os.getenv('SW_PYMONGO_TRACE_PARAMETERS') == 'True' else False # type: bool |
| pymongo_parameters_max_length = int(os.getenv('SW_PYMONGO_PARAMETERS_MAX_LENGTH') or '512') # type: int |
| ignore_suffix = os.getenv('SW_IGNORE_SUFFIX') or '.jpg,.jpeg,.js,.css,.png,.bmp,.gif,.ico,.mp3,' \ |
| '.mp4,.html,.svg ' # type: str |
| flask_collect_http_params = True if os.getenv('SW_FLASK_COLLECT_HTTP_PARAMS') and \ |
| os.getenv('SW_FLASK_COLLECT_HTTP_PARAMS') == 'True' else False # type: bool |
| http_params_length_threshold = int(os.getenv('SW_HTTP_PARAMS_LENGTH_THRESHOLD') or '1024') # type: int |
| django_collect_http_params = True if os.getenv('SW_DJANGO_COLLECT_HTTP_PARAMS') and \ |
| os.getenv('SW_DJANGO_COLLECT_HTTP_PARAMS') == 'True' else False # type: bool |
| correlation_element_max_number = int(os.getenv('SW_CORRELATION_ELEMENT_MAX_NUMBER') or '3') # type: int |
| correlation_value_max_length = int(os.getenv('SW_CORRELATION_VALUE_MAX_LENGTH') or '128') # type: int |
| trace_ignore = True if os.getenv('SW_TRACE_IGNORE') and \ |
| os.getenv('SW_TRACE_IGNORE') == 'True' else False # type: bool |
| trace_ignore_path = (os.getenv('SW_TRACE_IGNORE_PATH') or '').split(',') # type: List[str] |
| elasticsearch_trace_dsl = True if os.getenv('SW_ELASTICSEARCH_TRACE_DSL') and \ |
| os.getenv('SW_ELASTICSEARCH_TRACE_DSL') == 'True' else False # type: bool |
| |
| |
| def init( |
| service: str = None, |
| instance: str = None, |
| collector: str = None, |
| protocol_type: str = 'grpc', |
| token: str = None, |
| ): |
| global service_name |
| service_name = service or service_name |
| |
| global service_instance |
| service_instance = instance or service_instance |
| |
| global collector_address |
| collector_address = collector or collector_address |
| |
| global protocol |
| protocol = protocol_type or protocol |
| |
| global authentication |
| authentication = token or authentication |
| |
| |
| def serialize(): |
| from skywalking import config |
| return { |
| key: value for key, value in config.__dict__.items() if not ( |
| key.startswith('_') or key == 'TYPE_CHECKING' |
| or inspect.isfunction(value) |
| or inspect.ismodule(value) |
| or inspect.isbuiltin(value) |
| or inspect.isclass(value) |
| ) |
| } |
| |
| |
| def deserialize(data): |
| from skywalking import config |
| for key, value in data.items(): |
| if key in config.__dict__: |
| config.__dict__[key] = value |