Tryag File Manager - Edit By Mr.SaHr

//opt/nydus/pyvenv/lib64/python3.5/site-packages/nydus/log.py

# -*- coding: utf-8 -*-

"""\
© Copyright. All rights reserved.

"""

from contextlib import contextmanager
from functools import wraps
from threading import local
import logging

from archon.jobspec import JobSpec
from primordial import log as primordial_log
from primordial.log.envelope import EnvelopeV1
from primordial.log.extra import LogExtra

from nydus.util import first

_THREAD_LOCAL = local()
_CONTEXT = _THREAD_LOCAL.context = []
# CherryPy runs as a threadpool, so log context must be local to each thread.
# See https://docs.python.org/3.5/howto/logging-cookbook.html#using-filters-to-impart-contextual-information
# and https://docs.python.org/3.5/library/threading.html#thread-local-data.

def init_logging(filename, config, subtype):
    primordial_log.init_logging(
        name='nydus.settings',
        filename=filename,
        log_type='hfs',
        log_sub_type=subtype,
        datacenter=config.get('hfs_datacenter'),
        environment=config.get('hfs_environment'))

@contextmanager
def log_context(**kw):
    """Add given keyword arguments to the log context for the duration of the
    wrapped block.
    """
    _CONTEXT.append(kw)
    try:
        yield
    finally:
        _CONTEXT.pop()

def log_jobspec_arg(original):
    """Add first JobSpec parameter of wrapped function to the log context."""
    @wraps(original)
    def wrapper(*args, **kw):
        jobspec = first(lambda a: isinstance(a, JobSpec), args)
        with log_context(workflow_id=jobspec.workflow_guid,
                         jobspec_id=jobspec.guid):
            return original(*args, **kw)
    return wrapper

def log_nydus_headers(original):
    """Add nydus-workflow-id and nydus-jobspec-id headers to the log context.
    Add these values to the request as `nydus_context` workflow_id and
    jobspec_id for the receiver.
    """
    @wraps(original)
    def wrapper(self, req, *args, **kw):
        workflow_id = req.get_header('nydus-workflow-id')
        jobspec_id = req.get_header('nydus-jobspec-id')
        req.nydus_context = {}
        if workflow_id is not None:
            req.nydus_context['workflow_id'] = workflow_id
        if jobspec_id is not None:
            req.nydus_context['jobspec_id'] = jobspec_id
        with log_context(**req.nydus_context):
            return original(self, req, *args, **kw)
    return wrapper

class LogContextFilter(logging.Filter):
    """Add log context specified by calls to the log_context context manager
    to the payload element ("data") of the log record.
    """
    def filter(self, record):
        if _CONTEXT:
            if not hasattr(record, LogExtra.EXTRA_FIELD):
                record.extra = {}
            data = record.extra.setdefault(EnvelopeV1.PAYLOAD_FIELD, {})
            for d in _CONTEXT:
                data.update(d)
            # As part of the Nydus log payload, these items only belong in
            # extra/data.  But in order to use them in SystemdFormatter,
            # they must be attached directly to the log record.
            # record.workflow_id = record.jobspec_id = ...
        return True