airflow non_caching_file_handler 源码

  • 2022-10-20
  • 浏览 (282)

airflow non_caching_file_handler 代码

文件路径:/airflow/utils/log/non_caching_file_handler.py

# 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.
from __future__ import annotations

import logging
import os


class NonCachingFileHandler(logging.FileHandler):
    """
    This is an extension of the python FileHandler that advises the Kernel to not cache the file
    in PageCache when it is written. While there is nothing wrong with such cache (it will be cleaned
    when memory is needed), it causes ever-growing memory usage when scheduler is running as it keeps
    on writing new log files and the files are not rotated later on. This might lead to confusion
    for our users, who are monitoring memory usage of Scheduler - without realising that it is
    harmless and expected in this case.

    See https://github.com/apache/airflow/issues/14924

    Adding the advice to Kernel might help with not generating the cache memory growth in the first place.
    """

    def _open(self):
        wrapper = super()._open()
        try:
            fd = wrapper.fileno()
            os.posix_fadvise(fd, 0, 0, os.POSIX_FADV_DONTNEED)
        except Exception:
            # in case either file descriptor cannot be retrieved or fadvise is not available
            # we should simply return the wrapper retrieved by FileHandler's open method
            # the advise to the kernel is just an advise and if we cannot give it, we won't
            pass
        return wrapper

相关信息

airflow 源码目录

相关文章

airflow init 源码

airflow colored_log 源码

airflow file_processor_handler 源码

airflow file_task_handler 源码

airflow json_formatter 源码

airflow log_reader 源码

airflow logging_mixin 源码

airflow secrets_masker 源码

airflow task_handler_with_custom_formatter 源码

airflow timezone_aware 源码

0  赞