mercurial/revlogutils/nodemap.py

51655:8e24f4f86ba8

    use_mmap = revlog.opener.options.get(b"persistent-nodemap.mmap")

    test_race_hook_1()
    try:
        with revlog.opener(filename) as fd:
            if use_mmap:
                try:
                    data = util.buffer(util.mmapread(fd, data_length))
                except ValueError:
                    # raised when the read file is too small
                    data = b''

            tr.add(datafile, target_docket.data_length)
            with revlog.opener(datafile, b'r+') as fd:
                fd.seek(target_docket.data_length)
                fd.write(data)
                if feed_data:
                    if use_mmap:
                        fd.flush()
                        new_data = util.buffer(util.mmapread(fd, new_length))
                    else:
                        fd.seek(0)
                        new_data = fd.read(new_length)

        # simply empty them.
        tr.addabort(callback_id, abortck)
        with revlog.opener(datafile, b'w+') as fd:
            fd.write(data)
            if feed_data:
                if use_mmap:
                    fd.flush()
                    new_data = util.buffer(util.mmapread(fd, len(data)))
                else:
                    new_data = data
        target_docket.data_length = len(data)

51656:21442f87f7fa

    use_mmap = revlog.opener.options.get(b"persistent-nodemap.mmap")

    test_race_hook_1()
    try:
        with revlog.opener(filename) as fd:
            if use_mmap and revlog.opener.is_mmap_safe(filename):
                try:
                    data = util.buffer(util.mmapread(fd, data_length))
                except ValueError:
                    # raised when the read file is too small
                    data = b''

            tr.add(datafile, target_docket.data_length)
            with revlog.opener(datafile, b'r+') as fd:
                fd.seek(target_docket.data_length)
                fd.write(data)
                if feed_data:
                    if use_mmap and revlog.opener.is_mmap_safe(datafile):
                        fd.flush()
                        new_data = util.buffer(util.mmapread(fd, new_length))
                    else:
                        fd.seek(0)
                        new_data = fd.read(new_length)

        # simply empty them.
        tr.addabort(callback_id, abortck)
        with revlog.opener(datafile, b'w+') as fd:
            fd.write(data)
            if feed_data:
                if use_mmap and revlog.opener.is_mmap_safe(datafile):
                    fd.flush()
                    new_data = util.buffer(util.mmapread(fd, len(data)))
                else:
                    new_data = data
        target_docket.data_length = len(data)