记录烧瓶应用程序

Posted 2021-03-27

我正在开发一个带有烧瓶的webapp，它充当了进行计算的python库的接口（通常很耗时）。

每个对服务器的调用都用标识符标识，我想将对库的调用日志写入依赖于给定标识符的文件。

最小的工作示例如下。

computations.朋友

import time
import logging

logger = logging.getLogger(__name__)

def long_computation(identifier):
    logger.info('called computation with identifier %s', identifier)
    for i in range(100):
        logger.info('in step %d of identifier %s', i, identifier)
        time.sleep(1)
    logger.info('finished computation with identifier %s')

server.朋友

from flask import Flask, request
import logging
import threading
import computations

app = Flask(__name__)

def call_computation(identifier):
    fh = logging.FileHandler("computations-%s.log" % identifier)
    formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(name)s : %(message)s')
    fh.setFormatter(formatter)
    fh.setLevel(logging.INFO)

    computations.logger.setLevel(logging.INFO)
    computations.logger.addHandler(fh)
    computations.long_computation(identifier)
    computations.logger.removeHandler(fh)

@app.route('/node/<identifier>', methods=['GET','POST'])
def serve_node(identifier):
    thread = threading.Thread(target=call_computation, args=(identifier,))
    thread.start()
    return "I will compute it!"

当我打电话给服务器时，比如http://127.0.0.1:5000/node/A它会创建日志文件computations-A.log并正确记录到这个文件。但是，如果我再次调用服务器，在第一次计算结束之前说http://127.0.0.1:5000/node/B，那么它会创建日志文件computations-B.log，但两个计算的日志（对应于对call_computation的不同调用）都会转到这两个文件。也就是说，文件computations-A.log和computations-B.log都有例如以下行：

2018-08-02 20:31:57,524 INFO     computations : in step 56 of identifier B
2018-08-02 20:31:57,799 INFO     computations : in step 97 of identifier A

任何人都可以帮助我，以便调用库来转到合适的日志文件？请注意，我原则上不能修改进行计算的包，因此我无法在该包中创建更多的记录器。

提前致谢！

答案

解决方案是过滤日志记录。在server.py文件中创建logging.Filter的子类：

class MyFilter(logging.Filter):
    def __init__(self, thread_id):
        super(MyFilter, self).__init__()
        self.thread_id = thread_id

    def filter(self, record):
        return record.thread == self.thread_id

在设置处理程序时，添加此类的实例：

myfilter = MyFilter(threading.current_thread().ident)
fh.addFilter(myfilter)

这样，当日志记录到达过滤器时，如果创建日志的线程与创建过滤器的线程相同，那么它将传递到下一级别;否则会被过滤掉。