class PersistenceManager:
"""The PersistenceManager allows us to effectively store SVNCommits
to disk and retrieve them later using only their subversion revision
number as the key. It also returns the subversion revision number
for a given CVSRevision's unique key.
All information pertinent to each SVNCommit is stored in a series of
on-disk databases so that SVNCommits can be retrieved on-demand.
MODE is one of the constants DB_OPEN_NEW or DB_OPEN_READ.
In 'new' mode, PersistenceManager will initialize a new set of on-disk
databases and be fully-featured.
In 'read' mode, PersistenceManager will open existing on-disk databases
and the set_* methods will be unavailable."""
def __init__(self, mode):
self.mode = mode
if mode not in (DB_OPEN_NEW, DB_OPEN_READ):
raise RuntimeError, "Invalid 'mode' argument to PersistenceManager"
primer = (
SVNInitialProjectCommit,
SVNPrimaryCommit,
SVNPostCommit,
SVNBranchCommit,
SVNTagCommit,
)
serializer = PrimedPickleSerializer(primer)
self.svn_commit_db = IndexedDatabase(
artifact_manager.get_temp_file(config.SVN_COMMITS_INDEX_TABLE),
artifact_manager.get_temp_file(config.SVN_COMMITS_STORE),
mode, serializer)
self.cvs2svn_db = RecordTable(
artifact_manager.get_temp_file(config.CVS_REVS_TO_SVN_REVNUMS),
mode, SignedIntegerPacker(SVN_INVALID_REVNUM))
def get_svn_revnum(self, cvs_rev_id):
"""Return the Subversion revision number in which CVS_REV_ID was
committed, or SVN_INVALID_REVNUM if there is no mapping for
CVS_REV_ID."""
return self.cvs2svn_db.get(cvs_rev_id, SVN_INVALID_REVNUM)
def get_svn_commit(self, svn_revnum):
"""Return an SVNCommit that corresponds to SVN_REVNUM.
If no SVNCommit exists for revnum SVN_REVNUM, then return None."""
return self.svn_commit_db.get(svn_revnum, None)
def put_svn_commit(self, svn_commit):
"""Record the bidirectional mapping between SVN_REVNUM and
CVS_REVS and record associated attributes."""
if self.mode == DB_OPEN_READ:
raise RuntimeError, \
'Write operation attempted on read-only PersistenceManager'
self.svn_commit_db[svn_commit.revnum] = svn_commit
if isinstance(svn_commit, SVNRevisionCommit):
for cvs_rev in svn_commit.cvs_revs:
self.cvs2svn_db[cvs_rev.id] = svn_commit.revnum
def close(self):
self.cvs2svn_db.close()
self.cvs2svn_db = None
self.svn_commit_db.close()
self.svn_commit_db = None
