55
53
return BazaarObjectStore(repo, mapping)
58
MAX_INV_CACHE_SIZE = 50 * 1024 * 1024
61
class LRUInventoryCache(object):
63
def __init__(self, repository):
64
def approx_inv_size(inv):
65
# Very rough estimate, 1k per inventory entry
66
return len(inv) * 1024
67
self.repository = repository
68
self._cache = lru_cache.LRUSizeCache(max_size=MAX_INV_CACHE_SIZE,
69
after_cleanup_size=None, compute_size=approx_inv_size)
71
def get_inventory(self, revid):
73
return self._cache[revid]
75
inv = self.repository.get_inventory(revid)
76
self._cache.add(revid, inv)
79
def iter_inventories(self, revids):
80
invs = dict([(k, self._cache.get(k)) for k in revids])
81
for inv in self.repository.iter_inventories(
82
[r for r, v in invs.iteritems() if v is None]):
83
invs[inv.revision_id] = inv
84
self._cache.add(inv.revision_id, inv)
85
return (invs[r] for r in revids)
87
def get_inventories(self, revids):
88
return list(self.iter_inventories(revids))
90
def add(self, revid, inv):
91
self._cache.add(revid, inv)
94
def _check_expected_sha(expected_sha, object):
95
"""Check whether an object matches an expected SHA.
97
:param expected_sha: None or expected SHA as either binary or as hex digest
98
:param object: Object to verify
100
if expected_sha is None:
102
if len(expected_sha) == 40:
103
if expected_sha != object.sha().hexdigest():
104
raise AssertionError("Invalid sha for %r: %s" % (object,
106
elif len(expected_sha) == 20:
107
if expected_sha != object.sha().digest():
108
raise AssertionError("Invalid sha for %r: %s" % (object,
109
sha_to_hex(expected_sha)))
111
raise AssertionError("Unknown length %d for %r" % (len(expected_sha),
115
def _inventory_to_objects(inv, parent_invs, idmap,
116
unusual_modes, iter_files_bytes, has_ghost_parents):
117
"""Iterate over the objects that were introduced in a revision.
119
:param inv: Inventory to process
120
:param parent_invs: parent inventory SHA maps
122
:param unusual_modes: Unusual file modes
123
:param iter_files_bytes: Repository.iter_files_bytes-like callback
124
:return: Yields (path, object, ie) entries
129
for path, ie in inv.entries():
130
if ie.kind == "file":
131
if ie.revision != inv.revision_id:
132
for pinv in parent_invs:
134
pie = pinv[ie.file_id]
135
except errors.NoSuchId:
138
if (pie.text_sha1 == ie.text_sha1 and
139
pie.kind == ie.kind):
140
shamap[ie.file_id] = idmap.lookup_blob_id(
141
pie.file_id, pie.revision)
143
if not ie.file_id in shamap:
144
new_blobs.append((path, ie))
145
new_trees[urlutils.dirname(path)] = ie.parent_id
146
elif ie.kind == "symlink":
147
blob = symlink_to_blob(ie)
148
for pinv in parent_invs:
150
pie = pinv[ie.file_id]
151
except errors.NoSuchId:
154
if (ie.kind == pie.kind and
155
ie.symlink_target == pie.symlink_target):
159
new_trees[urlutils.dirname(path)] = ie.parent_id
160
shamap[ie.file_id] = blob.id
161
elif ie.kind == "directory":
162
for pinv in parent_invs:
164
pie = pinv[ie.file_id]
165
except errors.NoSuchId:
168
if (pie.kind == ie.kind and
169
pie.children.keys() == ie.children.keys()):
171
shamap[ie.file_id] = idmap.lookup_tree_id(
172
ie.file_id, inv.revision_id)
173
except (NotImplementedError, KeyError):
178
new_trees[path] = ie.file_id
180
raise AssertionError(ie.kind)
182
for (path, ie), chunks in iter_files_bytes(
183
[(ie.file_id, ie.revision, (path, ie))
184
for (path, ie) in new_blobs]):
188
shamap[ie.file_id] = obj.id
190
for fid in unusual_modes:
191
new_trees[inv.id2path(fid)] = inv[fid].parent_id
195
items = new_trees.items()
197
for path, file_id in items:
198
parent_id = inv[file_id].parent_id
199
if parent_id is not None:
200
parent_path = urlutils.dirname(path)
201
new_trees[parent_path] = parent_id
202
trees[path] = file_id
204
def ie_to_hexsha(ie):
206
return shamap[ie.file_id]
208
# Not all cache backends store the tree information,
209
# calculate again from scratch
210
ret = directory_to_tree(ie, ie_to_hexsha, unusual_modes)
215
for path in sorted(trees.keys(), reverse=True):
216
ie = inv[trees[path]]
217
assert ie.kind == "directory"
218
obj = directory_to_tree(ie, ie_to_hexsha, unusual_modes)
221
shamap[ie.file_id] = obj.id
224
56
class BazaarObjectStore(BaseObjectStore):
225
57
"""A Git-style object store backed onto a Bazaar repository."""
243
73
heads = graph.heads(self.repository.all_revision_ids())
245
75
heads = set([stop_revision])
246
missing_revids = self._cache.idmap.missing_revisions(heads)
76
missing_revids = self._idmap.missing_revisions(heads)
248
78
parents = graph.get_parent_map(heads)
250
80
for p in parents.values():
251
81
todo.update([x for x in p if x not in missing_revids])
252
heads = self._cache.idmap.missing_revisions(todo)
82
heads = self._idmap.missing_revisions(todo)
253
83
missing_revids.update(heads)
254
84
if NULL_REVISION in missing_revids:
255
85
missing_revids.remove(NULL_REVISION)
256
missing_revids = self.repository.has_revisions(missing_revids)
257
if not missing_revids:
259
86
self.start_write_group()
261
88
pb = ui.ui_factory.nested_progress_bar()
263
90
for i, revid in enumerate(graph.iter_topo_order(missing_revids)):
264
trace.mutter('processing %r', revid)
265
91
pb.update("updating git map", i, len(missing_revids))
266
92
self._update_sha_map_revision(revid)
286
112
return self.mapping.export_commit(rev, tree_sha, parent_lookup)
288
def _revision_to_objects(self, rev, inv):
114
def _update_sha_map_revision(self, revid):
115
inv = self.repository.get_inventory(revid)
116
rev = self.repository.get_revision(revid)
289
117
unusual_modes = extract_unusual_modes(rev)
290
present_parents = self.repository.has_revisions(rev.parent_ids)
291
has_ghost_parents = (len(rev.parent_ids) < len(present_parents))
292
parent_invs = self.parent_invs_cache.get_inventories(
293
[p for p in rev.parent_ids if p in present_parents])
295
for path, obj, ie in _inventory_to_objects(inv, parent_invs,
296
self._cache.idmap, unusual_modes,
297
self.repository.iter_files_bytes, has_ghost_parents):
302
if not rev.parent_ids:
118
tree_sha = self._get_ie_sha1(inv.root, inv, unusual_modes)
306
119
commit_obj = self._revision_to_commit(rev, tree_sha)
308
foreign_revid, mapping = mapping_registry.parse_revision_id(
121
foreign_revid, mapping = mapping_registry.parse_revision_id(revid)
310
122
except errors.InvalidRevisionId:
313
_check_expected_sha(foreign_revid, commit_obj)
314
yield None, commit_obj, None
316
def _get_updater(self, rev):
317
return self._cache.get_updater(rev)
319
def _update_sha_map_revision(self, revid):
320
rev = self.repository.get_revision(revid)
321
inv = self.parent_invs_cache.get_inventory(rev.revision_id)
322
updater = self._get_updater(rev)
323
for path, obj, ie in self._revision_to_objects(rev, inv):
324
updater.add_object(obj, ie)
325
commit_obj = updater.finish()
328
def _get_blob(self, fileid, revision, expected_sha):
125
if foreign_revid != commit_obj.id:
126
if not "fix-shamap" in debug.debug_flags:
127
raise AssertionError("recreated git commit had different sha1: expected %s, got %s" % (foreign_revid, commit_obj.id))
128
self._idmap.add_entry(commit_obj.id, "commit", (revid, tree_sha))
130
def _check_expected_sha(self, expected_sha, object):
131
if expected_sha is None:
133
if len(expected_sha) == 40:
134
if expected_sha != object.sha().hexdigest():
135
raise AssertionError("Invalid sha for %r: %s" % (object, expected_sha))
136
elif len(expected_sha) == 20:
137
if expected_sha != object.sha().digest():
138
raise AssertionError("Invalid sha for %r: %s" % (object, sha_to_hex(expected_sha)))
140
raise AssertionError("Unknown length %d for %r" % (len(expected_sha), expected_sha))
142
def _get_ie_object(self, entry, inv, unusual_modes):
143
if entry.kind == "directory":
144
return self._get_tree(entry.file_id, inv.revision_id, inv, unusual_modes)
145
elif entry.kind == "symlink":
146
return self._get_blob_for_symlink(entry.symlink_target)
147
elif entry.kind == "file":
148
return self._get_blob_for_file(entry.file_id, entry.revision)
150
raise AssertionError("unknown entry kind '%s'" % entry.kind)
152
def _get_ie_object_or_sha1(self, entry, inv, unusual_modes):
153
if entry.kind == "directory":
155
return self._idmap.lookup_tree(entry.file_id, inv.revision_id), None
157
ret = self._get_ie_object(entry, inv, unusual_modes)
163
self._idmap.add_entry(hexsha, "tree", (entry.file_id, inv.revision_id))
165
elif entry.kind in ("file", "symlink"):
167
return self._idmap.lookup_blob(entry.file_id, entry.revision), None
169
ret = self._get_ie_object(entry, inv, unusual_modes)
170
self._idmap.add_entry(ret.id, "blob", (entry.file_id, entry.revision))
173
raise AssertionError("unknown entry kind '%s'" % entry.kind)
175
def _get_ie_sha1(self, entry, inv, unusual_modes):
176
return self._get_ie_object_or_sha1(entry, inv, unusual_modes)[0]
178
def _get_blob_for_symlink(self, symlink_target, expected_sha=None):
179
"""Return a Git Blob object for symlink.
181
:param symlink_target: target of symlink.
183
if type(symlink_target) == unicode:
184
symlink_target = symlink_target.encode('utf-8')
186
blob._text = symlink_target
187
self._check_expected_sha(expected_sha, blob)
190
def _get_blob_for_file(self, fileid, revision, expected_sha=None):
329
191
"""Return a Git Blob object from a fileid and revision stored in bzr.
331
193
:param fileid: File id of the text
335
197
chunks = self.repository.iter_files_bytes([(fileid, revision, None)]).next()[1]
336
blob.chunked = chunks
337
if blob.id != expected_sha:
338
# Perhaps it's a symlink ?
339
inv = self.parent_invs_cache.get_inventory(revision)
341
assert entry.kind == 'symlink'
342
blob = symlink_to_blob(entry)
343
_check_expected_sha(expected_sha, blob)
198
blob._text = "".join(chunks)
199
self._check_expected_sha(expected_sha, blob)
202
def _get_blob(self, fileid, revision, expected_sha=None):
203
"""Return a Git Blob object from a fileid and revision stored in bzr.
205
:param fileid: File id of the text
206
:param revision: Revision of the text
208
inv = self.repository.get_inventory(revision)
211
if entry.kind == 'file':
212
return self._get_blob_for_file(entry.file_id, entry.revision,
213
expected_sha=expected_sha)
214
elif entry.kind == 'symlink':
215
return self._get_blob_for_symlink(entry.symlink_target,
216
expected_sha=expected_sha)
346
220
def _get_tree(self, fileid, revid, inv, unusual_modes, expected_sha=None):
347
221
"""Return a Git Tree object from a file id and a revision stored in bzr.
349
223
:param fileid: fileid in the tree.
350
224
:param revision: Revision of the tree.
352
def get_ie_sha1(entry):
353
if entry.kind == "directory":
355
return self._cache.idmap.lookup_tree_id(entry.file_id)
356
except (NotImplementedError, KeyError):
357
obj = self._get_tree(entry.file_id, revid, inv,
363
elif entry.kind in ("file", "symlink"):
364
return self._cache.idmap.lookup_blob_id(entry.file_id, entry.revision)
366
raise AssertionError("unknown entry kind '%s'" % entry.kind)
367
tree = directory_to_tree(inv[fileid], get_ie_sha1, unusual_modes)
368
_check_expected_sha(expected_sha, tree)
226
tree = directory_to_tree(inv[fileid],
227
lambda ie: self._get_ie_sha1(ie, inv, unusual_modes),
229
self._check_expected_sha(expected_sha, tree)
232
def _get_commit(self, rev, tree_sha, expected_sha=None):
233
commit = self._revision_to_commit(rev, tree_sha)
234
self._check_expected_sha(expected_sha, commit)
371
237
def get_parents(self, sha):
372
238
"""Retrieve the parents of a Git commit by SHA1.
411
277
raise AssertionError("Unknown object type '%s'" % type)
415
283
def _lookup_git_sha(self, sha):
416
284
# See if sha is in map
418
return self._cache.idmap.lookup_git_sha(sha)
286
return self._idmap.lookup_git_sha(sha)
420
288
# if not, see if there are any unconverted revisions and add them
421
289
# to the map, search for sha in map again
422
290
self._update_sha_map()
423
return self._cache.idmap.lookup_git_sha(sha)
291
return self._idmap.lookup_git_sha(sha)
425
293
def __getitem__(self, sha):
426
if self._cache.content_cache is not None:
428
return self._cache.content_cache[sha]
431
294
(type, type_data) = self._lookup_git_sha(sha)
432
295
# convert object to git object
433
296
if type == "commit":
434
(revid, tree_sha) = type_data
436
rev = self.repository.get_revision(revid)
298
rev = self.repository.get_revision(type_data[0])
437
299
except errors.NoSuchRevision:
438
trace.mutter('entry for %s %s in shamap: %r, but not found in '
439
'repository', type, sha, type_data)
300
trace.mutter('entry for %s %s in shamap: %r, but not found in repository', type, sha, type_data)
440
301
raise KeyError(sha)
441
commit = self._revision_to_commit(rev, tree_sha)
442
_check_expected_sha(sha, commit)
302
return self._get_commit(rev, type_data[1], expected_sha=sha)
444
303
elif type == "blob":
445
(fileid, revision) = type_data
446
return self._get_blob(fileid, revision, expected_sha=sha)
304
return self._get_blob(type_data[0], type_data[1], expected_sha=sha)
447
305
elif type == "tree":
448
(fileid, revid) = type_data
450
inv = self.parent_invs_cache.get_inventory(revid)
451
rev = self.repository.get_revision(revid)
307
inv = self.repository.get_inventory(type_data[1])
308
rev = self.repository.get_revision(type_data[1])
452
309
except errors.NoSuchRevision:
453
310
trace.mutter('entry for %s %s in shamap: %r, but not found in repository', type, sha, type_data)
454
311
raise KeyError(sha)
455
312
unusual_modes = extract_unusual_modes(rev)
457
return self._get_tree(fileid, revid, inv, unusual_modes,
314
return self._get_tree(type_data[0], type_data[1], inv, unusual_modes,
459
316
except errors.NoSuchRevision:
460
317
raise KeyError(sha)
462
319
raise AssertionError("Unknown object type '%s'" % type)
464
def generate_pack_contents(self, have, want):
465
"""Iterate over the contents of a pack file.
467
:param have: List of SHA1s of objects that should not be sent
468
:param want: List of SHA1s of objects that should be sent
471
for commit_sha in have:
473
(type, (revid, tree_sha)) = self._lookup_git_sha(commit_sha)
477
assert type == "commit"
480
for commit_sha in want:
481
if commit_sha in have:
483
(type, (revid, tree_sha)) = self._lookup_git_sha(commit_sha)
484
assert type == "commit"
488
processed.update(pending)
489
next_map = self.repository.get_parent_map(pending)
491
for item in next_map.iteritems():
493
next_pending.update(p for p in item[1] if p not in processed)
494
pending = next_pending
495
if NULL_REVISION in todo:
496
todo.remove(NULL_REVISION)
497
trace.mutter('sending revisions %r', todo)
499
pb = ui.ui_factory.nested_progress_bar()
501
for i, revid in enumerate(todo):
502
pb.update("generating git objects", i, len(todo))
503
rev = self.repository.get_revision(revid)
504
inv = self.parent_invs_cache.get_inventory(revid)
505
for path, obj, ie in self._revision_to_objects(rev, inv):
506
ret.append((obj, path))