我正在开发一个带有烧瓶的webapp,它充当了进行计算的python库的接口(通常很耗时)。
每个对服务器的调用都用标识符标识,我想将对库的调用日志写入依赖于给定标识符的文件。
最小的工作示例如下。
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')
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)
这样,当日志记录到达过滤器时,如果创建日志的线程与创建过滤器的线程相同,那么它将传递到下一级别;否则会被过滤掉。