Maintain annotated tags throughout incremental runs

Author: Vampire

src/main.cpp

@@ -211,6 +211,7 @@ int main(int argc, char **argv)
         repositories.insert(rule.name, repo);
 
         int repo_next = repo->setupIncremental(cutoff);
+        repo->restoreAnnotatedTags();
 
         /*
   * cutoff < resume_from => error exit eventually

src/repository.cpp

@@ -32,6 +32,15 @@ static const mark_t maxMark = ULONG_MAX;
 class FastImportRepository : public Repository
 {
 public:
+    struct AnnotatedTag
+    {
+        QString supportingRef;
+        QByteArray svnprefix;
+        QByteArray author;
+        QByteArray log;
+        uint dt;
+        int revnum;
+    };
     class Transaction : public Repository::Transaction
     {
         Q_DISABLE_COPY(Transaction)
@@ -69,6 +78,7 @@ class FastImportRepository : public Repository
     };
     FastImportRepository(const Rules::Repository &rule);
     int setupIncremental(int &cutoff);
+    void restoreAnnotatedTags();
     void restoreLog();
     ~FastImportRepository();
 
@@ -100,15 +110,6 @@ class FastImportRepository : public Repository
         QVector<int> marks;
         QByteArray note;
     };
-    struct AnnotatedTag
-    {
-        QString supportingRef;
-        QByteArray svnprefix;
-        QByteArray author;
-        QByteArray log;
-        uint dt;
-        int revnum;
-    };
 
     QHash<QString, Branch> branches;
     QHash<QString, AnnotatedTag> annotatedTags;
@@ -183,6 +184,7 @@ class ForwardingRepository : public Repository
     ForwardingRepository(const QString &n, Repository *r, const QString &p) : name(n), repo(r), prefix(p) {}
 
     int setupIncremental(int &) { return 1; }
+    void restoreAnnotatedTags() {}
     void restoreLog() {}
 
     void reloadBranches() { return repo->reloadBranches(); }
@@ -248,6 +250,33 @@ class ProcessCache: QLinkedList<FastImportRepository *>
 };
 static ProcessCache processCache;
 
+QDataStream &operator<<(QDataStream &out, const FastImportRepository::AnnotatedTag &annotatedTag)
+{
+    out << annotatedTag.supportingRef
+        << annotatedTag.svnprefix
+        << annotatedTag.author
+        << annotatedTag.log
+        << (quint64) annotatedTag.dt
+        << (qint64) annotatedTag.revnum;
+    return out;
+}
+
+QDataStream &operator>>(QDataStream &in, FastImportRepository::AnnotatedTag &annotatedTag)
+{
+    quint64 dt;
+    qint64 revnum;
+
+    in >> annotatedTag.supportingRef
+       >> annotatedTag.svnprefix
+       >> annotatedTag.author
+       >> annotatedTag.log
+       >> dt
+       >> revnum;
+    annotatedTag.dt = (uint) dt;
+    annotatedTag.revnum = (int) revnum;
+    return in;
+}
+
 Repository *createRepository(const Rules::Repository &rule, const QHash<QString, Repository *> &repositories)
 {
     if (rule.forwardTo.isEmpty())
@@ -267,6 +296,13 @@ static QString marksFileName(QString name)
     return name;
 }
 
+static QString annotatedTagsFileName(QString name)
+{
+    name.replace('/', '_');
+    name.prepend("annotatedTags-");
+    return name;
+}
+
 FastImportRepository::FastImportRepository(const Rules::Repository &rule)
     : name(rule.name), prefix(rule.forwardTo), fastImport(name), commitCount(0), outstandingTransactions(0),
       last_commit_mark(0), next_file_mark(maxMark - 1), processHasStarted(false)
@@ -451,6 +487,17 @@ int FastImportRepository::setupIncremental(int &cutoff)
     return cutoff;
 }
 
+void FastImportRepository::restoreAnnotatedTags()
+{
+    QFile annotatedTagsFile(name + "/" + annotatedTagsFileName(name));
+    if (!annotatedTagsFile.exists())
+        return;
+    annotatedTagsFile.open(QIODevice::ReadOnly);
+    QDataStream annotatedTagsStream(&annotatedTagsFile);
+    annotatedTagsStream >> annotatedTags;
+    annotatedTagsFile.close();
+}
+
 void FastImportRepository::restoreLog()
 {
     QString file = logFileName(name);
@@ -710,7 +757,13 @@ void FastImportRepository::finalizeTags()
     if (annotatedTags.isEmpty())
         return;
 
-    printf("Finalising tags for %s...", qPrintable(name));
+    QFile annotatedTagsFile(name + "/" + annotatedTagsFileName(name));
+    annotatedTagsFile.open(QIODevice::WriteOnly);
+    QDataStream annotatedTagsStream(&annotatedTagsFile);
+    annotatedTagsStream << annotatedTags;
+    annotatedTagsFile.close();
+
+    printf("Finalising annotated tags for %s...", qPrintable(name));
     startFastImport();
 
     QHash<QString, AnnotatedTag>::ConstIterator it = annotatedTags.constBegin();

src/repository.h

@@ -116,6 +116,7 @@ class Repository
                                 const QByteArray &commit = QByteArray()) = 0;
     };
     virtual int setupIncremental(int &cutoff) = 0;
+    virtual void restoreAnnotatedTags() = 0;
     virtual void restoreLog() = 0;
     virtual ~Repository() {}