From a75a1819f4c7fa5080ddb47545fe9012a842e5b3 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Wed, 20 Jul 2011 21:08:57 -0400
Subject: [PATCH] Misc fixes.

---
 src/com/gitblit/utils/MetricUtils.java |  164 ++++++++++++++++++++++++++++++------------------------
 1 files changed, 90 insertions(+), 74 deletions(-)

diff --git a/src/com/gitblit/utils/MetricUtils.java b/src/com/gitblit/utils/MetricUtils.java
index b1da273..acdacc4 100644
--- a/src/com/gitblit/utils/MetricUtils.java
+++ b/src/com/gitblit/utils/MetricUtils.java
@@ -35,81 +35,74 @@
 import com.gitblit.models.Metric;
 import com.gitblit.models.RefModel;
 
+/**
+ * Utility class for collecting metrics on a branch, tag, or other ref within
+ * the repository.
+ * 
+ * @author James Moger
+ * 
+ */
 public class MetricUtils {
 
 	private static final Logger LOGGER = LoggerFactory.getLogger(MetricUtils.class);
 
-	public static List<Metric> getDateMetrics(Repository r, boolean includeTotal, String format) {
+	/**
+	 * Returns the list of metrics for the specified commit reference, branch,
+	 * or tag within the repository. If includeTotal is true, the total of all
+	 * the metrics will be included as the first element in the returned list.
+	 * 
+	 * If the dateformat is unspecified an attempt is made to determine an
+	 * appropriate date format by determining the time difference between the
+	 * first commit on the branch and the most recent commit. This assumes that
+	 * the commits are linear.
+	 * 
+	 * @param repository
+	 * @param objectId
+	 *            if null or empty, HEAD is assumed.
+	 * @param includeTotal
+	 * @param dateFormat
+	 * @return list of metrics
+	 */
+	public static List<Metric> getDateMetrics(Repository repository, String objectId,
+			boolean includeTotal, String dateFormat) {
 		Metric total = new Metric("TOTAL");
 		final Map<String, Metric> metricMap = new HashMap<String, Metric>();
-	
-		if (JGitUtils.hasCommits(r)) {			
-			try {
-				RevWalk walk = new RevWalk(r);
-				ObjectId object = r.resolve(Constants.HEAD);
-				RevCommit lastCommit = walk.parseCommit(object);
-				walk.markStart(lastCommit);
-				SimpleDateFormat df = new SimpleDateFormat(format);
-				Iterable<RevCommit> revlog = walk;
-				for (RevCommit rev : revlog) {
-					Date d = JGitUtils.getCommitDate(rev);
-					String p = df.format(d);
-					if (!metricMap.containsKey(p)) {
-						metricMap.put(p, new Metric(p));
-					}
-					Metric m = metricMap.get(p);
-					m.count++;
-					total.count++;					
-				}
-			} catch (Throwable t) {
-				JGitUtils.LOGGER.error("Failed to mine log history for metrics", t);
-			}
+		if (StringUtils.isEmpty(objectId)) {
+			objectId = Constants.HEAD;
 		}
-		List<String> keys = new ArrayList<String>(metricMap.keySet());
-		Collections.sort(keys);
-		List<Metric> metrics = new ArrayList<Metric>();
-		for (String key : keys) {
-			metrics.add(metricMap.get(key));
-		}
-		if (includeTotal) {
-			metrics.add(0, total);
-		}
-		return metrics;
-	}
-
-	public static List<Metric> getDateMetrics(Repository r, boolean includeTotal) {
-		Metric total = new Metric("TOTAL");
-		final Map<String, Metric> metricMap = new HashMap<String, Metric>();
-	
-		if (JGitUtils.hasCommits(r)) {
-			final List<RefModel> tags = JGitUtils.getTags(r, -1);
+		if (JGitUtils.hasCommits(repository)) {
+			final List<RefModel> tags = JGitUtils.getTags(repository, true, -1);
 			final Map<ObjectId, RefModel> tagMap = new HashMap<ObjectId, RefModel>();
 			for (RefModel tag : tags) {
-				tagMap.put(tag.getCommitId(), tag);
+				tagMap.put(tag.getReferencedObjectId(), tag);
 			}
+			RevWalk revWalk = null;
 			try {
-				RevWalk walk = new RevWalk(r);
-				ObjectId object = r.resolve(Constants.HEAD);
-	
-				RevCommit firstCommit = JGitUtils.getFirstCommit(r, Constants.HEAD);
-				RevCommit lastCommit = walk.parseCommit(object);
-				int diffDays = (lastCommit.getCommitTime() - firstCommit.getCommitTime())
-						/ (60 * 60 * 24);
-				total.duration = diffDays;
+				revWalk = new RevWalk(repository);
+				ObjectId object = repository.resolve(objectId);
+				RevCommit lastCommit = revWalk.parseCommit(object);
+				revWalk.markStart(lastCommit);
+
 				DateFormat df;
-				if (diffDays <= 90) {
-					// Days
-					df = new SimpleDateFormat("yyyy-MM-dd");
-				} else if (diffDays > 90 && diffDays < 365) {
-					// Weeks
-					df = new SimpleDateFormat("yyyy-MM (w)");
+				if (StringUtils.isEmpty(dateFormat)) {
+					// dynamically determine date format
+					RevCommit firstCommit = JGitUtils.getFirstCommit(repository, Constants.HEAD);
+					int diffDays = (lastCommit.getCommitTime() - firstCommit.getCommitTime())
+							/ (60 * 60 * 24);
+					total.duration = diffDays;
+					if (diffDays <= 365) {
+						// Days
+						df = new SimpleDateFormat("yyyy-MM-dd");
+					} else {
+						// Months
+						df = new SimpleDateFormat("yyyy-MM");
+					}
 				} else {
-					// Months
-					df = new SimpleDateFormat("yyyy-MM");
+					// use specified date format
+					df = new SimpleDateFormat(dateFormat);
 				}
-				walk.markStart(lastCommit);
-	
-				Iterable<RevCommit> revlog = walk;
+
+				Iterable<RevCommit> revlog = revWalk;
 				for (RevCommit rev : revlog) {
 					Date d = JGitUtils.getCommitDate(rev);
 					String p = df.format(d);
@@ -125,7 +118,11 @@
 					}
 				}
 			} catch (Throwable t) {
-				JGitUtils.LOGGER.error("Failed to mine log history for metrics", t);
+				LOGGER.error("Failed to mine log history for date metrics", t);
+			} finally {
+				if (revWalk != null) {
+					revWalk.dispose();
+				}
 			}
 		}
 		List<String> keys = new ArrayList<String>(metricMap.keySet());
@@ -140,32 +137,51 @@
 		return metrics;
 	}
 
-	public static List<Metric> getAuthorMetrics(Repository r) {
-		Metric total = new Metric("TOTAL");
+	/**
+	 * Returns a list of author metrics for the specified repository.
+	 * 
+	 * @param repository
+	 * @param objectId
+	 *            if null or empty, HEAD is assumed.
+	 * @param byEmailAddress
+	 *            group metrics by author email address otherwise by author name
+	 * @return list of metrics
+	 */
+	public static List<Metric> getAuthorMetrics(Repository repository, String objectId,
+			boolean byEmailAddress) {
 		final Map<String, Metric> metricMap = new HashMap<String, Metric>();
-	
-		if (JGitUtils.hasCommits(r)) {
+		if (StringUtils.isEmpty(objectId)) {
+			objectId = Constants.HEAD;
+		}
+		if (JGitUtils.hasCommits(repository)) {
 			try {
-				RevWalk walk = new RevWalk(r);
-				ObjectId object = r.resolve(Constants.HEAD);
+				RevWalk walk = new RevWalk(repository);
+				ObjectId object = repository.resolve(objectId);
 				RevCommit lastCommit = walk.parseCommit(object);
 				walk.markStart(lastCommit);
-	
+
 				Iterable<RevCommit> revlog = walk;
 				for (RevCommit rev : revlog) {
-					String p = rev.getAuthorIdent().getName();
-					if (StringUtils.isEmpty(p)) {
-						p = rev.getAuthorIdent().getEmailAddress();
+					String p;
+					if (byEmailAddress) {
+						p = rev.getAuthorIdent().getEmailAddress().toLowerCase();
+						if (StringUtils.isEmpty(p)) {
+							p = rev.getAuthorIdent().getName().toLowerCase();
+						}
+					} else {
+						p = rev.getAuthorIdent().getName().toLowerCase();
+						if (StringUtils.isEmpty(p)) {
+							p = rev.getAuthorIdent().getEmailAddress().toLowerCase();
+						}
 					}
 					if (!metricMap.containsKey(p)) {
 						metricMap.put(p, new Metric(p));
 					}
 					Metric m = metricMap.get(p);
 					m.count++;
-					total.count++;
 				}
 			} catch (Throwable t) {
-				JGitUtils.LOGGER.error("Failed to mine log history for metrics", t);
+				LOGGER.error("Failed to mine log history for author metrics", t);
 			}
 		}
 		List<String> keys = new ArrayList<String>(metricMap.keySet());

--
Gitblit v1.9.1