Patchwork D7844: nodemap: code to parse the persistent binary nodemap data

login
register
mail settings
Submitter phabricator
Date Jan. 31, 2020, 3:29 p.m.
Message ID <726052b6b1862b2c5fad8baba480a246@localhost.localdomain>
Download mbox | patch
Permalink /patch/44808/
State Not Applicable
Headers show

Comments

phabricator - Jan. 31, 2020, 3:29 p.m.
marmoute edited the summary of this revision.
marmoute retitled this revision from "nodemap: code to parse a serialized nodemap" to "nodemap: code to parse the persistent binary nodemap data".
marmoute updated this revision to Diff 19760.

REPOSITORY
  rHG Mercurial

CHANGES SINCE LAST UPDATE
  https://phab.mercurial-scm.org/D7844?vs=19298&id=19760

CHANGES SINCE LAST ACTION
  https://phab.mercurial-scm.org/D7844/new/

REVISION DETAIL
  https://phab.mercurial-scm.org/D7844

AFFECTED FILES
  mercurial/revlogutils/nodemap.py

CHANGE DETAILS




To: marmoute, #hg-reviewers
Cc: mercurial-devel

Patch

diff --git a/mercurial/revlogutils/nodemap.py b/mercurial/revlogutils/nodemap.py
--- a/mercurial/revlogutils/nodemap.py
+++ b/mercurial/revlogutils/nodemap.py
@@ -308,3 +308,33 @@ 
         return block_map[id(item)]
     else:
         return _transform_rev(item)
+
+
+def parse_data(data):
+    """parse parse nodemap data into a nodemap Trie"""
+    if (len(data) % S_BLOCK.size) != 0:
+        msg = "nodemap data size is not a multiple of block size (%d): %d"
+        raise error.Abort(msg % (S_BLOCK.size, len(data)))
+    if not data:
+        return Block()
+    block_map = {}
+    for i in range(0, len(data), S_BLOCK.size):
+        block = _parse_block(block_map, data[i : i + S_BLOCK.size])
+    return block
+
+
+def _parse_block(block_map, block_data):
+    """Parse the persisted version of an individual block
+    """
+    block = Block()
+    ondisk_id = len(block_map)
+    block_map[ondisk_id] = block
+    values = S_BLOCK.unpack(block_data)
+    for idx, v in enumerate(values):
+        if v == NO_ENTRY:
+            continue
+        elif v >= 0:
+            block[idx] = block_map[v]
+        else:
+            block[idx] = _transform_rev(v)
+    return block