77
def recreate_search(self, repository, search_bytes, discard_excess=False):
78
"""Recreate a search from its serialised form.
80
:param discard_excess: If True, and the search refers to data we don't
81
have, just silently accept that fact - the verb calling
82
recreate_search trusts that clients will look for missing things
83
they expected and get it from elsewhere.
74
def recreate_search(self, repository, search_bytes):
85
75
lines = search_bytes.split('\n')
86
76
if lines[0] == 'ancestry-of':
88
78
search_result = graph.PendingAncestryResult(heads, repository)
89
79
return search_result, None
90
80
elif lines[0] == 'search':
91
return self.recreate_search_from_recipe(repository, lines[1:],
92
discard_excess=discard_excess)
81
return self.recreate_search_from_recipe(repository, lines[1:])
94
83
return (None, FailedSmartServerResponse(('BadSearch',)))
96
def recreate_search_from_recipe(self, repository, lines,
97
discard_excess=False):
98
"""Recreate a specific revision search (vs a from-tip search).
100
:param discard_excess: If True, and the search refers to data we don't
101
have, just silently accept that fact - the verb calling
102
recreate_search trusts that clients will look for missing things
103
they expected and get it from elsewhere.
85
def recreate_search_from_recipe(self, repository, lines):
105
86
start_keys = set(lines[0].split(' '))
106
87
exclude_keys = set(lines[1].split(' '))
107
88
revision_count = int(lines[2])
287
267
return SuccessfulSmartServerResponse(('ok', ), '\n'.join(lines))
290
class SmartServerRepositoryGetRevIdForRevno(SmartServerRepositoryReadLocked):
292
def do_readlocked_repository_request(self, repository, revno,
294
"""Find the revid for a given revno, given a known revno/revid pair.
299
found_flag, result = repository.get_rev_id_for_revno(revno, known_pair)
300
except errors.RevisionNotPresent, err:
301
if err.revision_id != known_pair[1]:
302
raise AssertionError(
303
'get_rev_id_for_revno raised RevisionNotPresent for '
304
'non-initial revision: ' + err.revision_id)
305
return FailedSmartServerResponse(
306
('nosuchrevision', err.revision_id))
308
return SuccessfulSmartServerResponse(('ok', result))
310
earliest_revno, earliest_revid = result
311
return SuccessfulSmartServerResponse(
312
('history-incomplete', earliest_revno, earliest_revid))
315
270
class SmartServerRequestHasRevision(SmartServerRepositoryRequest):
317
272
def do_repository_request(self, repository, revision_id):
420
375
self._to_format = network_format_registry.get(to_network_name)
421
if self._should_fake_unknown():
422
return FailedSmartServerResponse(
423
('UnknownMethod', 'Repository.get_stream'))
424
376
return None # Signal that we want a body.
426
def _should_fake_unknown(self):
427
"""Return True if we should return UnknownMethod to the client.
429
This is a workaround for bugs in pre-1.19 clients that claim to
430
support receiving streams of CHK repositories. The pre-1.19 client
431
expects inventory records to be serialized in the format defined by
432
to_network_name, but in pre-1.19 (at least) that format definition
433
tries to use the xml5 serializer, which does not correctly handle
434
rich-roots. After 1.19 the client can also accept inventory-deltas
435
(which avoids this issue), and those clients will use the
436
Repository.get_stream_1.19 verb instead of this one.
437
So: if this repository is CHK, and the to_format doesn't match,
438
we should just fake an UnknownSmartMethod error so that the client
439
will fallback to VFS, rather than sending it a stream we know it
442
from_format = self._repository._format
443
to_format = self._to_format
444
if not from_format.supports_chks:
445
# Source not CHK: that's ok
447
if (to_format.supports_chks and
448
from_format.repository_class is to_format.repository_class and
449
from_format._serializer == to_format._serializer):
450
# Source is CHK, but target matches: that's ok
451
# (e.g. 2a->2a, or CHK2->2a)
453
# Source is CHK, and target is not CHK or incompatible CHK. We can't
454
# generate a compatible stream.
457
378
def do_body(self, body_bytes):
458
379
repository = self._repository
459
380
repository.lock_read()
461
search_result, error = self.recreate_search(repository, body_bytes,
382
search_result, error = self.recreate_search(repository, body_bytes)
463
383
if error is not None:
464
384
repository.unlock()
489
409
repository.unlock()
492
class SmartServerRepositoryGetStream_1_19(SmartServerRepositoryGetStream):
494
def _should_fake_unknown(self):
495
"""Returns False; we don't need to workaround bugs in 1.19+ clients."""
499
412
def _stream_to_byte_stream(stream, src_format):
500
413
"""Convert a record stream to a self delimited byte stream."""
501
414
pack_writer = pack.ContainerSerialiser()
502
415
yield pack_writer.begin()
503
416
yield pack_writer.bytes_record(src_format.network_name(), '')
504
417
for substream_type, substream in stream:
505
if substream_type == 'inventory-deltas':
506
# This doesn't feel like the ideal place to issue this warning;
507
# however we don't want to do it in the Repository that's
508
# generating the stream, because that might be on the server.
509
# Instead we try to observe it as the stream goes by.
510
ui.ui_factory.warn_cross_format_fetch(src_format,
512
418
for record in substream:
513
419
if record.storage_kind in ('chunked', 'fulltext'):
514
420
serialised = record_to_fulltext_bytes(record)
515
elif record.storage_kind == 'inventory-delta':
516
serialised = record_to_inventory_delta_bytes(record)
517
elif record.storage_kind == 'absent':
518
raise ValueError("Absent factory for %s" % (record.key,))
520
422
serialised = record.get_bytes_as(record.storage_kind)
526
428
yield pack_writer.end()
529
class _ByteStreamDecoder(object):
530
"""Helper for _byte_stream_to_stream.
532
The expected usage of this class is via the function _byte_stream_to_stream
533
which creates a _ByteStreamDecoder, pops off the stream format and then
534
yields the output of record_stream(), the main entry point to
537
Broadly this class has to unwrap two layers of iterators:
541
This is complicated by wishing to return type, iterator_for_type, but
542
getting the data for iterator_for_type when we find out type: we can't
543
simply pass a generator down to the NetworkRecordStream parser, instead
544
we have a little local state to seed each NetworkRecordStream instance,
545
and gather the type that we'll be yielding.
547
:ivar byte_stream: The byte stream being decoded.
548
:ivar stream_decoder: A pack parser used to decode the bytestream
549
:ivar current_type: The current type, used to join adjacent records of the
550
same type into a single stream.
551
:ivar first_bytes: The first bytes to give the next NetworkRecordStream.
554
def __init__(self, byte_stream):
555
"""Create a _ByteStreamDecoder."""
556
self.stream_decoder = pack.ContainerPushParser()
557
self.current_type = None
558
self.first_bytes = None
559
self.byte_stream = byte_stream
561
def iter_stream_decoder(self):
562
"""Iterate the contents of the pack from stream_decoder."""
563
# dequeue pending items
564
for record in self.stream_decoder.read_pending_records():
566
# Pull bytes of the wire, decode them to records, yield those records.
567
for bytes in self.byte_stream:
568
self.stream_decoder.accept_bytes(bytes)
569
for record in self.stream_decoder.read_pending_records():
572
def iter_substream_bytes(self):
573
if self.first_bytes is not None:
574
yield self.first_bytes
575
# If we run out of pack records, single the outer layer to stop.
576
self.first_bytes = None
577
for record in self.iter_pack_records:
578
record_names, record_bytes = record
579
record_name, = record_names
580
substream_type = record_name[0]
581
if substream_type != self.current_type:
582
# end of a substream, seed the next substream.
583
self.current_type = substream_type
584
self.first_bytes = record_bytes
588
def record_stream(self):
589
"""Yield substream_type, substream from the byte stream."""
591
# Make and consume sub generators, one per substream type:
592
while self.first_bytes is not None:
593
substream = NetworkRecordStream(self.iter_substream_bytes())
594
# after substream is fully consumed, self.current_type is set to
595
# the next type, and self.first_bytes is set to the matching bytes.
596
yield self.current_type, substream.read()
598
def seed_state(self):
599
"""Prepare the _ByteStreamDecoder to decode from the pack stream."""
600
# Set a single generator we can use to get data from the pack stream.
601
self.iter_pack_records = self.iter_stream_decoder()
602
# Seed the very first subiterator with content; after this each one
604
list(self.iter_substream_bytes())
607
431
def _byte_stream_to_stream(byte_stream):
608
432
"""Convert a byte stream into a format and a stream.
610
434
:param byte_stream: A bytes iterator, as output by _stream_to_byte_stream.
611
435
:return: (RepositoryFormat, stream_generator)
613
decoder = _ByteStreamDecoder(byte_stream)
437
stream_decoder = pack.ContainerPushParser()
439
"""Closure to return the substreams."""
440
# May have fully parsed records already.
441
for record in stream_decoder.read_pending_records():
442
record_names, record_bytes = record
443
record_name, = record_names
444
substream_type = record_name[0]
445
substream = NetworkRecordStream([record_bytes])
446
yield substream_type, substream.read()
447
for bytes in byte_stream:
448
stream_decoder.accept_bytes(bytes)
449
for record in stream_decoder.read_pending_records():
450
record_names, record_bytes = record
451
record_name, = record_names
452
substream_type = record_name[0]
453
substream = NetworkRecordStream([record_bytes])
454
yield substream_type, substream.read()
614
455
for bytes in byte_stream:
615
decoder.stream_decoder.accept_bytes(bytes)
616
for record in decoder.stream_decoder.read_pending_records(max=1):
456
stream_decoder.accept_bytes(bytes)
457
for record in stream_decoder.read_pending_records(max=1):
617
458
record_names, src_format_name = record
618
459
src_format = network_format_registry.get(src_format_name)
619
return src_format, decoder.record_stream()
460
return src_format, record_stream()
622
463
class SmartServerRepositoryUnlock(SmartServerRepositoryRequest):
766
606
return SuccessfulSmartServerResponse(('ok', ))
769
class SmartServerRepositoryInsertStream_1_19(SmartServerRepositoryInsertStreamLocked):
770
"""Insert a record stream from a RemoteSink into a repository.
772
Same as SmartServerRepositoryInsertStreamLocked, except:
773
- the lock token argument is optional
774
- servers that implement this verb accept 'inventory-delta' records in the
780
def do_repository_request(self, repository, resume_tokens, lock_token=None):
781
"""StreamSink.insert_stream for a remote repository."""
782
SmartServerRepositoryInsertStreamLocked.do_repository_request(
783
self, repository, resume_tokens, lock_token)
786
609
class SmartServerRepositoryInsertStream(SmartServerRepositoryInsertStreamLocked):
787
610
"""Insert a record stream from a RemoteSink into an unlocked repository.
added
removed
bzrlib/smart/repository.py
