diff src/org/tmatesoft/hg/internal/RevlogStream.java @ 198:33a7d76f067b

Performance optimization: reduce memory to keep revlog cached info
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Wed, 20 Apr 2011 05:40:14 +0200
parents 71ddbf8603e8
children 706bcc7cfee4
line wrap: on
line diff
--- a/src/org/tmatesoft/hg/internal/RevlogStream.java	Tue Apr 19 03:49:29 2011 +0200
+++ b/src/org/tmatesoft/hg/internal/RevlogStream.java	Wed Apr 20 05:40:14 2011 +0200
@@ -22,7 +22,6 @@
 import java.io.File;
 import java.io.IOException;
 import java.util.ArrayList;
-import java.util.Collections;
 import java.util.LinkedList;
 import java.util.List;
 
@@ -33,7 +32,7 @@
 
 /**
  * ? Single RevlogStream per file per repository with accessor to record access session (e.g. with back/forward operations), 
- * or numerous RevlogStream with separate representation of the underlaying data (cached, lazy ChunkStream)?
+ * or numerous RevlogStream with separate representation of the underlying data (cached, lazy ChunkStream)?
  * 
  * @see http://mercurial.selenic.com/wiki/Revlog
  * @see http://mercurial.selenic.com/wiki/RevlogNG
@@ -43,7 +42,16 @@
  */
 public class RevlogStream {
 
-	private List<IndexEntry> index; // indexed access highly needed
+	/*
+	 * makes sense for index with inline data only - actual offset of the record in the .i file (record entry + revision * record size))
+	 * 
+	 * long[] in fact (there are 8-bytes field in the revlog)
+	 * However, (a) DataAccess currently doesn't operate with long seek/length
+	 * and, of greater significance, (b) files with inlined data are designated for smaller files,  
+	 * guess, about 130 Kb, and offset there won't ever break int capacity
+	 */
+	private int[] indexRecordOffset;  
+	private int[] baseRevisions;
 	private boolean inline = false;
 	private final File indexFile;
 	private final DataAccessProvider dataAccess;
@@ -66,7 +74,7 @@
 	
 	public int revisionCount() {
 		initOutline();
-		return index.size();
+		return baseRevisions.length;
 	}
 	
 	public int dataLength(int revision) {
@@ -78,7 +86,7 @@
 			revision = indexSize - 1;
 		}
 		try {
-			int recordOffset = inline ? index.get(revision).getIntOffset() : revision * REVLOGV1_RECORD_SIZE;
+			int recordOffset = getIndexOffsetInt(revision);
 			daIndex.seek(recordOffset + 12); // 6+2+4
 			int actualLen = daIndex.readInt();
 			return actualLen; 
@@ -100,7 +108,7 @@
 		}
 		DataAccess daIndex = getIndexStream();
 		try {
-			int recordOffset = inline ? index.get(revision).getIntOffset() : revision * REVLOGV1_RECORD_SIZE;
+			int recordOffset = getIndexOffsetInt(revision);
 			daIndex.seek(recordOffset + 32);
 			byte[] rv = new byte[20];
 			daIndex.readBytes(rv, 0, 20);
@@ -123,7 +131,7 @@
 		}
 		DataAccess daIndex = getIndexStream();
 		try {
-			int recordOffset = inline ? index.get(revision).getIntOffset() : revision * REVLOGV1_RECORD_SIZE;
+			int recordOffset = getIndexOffsetInt(revision);
 			daIndex.seek(recordOffset + 20);
 			int linkRev = daIndex.readInt();
 			return linkRev;
@@ -174,7 +182,7 @@
 	// ? boolean needsNodeid
 	public void iterate(int start, int end, boolean needData, Inspector inspector) {
 		initOutline();
-		final int indexSize = index.size();
+		final int indexSize = revisionCount();
 		if (indexSize == 0) {
 			return;
 		}
@@ -202,22 +210,21 @@
 			DataAccess lastUserData = null;
 			int i;
 			boolean extraReadsToBaseRev = false;
-			if (needData && index.get(start).baseRevision < start) {
-				i = index.get(start).baseRevision;
+			if (needData && getBaseRevision(start) < start) {
+				i = getBaseRevision(start);
 				extraReadsToBaseRev = true;
 			} else {
 				i = start;
 			}
 			
-			daIndex.seek(inline ? index.get(i).getIntOffset() : i * REVLOGV1_RECORD_SIZE);
+			daIndex.seek(getIndexOffsetInt(i));
 			for (; i <= end; i++ ) {
 				if (inline && needData) {
 					// inspector reading data (though FilterDataAccess) may have affected index position
-					daIndex.seek(index.get(i).getIntOffset());
+					daIndex.seek(getIndexOffsetInt(i));
 				}
 				long l = daIndex.readLong(); // 0
-				@SuppressWarnings("unused")
-				long offset = l >>> 16;
+				long offset = i == 0 ? 0 : (l >>> 16);
 				@SuppressWarnings("unused")
 				int flags = (int) (l & 0X0FFFF);
 				int compressedLen = daIndex.readInt(); // +8
@@ -232,12 +239,13 @@
 				DataAccess userDataAccess = null;
 				if (needData) {
 					final byte firstByte;
-					int streamOffset = index.get(i).getIntOffset();
+					int streamOffset;
 					DataAccess streamDataAccess;
 					if (inline) {
 						streamDataAccess = daIndex;
-						streamOffset += REVLOGV1_RECORD_SIZE; // don't need to do seek as it's actual position in the index stream
+						streamOffset = getIndexOffsetInt(i) + REVLOGV1_RECORD_SIZE; // don't need to do seek as it's actual position in the index stream
 					} else {
+						streamOffset = (int) offset;
 						streamDataAccess = daData;
 						daData.seek(streamOffset);
 					}
@@ -291,12 +299,24 @@
 			}
 		}
 	}
-	
+
+	private int getBaseRevision(int revision) {
+		return baseRevisions[revision];
+	}
+
+	/**
+	 * @return  offset of the revision's record in the index (.i) stream
+	 */
+	private int getIndexOffsetInt(int revision) {
+		return inline ? indexRecordOffset[revision] : revision * REVLOGV1_RECORD_SIZE;
+	}
+
 	private void initOutline() {
-		if (index != null && !index.isEmpty()) {
+		if (baseRevisions != null && baseRevisions.length > 0) {
 			return;
 		}
-		ArrayList<IndexEntry> res = new ArrayList<IndexEntry>();
+		ArrayList<Integer> resBases = new ArrayList<Integer>();
+		ArrayList<Integer> resOffsets = new ArrayList<Integer>();
 		DataAccess da = getIndexStream();
 		try {
 			int versionField = da.readInt();
@@ -315,17 +335,25 @@
 //				int parent1Revision = di.readInt();
 //				int parent2Revision = di.readInt();
 //				byte[] nodeid = new byte[32];
+				resBases.add(baseRevision);
 				if (inline) {
-					res.add(new IndexEntry(offset + REVLOGV1_RECORD_SIZE * res.size(), baseRevision));
+					int o = (int) offset;
+					if (o != offset) {
+						// just in case, can't happen, ever, unless HG (or some other bad tool) produces index file 
+						// with inlined data of size greater than 2 Gb.
+						throw new HgBadStateException("Data too big, offset didn't fit to sizeof(int)");
+					}
+					resOffsets.add(o + REVLOGV1_RECORD_SIZE * resOffsets.size());
 					da.skip(3*4 + 32 + compressedLen); // Check: 44 (skip) + 20 (read) = 64 (total RevlogNG record size)
 				} else {
-					res.add(new IndexEntry(offset, baseRevision));
 					da.skip(3*4 + 32);
 				}
 				if (da.isEmpty()) {
 					// fine, done then
-					res.trimToSize();
-					index = res;
+					baseRevisions = toArray(resBases);
+					if (inline) {
+						indexRecordOffset = toArray(resOffsets);
+					}
 					break;
 				} else {
 					// start reading next record
@@ -335,36 +363,21 @@
 			}
 		} catch (IOException ex) {
 			ex.printStackTrace(); // log error
-			// too bad, no outline then.
-			index = Collections.emptyList();
+			// too bad, no outline then, but don't fail with NPE
+			baseRevisions = new int[0];
 		} finally {
 			da.done();
 		}
-		
 	}
 	
-
-	// perhaps, package-local or protected, if anyone else from low-level needs them
-	// XXX think over if we should keep offset in case of separate data file - we read the field anyway. Perhaps, distinct entry classes for Inline and non-inline indexes?
-	private static class IndexEntry {
-		private final int/*long*/ offset; // for separate .i and .d - copy of index record entry, for inline index - actual offset of the record in the .i file (record entry + revision * record size))
-		//public final int length; // data past fixed record (need to decide whether including header size or not), and whether length is of compressed data or not
-		public final int baseRevision;
-
-		public IndexEntry(long o, int baseRev) {
-			offset = (int) o;
-			// Index file stores offsets as long, but since DataAccess doesn't support long length() and others yet, 
-			// no reason to operate with long offsets 
-			if (o != offset) {
-				throw new HgBadStateException("Data too big, offset didn't fit to sizeof(int)");
-			}
-			baseRevision = baseRev;
+	private static int[] toArray(List<Integer> l) {
+		int[] rv = new int[l.size()];
+		for (int i = 0; i < rv.length; i++) {
+			rv[i] = l.get(i);
 		}
-
-		public int getIntOffset() {
-			return offset;
-		}
+		return rv;
 	}
+	
 
 	// mpatch.c : apply()
 	// FIXME need to implement patch merge (fold, combine, gather and discard from aforementioned mpatch.[c|py]), also see Revlog and Mercurial PDF