Java tutorial
/* * Copyright (C) 2008-2018, Robin Rosenberg <robin.rosenberg@dewire.com> * Copyright (C) 2008, Shawn O. Pearce <spearce@spearce.org> * and other copyright owners as documented in the project's IP log. * * This program and the accompanying materials are made available * under the terms of the Eclipse Distribution License v1.0 which * accompanies this distribution, is reproduced below, and is * available at http://www.eclipse.org/org/documents/edl-v10.php * * All rights reserved. * * Redistribution and use in source and binary forms, with or * without modification, are permitted provided that the following * conditions are met: * * - Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * - Redistributions in binary form must reproduce the above * copyright notice, this list of conditions and the following * disclaimer in the documentation and/or other materials provided * with the distribution. * * - Neither the name of the Eclipse Foundation, Inc. nor the * names of its contributors may be used to endorse or promote * products derived from this software without specific prior * written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ package org.eclipse.jgit.revplot; import static org.eclipse.jgit.lib.Constants.R_HEADS; import static org.eclipse.jgit.lib.Constants.R_REMOTES; import static org.eclipse.jgit.lib.Constants.R_TAGS; import java.io.IOException; import java.util.Arrays; import java.util.Collection; import java.util.Collections; import java.util.Comparator; import java.util.HashMap; import java.util.HashSet; import java.util.Map; import java.util.Set; import org.eclipse.jgit.errors.IncorrectObjectTypeException; import org.eclipse.jgit.errors.MissingObjectException; import org.eclipse.jgit.internal.JGitText; import org.eclipse.jgit.lib.AnyObjectId; import org.eclipse.jgit.lib.PersonIdent; import org.eclipse.jgit.lib.Ref; import org.eclipse.jgit.lib.Repository; import org.eclipse.jgit.revwalk.RevCommit; import org.eclipse.jgit.revwalk.RevObject; import org.eclipse.jgit.revwalk.RevSort; import org.eclipse.jgit.revwalk.RevTag; import org.eclipse.jgit.revwalk.RevWalk; /** * Specialized RevWalk for visualization of a commit graph. */ public class PlotWalk extends RevWalk { private Map<AnyObjectId, Set<Ref>> additionalRefMap; private Map<AnyObjectId, Set<Ref>> reverseRefMap; private Repository repository; /** {@inheritDoc} */ @Override public void dispose() { super.dispose(); if (reverseRefMap != null) { reverseRefMap.clear(); reverseRefMap = null; } if (additionalRefMap != null) { additionalRefMap.clear(); additionalRefMap = null; } repository = null; } /** * Create a new revision walker for a given repository. * * @param repo * the repository the walker will obtain data from. */ public PlotWalk(Repository repo) { super(repo); super.sort(RevSort.TOPO, true); additionalRefMap = new HashMap<>(); repository = repo; } /** * Add additional refs to the walk * * @param refs * additional refs * @throws java.io.IOException */ public void addAdditionalRefs(Iterable<Ref> refs) throws IOException { for (Ref ref : refs) { Set<Ref> set = additionalRefMap.get(ref.getObjectId()); if (set == null) set = Collections.singleton(ref); else { set = new HashSet<>(set); set.add(ref); } additionalRefMap.put(ref.getObjectId(), set); } } /** {@inheritDoc} */ @Override public void sort(RevSort s, boolean use) { if (s == RevSort.TOPO && !use) throw new IllegalArgumentException(JGitText.get().topologicalSortRequired); super.sort(s, use); } /** {@inheritDoc} */ @Override protected RevCommit createCommit(AnyObjectId id) { return new PlotCommit(id); } /** {@inheritDoc} */ @Override public RevCommit next() throws MissingObjectException, IncorrectObjectTypeException, IOException { PlotCommit<?> pc = (PlotCommit) super.next(); if (pc != null) pc.refs = getRefs(pc); return pc; } private Ref[] getRefs(AnyObjectId commitId) { if (reverseRefMap == null) { reverseRefMap = repository.getAllRefsByPeeledObjectId(); for (Map.Entry<AnyObjectId, Set<Ref>> entry : additionalRefMap.entrySet()) { Set<Ref> set = reverseRefMap.get(entry.getKey()); Set<Ref> additional = entry.getValue(); if (set != null) { if (additional.size() == 1) { // It's an unmodifiable singleton set... additional = new HashSet<>(additional); } additional.addAll(set); } reverseRefMap.put(entry.getKey(), additional); } additionalRefMap.clear(); additionalRefMap = null; } Collection<Ref> list = reverseRefMap.get(commitId); if (list == null) { return PlotCommit.NO_REFS; } Ref[] tags = list.toArray(new Ref[0]); Arrays.sort(tags, new PlotRefComparator()); return tags; } class PlotRefComparator implements Comparator<Ref> { @Override public int compare(Ref o1, Ref o2) { try { RevObject obj1 = parseAny(o1.getObjectId()); RevObject obj2 = parseAny(o2.getObjectId()); long t1 = timeof(obj1); long t2 = timeof(obj2); if (t1 > t2) return -1; if (t1 < t2) return 1; } catch (IOException e) { // ignore } int cmp = kind(o1) - kind(o2); if (cmp == 0) cmp = o1.getName().compareTo(o2.getName()); return cmp; } long timeof(RevObject o) { if (o instanceof RevCommit) return ((RevCommit) o).getCommitTime(); if (o instanceof RevTag) { RevTag tag = (RevTag) o; try { parseBody(tag); } catch (IOException e) { return 0; } PersonIdent who = tag.getTaggerIdent(); return who != null ? who.getWhen().getTime() : 0; } return 0; } int kind(Ref r) { if (r.getName().startsWith(R_TAGS)) return 0; if (r.getName().startsWith(R_HEADS)) return 1; if (r.getName().startsWith(R_REMOTES)) return 2; return 3; } } }