~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/trace.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-02 19:56:24 UTC
  • mto: This revision was merged to the branch mainline in revision 4469.
  • Revision ID: john@arbash-meinel.com-20090602195624-utljsyz0qgmq63lg
Add a chunks_to_gzip function.
This allows the _record_to_data code to build up a list of chunks,
rather than requiring a single string.
It should be ~ the same performance when using a single string, since
we are only adding a for() loop over the chunks and an if check.
We could possibly just remove the if check and not worry about adding
some empty strings in there.

Show diffs side-by-side

added added

removed removed

Lines of Context:
20
20
and then delegates it.
21
21
"""
22
22
 
23
 
from bzrlib.transport import decorator
24
 
 
25
 
 
26
 
class TransportTraceDecorator(decorator.TransportDecorator):
 
23
from bzrlib.transport.decorator import TransportDecorator, DecoratorServer
 
24
 
 
25
 
 
26
class TransportTraceDecorator(TransportDecorator):
27
27
    """A tracing decorator for Transports.
28
28
 
29
29
    Calls that potentially perform IO are logged to self._activity. The
33
33
    Not all operations are logged at this point, if you need an unlogged
34
34
    operation please add a test to the tests of this transport, for the logging
35
35
    of the operation you want logged.
36
 
 
37
 
    See also TransportLogDecorator, that records a machine-readable log in 
38
 
    memory for eg testing.
39
36
    """
40
37
 
41
38
    def __init__(self, url, _decorated=None, _from_transport=None):
43
40
 
44
41
        _decorated is a private parameter for cloning.
45
42
        """
46
 
        super(TransportTraceDecorator, self).__init__(url, _decorated)
 
43
        TransportDecorator.__init__(self, url, _decorated)
47
44
        if _from_transport is None:
48
45
            # newly created
49
46
            self._activity = []
129
126
 
130
127
    def readv(self, relpath, offsets, adjust_for_latency=False,
131
128
        upper_limit=None):
132
 
        # we override at the readv() level rather than _readv() so that any
133
 
        # latency adjustments will be done by the underlying transport
 
129
        """See Transport.readv."""
134
130
        self._trace(('readv', relpath, offsets, adjust_for_latency,
135
131
            upper_limit))
136
132
        return self._decorated.readv(relpath, offsets, adjust_for_latency,
169
165
        self._activity.append(operation_tuple)
170
166
 
171
167
 
 
168
class TraceServer(DecoratorServer):
 
169
    """Server for the TransportTraceDecorator for testing with."""
 
170
 
 
171
    def get_decorator_class(self):
 
172
        return TransportTraceDecorator
 
173
 
 
174
 
172
175
def get_test_permutations():
173
176
    """Return the permutations to be used in testing."""
174
 
    from bzrlib.tests import test_server
175
 
    return [(TransportTraceDecorator, test_server.TraceServer)]
 
177
    return [(TransportTraceDecorator, TraceServer)]