Skip to content

Commit

Permalink
enabling stdout logs
Browse files Browse the repository at this point in the history
refactoring
  • Loading branch information
bsquizz authored and tahmidefaz committed May 1, 2020
1 parent bb9cf94 commit def379d
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 12 deletions.
3 changes: 1 addition & 2 deletions app/__init__.py
Expand Up @@ -18,8 +18,6 @@
from app.validators import verify_uuid_format # noqa: 401
from tasks import init_tasks

logger = get_logger(__name__)

IDENTITY_HEADER = "x-rh-identity"
REQUEST_ID_HEADER = "x-rh-insights-request-id"
UNKNOWN_REQUEST_ID_VALUE = "-1"
Expand All @@ -41,6 +39,7 @@ def create_app(config_name, start_tasks=False, start_payload_tracker=False):
# This feels like a hack but it is needed. The logging configuration
# needs to be setup before the flask app is initialized.
configure_logging(config_name)
logger = get_logger(__name__) # we need to call the logger after configuring it, not before

app_config = Config(RuntimeEnvironment.server)
app_config.log_configuration(config_name)
Expand Down
31 changes: 21 additions & 10 deletions app/logging.py
Expand Up @@ -13,6 +13,8 @@

threadctx = local()

session_boto3 = None


def configure_logging(config_name):
env_var_name = "INVENTORY_LOGGING_CONFIG_FILE"
Expand Down Expand Up @@ -43,7 +45,6 @@ def _configure_watchtower_logging_handler():
aws_region_name = os.getenv("AWS_REGION_NAME", None)
log_group = os.getenv("AWS_LOG_GROUP", "platform")
stream_name = os.getenv("AWS_LOG_STREAM", _get_hostname()) # default to hostname
create_log_group = str(os.getenv("AWS_CREATE_LOG_GROUP")).lower() == "true"

if all([aws_access_key_id, aws_secret_access_key, aws_region_name, stream_name]):
print(f"Configuring watchtower logging (log_group={log_group}, stream_name={stream_name})")
Expand All @@ -53,19 +54,24 @@ def _configure_watchtower_logging_handler():
region_name=aws_region_name,
)

root = logging.getLogger()
handler = watchtower.CloudWatchLogHandler(
boto3_session=boto3_session,
log_group=log_group,
stream_name=stream_name,
create_log_group=create_log_group
)
handler.setFormatter(logstash_formatter.LogstashFormatterV1())
root.addHandler(handler)
global session_boto3
session_boto3 = boto3_session
else:
print("Unable to configure watchtower logging. Please verify watchtower logging configuration!")


def _get_cloudwatch_handler():
global session_boto3
group = os.getenv("AWS_LOG_GROUP", "platform")
stream = os.getenv("AWS_LOG_STREAM", _get_hostname())
create_log_group = str(os.getenv("AWS_CREATE_LOG_GROUP")).lower() == "true"
handler = watchtower.CloudWatchLogHandler(
boto3_session=session_boto3, log_group=group, stream_name=stream, create_log_group=create_log_group
)
handler.setFormatter(logstash_formatter.LogstashFormatterV1())
return handler


def _get_hostname():
return os.uname()[1]

Expand Down Expand Up @@ -122,5 +128,10 @@ def get_logger(name):
log_level = os.getenv("INVENTORY_LOG_LEVEL", "INFO").upper()
logger = logging.getLogger(LOGGER_PREFIX + name)
logger.addFilter(ContextualFilter())

global session_boto3
if session_boto3 is not None:
logger.addHandler(_get_cloudwatch_handler())

logger.setLevel(log_level)
return logger

0 comments on commit def379d

Please sign in to comment.