From 4110287efbeb1304cda70fd6ce7ea42422d327f7 Mon Sep 17 00:00:00 2001 From: marcsello Date: Thu, 5 Aug 2021 15:55:03 +0200 Subject: [PATCH] Fixed logging --- model_service/app.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/model_service/app.py b/model_service/app.py index 82b38fa..2e4661b 100644 --- a/model_service/app.py +++ b/model_service/app.py @@ -74,8 +74,19 @@ app.add_url_rule("/healthz", "healthcheck", view_func=lambda: health.run()) if __name__ == "__main__": app.run(debug=True) else: - import logging + import os - gunicorn_logger = logging.getLogger('gunicorn.error') - app.logger.handlers = gunicorn_logger.handlers - app.logger.setLevel(gunicorn_logger.level) + if "gunicorn" in os.environ.get("SERVER_SOFTWARE", ""): + import logging + + gunicorn_logger = logging.getLogger('gunicorn.error') + app.logger.handlers = gunicorn_logger.handlers + app.logger.setLevel(gunicorn_logger.level) + + jaeger_logger = logging.getLogger('jaeger_tracing') + jaeger_logger.handlers = gunicorn_logger.handlers + jaeger_logger.setLevel(gunicorn_logger.level) + + app.logger.info("Gunicorn environment detected!") + else: + app.logger.info("Not gunicorn")