mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 23:38:50 +03:00
ab3a7cb21f
Summary: In preparation for merging fb-mercurial sources to the Eden repository, move everything from the top-level directory into an `eden/scm` subdirectory.
74 lines
2.4 KiB
Python
74 lines
2.4 KiB
Python
# Portions Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# dirstateguard.py - class to allow restoring dirstate after failure
|
|
#
|
|
# Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from . import error, util
|
|
from .i18n import _
|
|
|
|
|
|
class dirstateguard(util.transactional):
|
|
"""Restore dirstate at unexpected failure.
|
|
|
|
At the construction, this class does:
|
|
|
|
- write current ``repo.dirstate`` out, and
|
|
- save ``.hg/dirstate`` into the backup file
|
|
|
|
This restores ``.hg/dirstate`` from backup file, if ``release()``
|
|
is invoked before ``close()``.
|
|
|
|
This just removes the backup file at ``close()`` before ``release()``.
|
|
"""
|
|
|
|
def __init__(self, repo, name):
|
|
self._repo = repo
|
|
self._active = False
|
|
self._closed = False
|
|
self._backupname = "dirstate.backup.%s.%d" % (name, id(self))
|
|
repo.dirstate.savebackup(repo.currenttransaction(), self._backupname)
|
|
self._active = True
|
|
|
|
def __del__(self):
|
|
if self._active: # still active
|
|
# this may occur, even if this class is used correctly:
|
|
# for example, releasing other resources like transaction
|
|
# may raise exception before ``dirstateguard.release`` in
|
|
# ``release(tr, ....)``.
|
|
self._abort()
|
|
|
|
def close(self):
|
|
if not self._active: # already inactivated
|
|
msg = _("can't close already inactivated backup: %s") % self._backupname
|
|
raise error.Abort(msg)
|
|
|
|
self._repo.dirstate.clearbackup(
|
|
self._repo.currenttransaction(), self._backupname
|
|
)
|
|
self._active = False
|
|
self._closed = True
|
|
|
|
def _abort(self):
|
|
self._repo.dirstate.restorebackup(
|
|
self._repo.currenttransaction(), self._backupname
|
|
)
|
|
self._active = False
|
|
|
|
def release(self):
|
|
if not self._closed:
|
|
if not self._active: # already inactivated
|
|
msg = (
|
|
_("can't release already inactivated backup: %s") % self._backupname
|
|
)
|
|
raise error.Abort(msg)
|
|
self._abort()
|