oioswift-proxy-server

#!/usr/bin/env python
# Copyright (c) 2010-2012 OpenStack Foundation
# Copyright (c) 2019-2020 OpenIO SAS
#
# 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.

from __future__ import print_function

import sys
from optparse import OptionParser

# Immediately change "wsgi" before importing the other modules
# for this modification to be propagated everywhere
from oioswift.common import wsgi as oiowsgi
from swift.common import wsgi

orig_run_server = wsgi.run_server


def run_server_wrapper(conf, logger, sock, global_conf=None):
    if global_conf is None and GLOBAL_CONF is not None:
        global_conf = {k: v for k, v in GLOBAL_CONF.items()
                       if k.startswith('oio')}
    return orig_run_server(conf, logger, sock, global_conf)


wsgi.make_env = oiowsgi.oio_make_env
wsgi.make_subrequest = oiowsgi.oio_make_subrequest
wsgi.run_server = run_server_wrapper


from oioswift.server import global_conf_callback  # noqa
from swift.common.utils import parse_options  # noqa


GLOBAL_CONF = None


def global_conf_callback_wrapper(preloaded_app_conf, global_conf):
    global GLOBAL_CONF
    global_conf_callback(preloaded_app_conf, global_conf)
    GLOBAL_CONF = global_conf


def run_objgraph(types):
    import objgraph
    import os
    import random
    objgraph.show_most_common_types(limit=50, shortnames=False)
    for type_ in types:
        count = objgraph.count(type_)
        print('%s objects: %d' % (type_, count))
        if count:
            objgraph.show_backrefs(
                random.choice(objgraph.by_type(type_)), max_depth=20,
                filename='/tmp/backrefs_%s_%d.dot' % (type_, os.getpid()))


if __name__ == '__main__':
    parser = OptionParser(usage="%prog CONFIG [options]")
    parser.add_option('--objgraph', action='store_true',
                      help=('Run objgraph, show most common '
                            'types before exiting'))
    parser.add_option('--show-backrefs', action='append', default=list(),
                      help=('Draw backreference graph for one randomly '
                            'chosen object of that type. Can be used '
                            'multiple times.'))
    conf_file, options = parse_options(parser)
    res = wsgi.run_wsgi(conf_file, 'proxy-server',
                        global_conf_callback=global_conf_callback_wrapper,
                        **options)
    if options.get('objgraph'):
        run_objgraph(options.get('show_backrefs', list()))
    sys.exit(res)