From 62d27e0bd56b765bd32c22f4062e5b864fcf29c0 Mon Sep 17 00:00:00 2001
From: Panu Matilainen <pmatilai@redhat.com>
Date: Tue, 24 Feb 2009 09:21:20 +0200
Subject: [PATCH] Calculate files in transaction just once - move file counting
 from rpmtsRun() and rpmFindBaseNamesInDB() to single   helper, store result
 in rpmts struct

---
 lib/rpmts_internal.h |  1 +
 lib/transaction.c    | 66 ++++++++++++++++++----------------------------------
 2 files changed, 24 insertions(+), 43 deletions(-)

diff --git a/lib/rpmts_internal.h b/lib/rpmts_internal.h
index 379bb85..97bb7e0 100644
--- a/lib/rpmts_internal.h
+++ b/lib/rpmts_internal.h
@@ -52,6 +52,7 @@ struct rpmts_s {
     unsigned int filesystemCount;	/*!< No. of mounted filesystems. */
     const char ** filesystems;	/*!< Mounted filesystem names. */
     rpmDiskSpaceInfo dsi;	/*!< Per filesystem disk/inode usage. */
+    uint64_t fileCount;		/*!< Total number of files in transaction */
 
     rpmdb rdb;			/*!< Install database handle. */
     int dbmode;			/*!< Install database open mode. */
diff --git a/lib/transaction.c b/lib/transaction.c
index 9b7fb12..33195a4 100644
--- a/lib/transaction.c
+++ b/lib/transaction.c
@@ -42,6 +42,22 @@ static int osOkay(const char * pkgOs)
     return (rpmMachineScore(RPM_MACHTABLE_INSTOS, pkgOs) ? 1 : 0);
 }
 
+/* Calculate total number of files involved in transaction */
+static uint64_t countFiles(rpmts ts)
+{
+    uint64_t fc = 0;
+    rpmtsi pi = rpmtsiInit(ts);
+    rpmte p;
+    rpmfi fi;
+    while ((p = rpmtsiNext(pi, 0)) != NULL) {
+	if ((fi = rpmteFI(p)) == NULL)
+	    continue;   /* XXX can't happen */
+	fc += rpmfiFC(fi);
+    }
+    pi = rpmtsiFree(pi);
+    return fc;
+}
+
 /**
  * handleInstInstalledFiles.
  * @param ts		transaction set
@@ -604,22 +620,12 @@ rpmdbMatchIterator rpmFindBaseNamesInDB(rpmts ts)
 {
     rpmtsi pi;  rpmte p;
     rpmfi fi;
-    int fc=0;
     rpmdbMatchIterator mi;
     int i, xx;
     const char * baseName;
 
-    /* get number of files in transaction */
-    // XXX move to ts
-    pi = rpmtsiInit(ts);
-    while ((p = rpmtsiNext(pi, 0)) != NULL) {
-	if ((fi = rpmteFI(p)) == NULL)
-	    continue;   /* XXX can't happen */
-	fc += rpmfiFC(fi);
-    }
-    pi = rpmtsiFree(pi);
-
-    rpmStringSet baseNames = rpmStringSetCreate(fc, hashFunctionString, strcmp, NULL);
+    rpmStringSet baseNames = rpmStringSetCreate(ts->fileCount, 
+					hashFunctionString, strcmp, NULL);
 
     mi = rpmdbInitIterator(rpmtsGetRdb(ts), RPMTAG_BASENAMES, NULL, 0);
 
@@ -896,7 +902,6 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
     rpm_color_t tscolor = rpmtsColor(ts);
     int i;
     int rc = 0;
-    int totalFileCount = 0;
     rpmfi fi;
     fingerPrintCache fpc;
     rpmps ps;
@@ -971,11 +976,7 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
      * For packages being installed:
      * - verify package arch/os.
      * - verify package epoch:version-release is newer.
-     * - count files.
-     * For packages being removed:
-     * - count files.
      */
-
     rpmlog(RPMLOG_DEBUG, "sanity checking %d elements\n", rpmtsNElements(ts));
     ps = rpmtsProblems(ts);
     /* The ordering doesn't matter here */
@@ -983,11 +984,6 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
     /* XXX Only added packages need be checked. */
     while ((p = rpmtsiNext(pi, TR_ADDED)) != NULL) {
 	rpmdbMatchIterator mi;
-	int fc;
-
-	if ((fi = rpmteFI(p)) == NULL)
-	    continue;	/* XXX can't happen */
-	fc = rpmfiFC(fi);
 
 	if (!(rpmtsFilterFlags(ts) & RPMPROB_FILTER_IGNOREARCH))
 	    if (!archOkay(rpmteA(p)))
@@ -1035,27 +1031,10 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
 	    }
 	    mi = rpmdbFreeIterator(mi);
 	}
-
-	/* Count no. of files (if any). */
-	totalFileCount += fc;
-
     }
     pi = rpmtsiFree(pi);
     ps = rpmpsFree(ps);
 
-    /* The ordering doesn't matter here */
-    pi = rpmtsiInit(ts);
-    while ((p = rpmtsiNext(pi, TR_REMOVED)) != NULL) {
-	int fc;
-
-	if ((fi = rpmteFI(p)) == NULL)
-	    continue;	/* XXX can't happen */
-	fc = rpmfiFC(fi);
-
-	totalFileCount += fc;
-    }
-    pi = rpmtsiFree(pi);
-
 
     /* Run pre-transaction scripts, but only if there are no known
      * problems up to this point and not disabled otherwise. */
@@ -1075,7 +1054,8 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
      * calling fpLookupList only once. I'm not sure that the speedup is
      * worth the trouble though.
      */
-    rpmlog(RPMLOG_DEBUG, "computing %d file fingerprints\n", totalFileCount);
+    ts->fileCount = countFiles(ts);
+    rpmlog(RPMLOG_DEBUG, "computing %" PRIu64 " file fingerprints\n", ts->fileCount);
 
     numAdded = numRemoved = 0;
     pi = rpmtsiInit(ts);
@@ -1114,10 +1094,10 @@ int rpmtsRun(rpmts ts, rpmps okProbs, rpmprobFilterFlags ignoreSet)
 	(void) rpmtsSetChrootDone(ts, 1);
     }
 
-    ts->ht = rpmFpHashCreate(totalFileCount/2+1, fpHashFunction, fpEqual,
+    ts->ht = rpmFpHashCreate(ts->fileCount/2+1, fpHashFunction, fpEqual,
 			     NULL, NULL);
-    rpmFpHash symlinks = rpmFpHashCreate(totalFileCount/16+16, fpHashFunction, fpEqual, NULL, NULL);
-    fpc = fpCacheCreate(totalFileCount/2 + 10001);
+    rpmFpHash symlinks = rpmFpHashCreate(ts->fileCount/16+16, fpHashFunction, fpEqual, NULL, NULL);
+    fpc = fpCacheCreate(ts->fileCount/2 + 10001);
 
     /* ===============================================
      * Add fingerprint for each file not skipped.
-- 
2.7.4