From 6e17f4f915e3fbd57513fa29ed15cb117aeef6c4 Mon Sep 17 00:00:00 2001
From: Zac Medico <zmedico@gentoo.org>
Date: Sun, 18 May 2008 22:18:08 +0000
Subject: [PATCH] * To avoid excessive IO on each call to LinkageMap.rebuild(),
 cache   NEEDED.* files via vardbapi.aux_get().

* Make vardbapi.aux_get() preserve newlines in multi-line metadata
  such as NEEDED.* and CONTENTS.

svn path=/main/trunk/; revision=10359
---
 pym/portage/dbapi/vartree.py | 37 ++++++++++++++++++++++++++++--------
 1 file changed, 29 insertions(+), 8 deletions(-)

diff --git a/pym/portage/dbapi/vartree.py b/pym/portage/dbapi/vartree.py
index edd588e6a..8506a23e4 100644
--- a/pym/portage/dbapi/vartree.py
+++ b/pym/portage/dbapi/vartree.py
@@ -137,7 +137,9 @@ class LinkageMap(object):
 		obj_properties = {}
 		lines = []
 		for cpv in self._dbapi.cpv_all():
-			lines += grabfile(self._dbapi.getpath(cpv, filename="NEEDED.ELF.2"))
+			lines += self._dbapi.aux_get(cpv, ["NEEDED.ELF.2"])[0].split('\n')
+		# Cache NEEDED.* files avoid doing excessive IO for every rebuild.
+		self._dbapi.flush_cache()
 		
 		if include_file:
 			lines += grabfile(include_file)
@@ -236,6 +238,9 @@ class vardbapi(dbapi):
 	_excluded_dirs = re.compile(r'^(\..*|-MERGING-.*|' + \
 		"|".join(_excluded_dirs) + r')$')
 
+	_aux_cache_keys_re = re.compile(r'^NEEDED\..*$')
+	_aux_multi_line_re = re.compile(r'^(CONTENTS|NEEDED\..*)$')
+
 	def __init__(self, root, categories=None, settings=None, vartree=None):
 		"""
 		The categories parameter is unused since the dbapi class
@@ -569,8 +574,17 @@ class vardbapi(dbapi):
 		unrecognized, the cache will simple be recreated from scratch (it is
 		completely disposable).
 		"""
-		if not self._aux_cache_keys.intersection(wants):
+		cache_these_wants = self._aux_cache_keys.intersection(wants)
+		for x in wants:
+			if self._aux_cache_keys_re.match(x) is not None:
+				cache_these_wants.add(x)
+
+		if not cache_these_wants:
 			return self._aux_get(mycpv, wants)
+
+		cache_these = set(self._aux_cache_keys)
+		cache_these.update(cache_these_wants)
+
 		if self._aux_cache is None:
 			try:
 				f = open(self._aux_cache_filename)
@@ -600,29 +614,33 @@ class vardbapi(dbapi):
 		pkg_data = self._aux_cache["packages"].get(mycpv)
 		mydata = {}
 		cache_valid = False
+		cache_incomplete = False
 		if pkg_data:
 			cache_mtime, metadata = pkg_data
 			cache_valid = cache_mtime == mydir_mtime
 		if cache_valid:
-			cache_incomplete = self._aux_cache_keys.difference(metadata)
+			cache_incomplete = cache_these.difference(metadata)
 			if cache_incomplete:
 				# Allow self._aux_cache_keys to change without a cache version
 				# bump and efficiently recycle partial cache whenever possible.
 				cache_valid = False
 				pull_me = cache_incomplete.union(wants)
 			else:
-				pull_me = set(wants).difference(self._aux_cache_keys)
+				pull_me = set(wants).difference(cache_these)
 			mydata.update(metadata)
 		else:
-			pull_me = self._aux_cache_keys.union(wants)
+			pull_me = cache_these
+
 		if pull_me:
 			# pull any needed data and cache it
 			aux_keys = list(pull_me)
 			for k, v in izip(aux_keys, self._aux_get(mycpv, aux_keys)):
 				mydata[k] = v
-			if not cache_valid:
+			if not cache_valid or cache_incomplete:
 				cache_data = {}
-				for aux_key in self._aux_cache_keys:
+				if cache_incomplete:
+					cache_data.update(metadata)
+				for aux_key in cache_these:
 					cache_data[aux_key] = mydata[aux_key]
 				self._aux_cache["packages"][mycpv] = (mydir_mtime, cache_data)
 				self._aux_cache["modified"] = True
@@ -646,7 +664,10 @@ class vardbapi(dbapi):
 					myd = myf.read()
 				finally:
 					myf.close()
-				myd = " ".join(myd.split())
+				# Preserve \n for metadata that is known to
+				# contain multiple lines.
+				if self._aux_multi_line_re.match(x) is None:
+					myd = " ".join(myd.split())
 			except IOError:
 				myd = ""
 			if x == "EAPI" and not myd:
-- 
2.26.2