1 files changed, 106 insertions, 0 deletions
diff --git a/cvs2svn_lib/persistence_manager.py b/cvs2svn_lib/persistence_manager.py
new file mode 100644
index 0000000..8a622ab
--- /dev/null
+++ b/cvs2svn_lib/persistence_manager.py
@@ -0,0 +1,106 @@
+# (Be in -*- python -*- mode.)
+#
+# ====================================================================
+# Copyright (c) 2000-2008 CollabNet.  All rights reserved.
+#
+# This software is licensed as described in the file COPYING, which
+# you should have received as part of this distribution.  The terms
+# are also available at http://subversion.tigris.org/license-1.html.
+# If newer versions of this license are posted there, you may use a
+# newer version instead, at your option.
+#
+# This software consists of voluntary contributions made by many
+# individuals.  For exact contribution history, see the revision
+# history and logs, available at http://cvs2svn.tigris.org/.
+# ====================================================================
+
+"""This module contains class PersistenceManager."""
+
+
+from cvs2svn_lib import config
+from cvs2svn_lib.common import DB_OPEN_NEW
+from cvs2svn_lib.common import DB_OPEN_READ
+from cvs2svn_lib.common import SVN_INVALID_REVNUM
+from cvs2svn_lib.artifact_manager import artifact_manager
+from cvs2svn_lib.record_table import SignedIntegerPacker
+from cvs2svn_lib.record_table import RecordTable
+from cvs2svn_lib.serializer import PrimedPickleSerializer
+from cvs2svn_lib.database import IndexedDatabase
+from cvs2svn_lib.svn_commit import SVNRevisionCommit
+from cvs2svn_lib.svn_commit import SVNInitialProjectCommit
+from cvs2svn_lib.svn_commit import SVNPrimaryCommit
+from cvs2svn_lib.svn_commit import SVNBranchCommit
+from cvs2svn_lib.svn_commit import SVNTagCommit
+from cvs2svn_lib.svn_commit import SVNPostCommit
+
+
+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
+
+