oslo.rootwrap/tests/run_daemon.py
Yuriy Taraday ed45e5aa6d Add an option to run rootwrap as a daemon
Benchmark for daemon mode is to be added in next commit.

Current results on my machine for 'ip a':
        method          :    min       avg       max       dev
                   ip a :   3.701ms   4.966ms  12.720ms   1.206ms
              sudo ip a :  11.531ms  13.538ms  19.685ms   1.525ms
sudo rootwrap conf ip a :  92.253ms 119.431ms 211.236ms  17.948ms
     daemon.run('ip a') :   6.718ms   9.914ms 189.873ms  18.201ms

Implements: blueprint rootwrap-daemon-mode
Change-Id: Iace26738f910a18a5d1d3479fad949027e5a3816
2014-08-13 20:42:59 +04:00

58 lines
1.5 KiB
Python

# Copyright (c) 2014 Mirantis Inc.
# All Rights Reserved.
#
# Licensed 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.
import logging
import subprocess
import sys
import threading
from oslo.rootwrap import cmd
def forward_stream(fr, to):
while True:
line = fr.readline()
if not line:
break
to.write(line)
def forwarding_popen(f, old_popen=subprocess.Popen):
def popen(*args, **kwargs):
p = old_popen(*args, **kwargs)
t = threading.Thread(target=forward_stream, args=(p.stderr, f))
t.daemon = True
t.start()
return p
return popen
class nonclosing(object):
def __init__(self, f):
self._f = f
def __getattr__(self, name):
return getattr(self._f, name)
def close(self):
pass
log_format = ("%(asctime)s | [%(process)5s]+%(levelname)5s | "
"%(message)s")
if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG, format=log_format)
sys.stderr = nonclosing(sys.stderr)
cmd.daemon()