← Back to branch summary

~loggerhead-team/loggerhead/trunk-rich

« back to all changes in this revision

Viewing changes to __init__.py

  • Committer: Robey Pointer
  • Date: 2007-01-02 04:34:09 UTC
  • Revision ID: robey@lag.net-20070102043409-5slhxuzq5omtim2f
exponential backoff isn't really working for the lockfile, so keep it down
to 0.1 second pauses while trying to grab the lock.  also, flush the caches
at shutdown, and pause during cache rebuilding to give normal requests a
chance to get the lock.  (with threading.Lock, wakeups were automatic, but
with the file lock, there's a lot of sleeping going on.)

Show diffs side-by-side

added added

removed removed

Lines of Context:
__init__.py
1
 
# Copyright 2009, 2010, 2011 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
# This file allows loggerhead to be treated as a plugin for bzr.
19
 
#
20
 
# XXX: Because loggerhead already contains a loggerhead directory, much of the
21
 
# code is going to appear loaded at bzrlib.plugins.loggerhead.loggerhead.
22
 
# This seems like the easiest thing, because bzrlib wants the top-level plugin
23
 
# directory to be the module, but when it's used as a library people expect
24
 
# the source directory to contain a directory called loggerhead.  -- mbp
25
 
# 20090123
26
 
 
27
 
"""Loggerhead web viewer for Bazaar branches.
28
 
 
29
 
This provides a new option "--http" to the "bzr serve" command, that
30
 
starts a web server to browse the contents of a branch.
31
 
"""
32
 
 
33
 
import sys
34
 
 
35
 
from info import (
36
 
    bzr_plugin_version as version_info,
37
 
    bzr_compatible_versions,
38
 
    )
39
 
 
40
 
if __name__ == 'bzrlib.plugins.loggerhead':
41
 
    import bzrlib
42
 
    from bzrlib.api import require_any_api
43
 
    from bzrlib import commands
44
 
 
45
 
    require_any_api(bzrlib, bzr_compatible_versions)
46
 
 
47
 
    from bzrlib.transport import transport_server_registry
48
 
 
49
 
    DEFAULT_HOST = '0.0.0.0'
50
 
    DEFAULT_PORT = 8080
51
 
    HELP = ('Loggerhead, a web-based code viewer and server. (default port: %d)' %
52
 
            (DEFAULT_PORT,))
53
 
 
54
 
    def _ensure_loggerhead_path():
55
 
        """Ensure that you can 'import loggerhead' and get the root."""
56
 
        # loggerhead internal code will try to 'import loggerhead', so
57
 
        # let's put it on the path if we can't find it in the existing path
58
 
        try:
59
 
            import loggerhead.apps.transport
60
 
        except ImportError:
61
 
            import os.path, sys
62
 
            sys.path.append(os.path.dirname(__file__))
63
 
 
64
 
    def serve_http(transport, host=None, port=None, inet=None):
65
 
        from paste.httpexceptions import HTTPExceptionHandler
66
 
        from paste.httpserver import serve
67
 
 
68
 
        _ensure_loggerhead_path()
69
 
 
70
 
        from loggerhead.apps.transport import BranchesFromTransportRoot
71
 
        from loggerhead.config import LoggerheadConfig
72
 
        from loggerhead.main import setup_logging
73
 
 
74
 
        if host is None:
75
 
            host = DEFAULT_HOST
76
 
        if port is None:
77
 
            port = DEFAULT_PORT
78
 
        argv = ['--host', host, '--port', str(port), '--', transport.base]
79
 
        if not transport.is_readonly():
80
 
            argv.insert(0, '--allow-writes')
81
 
        config = LoggerheadConfig(argv)
82
 
        setup_logging(config, init_logging=False, log_file=sys.stderr)
83
 
        app = BranchesFromTransportRoot(transport.base, config)
84
 
        app = HTTPExceptionHandler(app)
85
 
        serve(app, host=host, port=port)
86
 
 
87
 
    transport_server_registry.register('http', serve_http, help=HELP)
88
 
 
89
 
    class cmd_load_test_loggerhead(commands.Command):
90
 
        """Run a load test against a live loggerhead instance.
91
 
 
92
 
        Pass in the name of a script file to run. See loggerhead/load_test.py
93
 
        for a description of the file format.
94
 
        """
95
 
 
96
 
        takes_args = ["filename"]
97
 
 
98
 
        def run(self, filename):
99
 
            from bzrlib.plugins.loggerhead.loggerhead import load_test
100
 
            script = load_test.run_script(filename)
101
 
            for thread_id in sorted(script._threads):
102
 
                worker = script._threads[thread_id][0]
103
 
                for url, success, time in worker.stats:
104
 
                    self.outf.write(' %5.3fs %s %s\n'
105
 
                                    % (time, str(success)[0], url))
106
 
 
107
 
    commands.register_command(cmd_load_test_loggerhead)
108
 
 
109
 
    def load_tests(standard_tests, module, loader):
110
 
        _ensure_loggerhead_path()
111
 
        standard_tests.addTests(loader.loadTestsFromModuleNames(
112
 
            ['bzrlib.plugins.loggerhead.loggerhead.tests']))
113
 
        return standard_tests