225
229
# There may be duplicates, but I don't think it is worth worrying
227
231
self._nodes[reference] = ('a', (), '')
232
self._absent_keys.update(absent_references)
233
self._absent_keys.discard(key)
228
234
self._nodes[key] = ('', node_refs, value)
230
235
if self._nodes_by_key is not None and self._key_length > 1:
231
236
self._update_nodes_by_key(key, value, node_refs)
238
def clear_cache(self):
239
"""See GraphIndex.clear_cache()
241
This is a no-op, but we need the api to conform to a generic 'Index'
233
245
def finish(self):
234
246
lines = [_SIGNATURE]
235
247
lines.append(_OPTION_NODE_REFS + str(self.reference_lists) + '\n')
236
248
lines.append(_OPTION_KEY_ELEMENTS + str(self._key_length) + '\n')
237
lines.append(_OPTION_LEN + str(len(self._keys)) + '\n')
249
key_count = len(self._nodes) - len(self._absent_keys)
250
lines.append(_OPTION_LEN + str(key_count) + '\n')
238
251
prefix_length = sum(len(x) for x in lines)
239
252
# references are byte offsets. To avoid having to do nasty
240
253
# polynomial work to resolve offsets (references to later in the
703
751
# the last thing looked up was a terminal element
704
752
yield (self, ) + key_dict
754
def _find_ancestors(self, keys, ref_list_num, parent_map, missing_keys):
755
"""See BTreeIndex._find_ancestors."""
756
# The api can be implemented as a trivial overlay on top of
757
# iter_entries, it is not an efficient implementation, but it at least
761
for index, key, value, refs in self.iter_entries(keys):
762
parent_keys = refs[ref_list_num]
764
parent_map[key] = parent_keys
765
search_keys.update(parent_keys)
766
# Figure out what, if anything, was missing
767
missing_keys.update(set(keys).difference(found_keys))
768
search_keys = search_keys.difference(parent_map)
706
771
def key_count(self):
707
772
"""Return an estimate of the number of keys in this index.
1186
1267
self._indices = indices
1187
1268
self._reload_func = reload_func
1269
# Sibling indices are other CombinedGraphIndex that we should call
1270
# _move_to_front_by_name on when we auto-reorder ourself.
1271
self._sibling_indices = []
1272
# A list of names that corresponds to the instances in self._indices,
1273
# so _index_names[0] is always the name for _indices[0], etc. Sibling
1274
# indices must all use the same set of names as each other.
1275
self._index_names = [None] * len(self._indices)
1189
1277
def __repr__(self):
1190
1278
return "%s(%s)" % (
1191
1279
self.__class__.__name__,
1192
1280
', '.join(map(repr, self._indices)))
1282
def clear_cache(self):
1283
"""See GraphIndex.clear_cache()"""
1284
for index in self._indices:
1194
1287
def get_parent_map(self, keys):
1195
"""See graph._StackedParentsProvider.get_parent_map"""
1288
"""See graph.StackedParentsProvider.get_parent_map"""
1196
1289
search_keys = set(keys)
1197
1290
if NULL_REVISION in search_keys:
1198
1291
search_keys.discard(NULL_REVISION)
1288
1391
seen_keys = set()
1291
1395
for index in self._indices:
1292
1397
for node in index.iter_entries_prefix(keys):
1293
1398
if node[1] in seen_keys:
1295
1400
seen_keys.add(node[1])
1404
hit_indices.append(index)
1298
1406
except errors.NoSuchFile:
1299
1407
self._reload_or_raise()
1408
self._move_to_front(hit_indices)
1410
def _move_to_front(self, hit_indices):
1411
"""Rearrange self._indices so that hit_indices are first.
1413
Order is maintained as much as possible, e.g. the first unhit index
1414
will be the first index in _indices after the hit_indices, and the
1415
hit_indices will be present in exactly the order they are passed to
1418
_move_to_front propagates to all objects in self._sibling_indices by
1419
calling _move_to_front_by_name.
1421
if self._indices[:len(hit_indices)] == hit_indices:
1422
# The 'hit_indices' are already at the front (and in the same
1423
# order), no need to re-order
1425
hit_names = self._move_to_front_by_index(hit_indices)
1426
for sibling_idx in self._sibling_indices:
1427
sibling_idx._move_to_front_by_name(hit_names)
1429
def _move_to_front_by_index(self, hit_indices):
1430
"""Core logic for _move_to_front.
1432
Returns a list of names corresponding to the hit_indices param.
1434
indices_info = zip(self._index_names, self._indices)
1435
if 'index' in debug.debug_flags:
1436
mutter('CombinedGraphIndex reordering: currently %r, promoting %r',
1437
indices_info, hit_indices)
1440
new_hit_indices = []
1443
for offset, (name, idx) in enumerate(indices_info):
1444
if idx in hit_indices:
1445
hit_names.append(name)
1446
new_hit_indices.append(idx)
1447
if len(new_hit_indices) == len(hit_indices):
1448
# We've found all of the hit entries, everything else is
1450
unhit_names.extend(self._index_names[offset+1:])
1451
unhit_indices.extend(self._indices[offset+1:])
1454
unhit_names.append(name)
1455
unhit_indices.append(idx)
1457
self._indices = new_hit_indices + unhit_indices
1458
self._index_names = hit_names + unhit_names
1459
if 'index' in debug.debug_flags:
1460
mutter('CombinedGraphIndex reordered: %r', self._indices)
1463
def _move_to_front_by_name(self, hit_names):
1464
"""Moves indices named by 'hit_names' to front of the search order, as
1465
described in _move_to_front.
1467
# Translate names to index instances, and then call
1468
# _move_to_front_by_index.
1469
indices_info = zip(self._index_names, self._indices)
1471
for name, idx in indices_info:
1472
if name in hit_names:
1473
hit_indices.append(idx)
1474
self._move_to_front_by_index(hit_indices)
1476
def find_ancestry(self, keys, ref_list_num):
1477
"""Find the complete ancestry for the given set of keys.
1479
Note that this is a whole-ancestry request, so it should be used
1482
:param keys: An iterable of keys to look for
1483
:param ref_list_num: The reference list which references the parents
1485
:return: (parent_map, missing_keys)
1487
# XXX: make this call _move_to_front?
1488
missing_keys = set()
1490
keys_to_lookup = set(keys)
1492
while keys_to_lookup:
1493
# keys that *all* indexes claim are missing, stop searching them
1495
all_index_missing = None
1496
# print 'gen\tidx\tsub\tn_keys\tn_pmap\tn_miss'
1497
# print '%4d\t\t\t%4d\t%5d\t%5d' % (generation, len(keys_to_lookup),
1499
# len(missing_keys))
1500
for index_idx, index in enumerate(self._indices):
1501
# TODO: we should probably be doing something with
1502
# 'missing_keys' since we've already determined that
1503
# those revisions have not been found anywhere
1504
index_missing_keys = set()
1505
# Find all of the ancestry we can from this index
1506
# keep looking until the search_keys set is empty, which means
1507
# things we didn't find should be in index_missing_keys
1508
search_keys = keys_to_lookup
1510
# print ' \t%2d\t\t%4d\t%5d\t%5d' % (
1511
# index_idx, len(search_keys),
1512
# len(parent_map), len(index_missing_keys))
1515
# TODO: ref_list_num should really be a parameter, since
1516
# CombinedGraphIndex does not know what the ref lists
1518
search_keys = index._find_ancestors(search_keys,
1519
ref_list_num, parent_map, index_missing_keys)
1520
# print ' \t \t%2d\t%4d\t%5d\t%5d' % (
1521
# sub_generation, len(search_keys),
1522
# len(parent_map), len(index_missing_keys))
1523
# Now set whatever was missing to be searched in the next index
1524
keys_to_lookup = index_missing_keys
1525
if all_index_missing is None:
1526
all_index_missing = set(index_missing_keys)
1528
all_index_missing.intersection_update(index_missing_keys)
1529
if not keys_to_lookup:
1531
if all_index_missing is None:
1532
# There were no indexes, so all search keys are 'missing'
1533
missing_keys.update(keys_to_lookup)
1534
keys_to_lookup = None
1536
missing_keys.update(all_index_missing)
1537
keys_to_lookup.difference_update(all_index_missing)
1538
return parent_map, missing_keys
1301
1540
def key_count(self):
1302
1541
"""Return an estimate of the number of keys in this index.