diff options
author | 2015-11-24 01:35:18 -0800 | |
---|---|---|
committer | 2015-11-24 01:35:18 -0800 | |
commit | a7f8202e77db600354d89cf15c4d5636bffb0415 (patch) | |
tree | 7323ee4817f1d1b4adf863204f14a10710724266 /utils/dispatcher/file.py | |
parent | 68e74fd78b2485e5914ce34a5b30f4485029e021 (diff) |
dispathcer to transfer test result into DB
JIRA: BOTTLENECK-30
Change-Id: I5be6641512eb0c2b1e4e3ea92a5dea720e97f770
Signed-off-by: MatthewLi <matthew.lijun@huawei.com>
Diffstat (limited to 'utils/dispatcher/file.py')
-rw-r--r-- | utils/dispatcher/file.py | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/utils/dispatcher/file.py b/utils/dispatcher/file.py new file mode 100644 index 00000000..6e1df828 --- /dev/null +++ b/utils/dispatcher/file.py @@ -0,0 +1,66 @@ +############################################################################## +# Copyright (c) 2015 Huawei Technologies Co.,Ltd and others. +# liangqi1@huawei.com matthew.lijun@huawei.com +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +import logging +import json + +from oslo_config import cfg + +from utils.dispatcher.base import Base as DispatchBase + +CONF = cfg.CONF +opts = [ + cfg.StrOpt('file_path', + default='/tmp/bottlenecks.out', + help='Name and the location of the file to record ' + 'data.'), + cfg.IntOpt('max_bytes', + default=0, + help='The max size of the file.'), + cfg.IntOpt('backup_count', + default=0, + help='The max number of the files to keep.'), +] +CONF.register_opts(opts, group="dispatcher_file") + + +class FileDispatcher(DispatchBase): + """Dispatcher class for recording data to a file. + """ + + __dispatcher_type__ = "File" + + def __init__(self, conf): + super(FileDispatcher, self).__init__(conf) + self.log = None + + # if the directory and path are configured, then log to the file + if CONF.dispatcher_file.file_path: + dispatcher_logger = logging.Logger('dispatcher.file') + dispatcher_logger.setLevel(logging.INFO) + # create rotating file handler which logs result + rfh = logging.handlers.RotatingFileHandler( + self.conf.get('file_path', CONF.dispatcher_file.file_path), + maxBytes=CONF.dispatcher_file.max_bytes, + backupCount=CONF.dispatcher_file.backup_count, + encoding='utf8') + + rfh.setLevel(logging.INFO) + # Only wanted the data to be saved in the file, not the + # project root logger. + dispatcher_logger.propagate = False + dispatcher_logger.addHandler(rfh) + self.log = dispatcher_logger + + def record_result_data(self, data): + if self.log: + self.log.info(json.dumps(data)) + + def flush_result_data(self): + pass |