← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/smart/protocol.py

  • Committer: Tarmac
  • Author(s): Vincent Ladeuil
  • Date: 2017-01-30 14:42:05 UTC
  • mfrom: (6620.1.1 trunk)
  • Revision ID: tarmac-20170130144205-r8fh2xpmiuxyozpv
Merge  2.7 into trunk including fix for bug #1657238 [r=vila]

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/smart/protocol.py
1
 
# Copyright (C) 2006, 2007, 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
18
18
client and server.
19
19
"""
20
20
 
 
21
from __future__ import absolute_import
 
22
 
21
23
import collections
22
24
from cStringIO import StringIO
23
25
import struct
24
26
import sys
25
27
import thread
26
 
import threading
27
28
import time
28
29
 
29
30
import bzrlib
62
63
 
63
64
def _encode_tuple(args):
64
65
    """Encode the tuple args to a bytestream."""
65
 
    return '\x01'.join(args) + '\n'
 
66
    joined = '\x01'.join(args) + '\n'
 
67
    if type(joined) is unicode:
 
68
        # XXX: We should fix things so this never happens!  -AJB, 20100304
 
69
        mutter('response args contain unicode, should be only bytes: %r',
 
70
               joined)
 
71
        joined = joined.encode('ascii')
 
72
    return joined
66
73
 
67
74
 
68
75
class Requester(object):
648
655
        """Make a remote call with a readv array.
649
656
 
650
657
        The body is encoded with one line per readv offset pair. The numbers in
651
 
        each pair are separated by a comma, and no trailing \n is emitted.
 
658
        each pair are separated by a comma, and no trailing \\n is emitted.
652
659
        """
653
660
        if 'hpss' in debug.debug_flags:
654
661
            mutter('hpss call w/readv: %s', repr(args)[1:-1])
1075
1082
        self._real_write_func = write_func
1076
1083
 
1077
1084
    def _write_func(self, bytes):
1078
 
        # TODO: It is probably more appropriate to use sum(map(len, _buf))
1079
 
        #       for total number of bytes to write, rather than buffer based on
1080
 
        #       the number of write() calls
1081
1085
        # TODO: Another possibility would be to turn this into an async model.
1082
1086
        #       Where we let another thread know that we have some bytes if
1083
1087
        #       they want it, but we don't actually block for it
1225
1229
                    if first_chunk is None:
1226
1230
                        first_chunk = chunk
1227
1231
                    self._write_prefixed_body(chunk)
 
1232
                    self.flush()
1228
1233
                    if 'hpssdetail' in debug.debug_flags:
1229
1234
                        # Not worth timing separately, as _write_func is
1230
1235
                        # actually buffered
1285
1290
        _ProtocolThreeEncoder.__init__(self, medium_request.accept_bytes)
1286
1291
        self._medium_request = medium_request
1287
1292
        self._headers = {}
 
1293
        self.body_stream_started = None
1288
1294
 
1289
1295
    def set_headers(self, headers):
1290
1296
        self._headers = headers.copy()
1325
1331
        """Make a remote call with a readv array.
1326
1332
 
1327
1333
        The body is encoded with one line per readv offset pair. The numbers in
1328
 
        each pair are separated by a comma, and no trailing \n is emitted.
 
1334
        each pair are separated by a comma, and no trailing \\n is emitted.
1329
1335
        """
1330
1336
        if 'hpss' in debug.debug_flags:
1331
1337
            mutter('hpss call w/readv: %s', repr(args)[1:-1])
1350
1356
            if path is not None:
1351
1357
                mutter('                  (to %s)', path)
1352
1358
            self._request_start_time = osutils.timer_func()
 
1359
        self.body_stream_started = False
1353
1360
        self._write_protocol_version()
1354
1361
        self._write_headers(self._headers)
1355
1362
        self._write_structure(args)
1357
1364
        #       have finished sending the stream.  We would notice at the end
1358
1365
        #       anyway, but if the medium can deliver it early then it's good
1359
1366
        #       to short-circuit the whole request...
 
1367
        # Provoke any ConnectionReset failures before we start the body stream.
 
1368
        self.flush()
 
1369
        self.body_stream_started = True
1360
1370
        for exc_info, part in _iter_with_errors(stream):
1361
1371
            if exc_info is not None:
1362
1372
                # Iterating the stream failed.  Cleanly abort the request.