Proof of concept #1026

Paul Martin

2016-04-16 eecaad8b8e2c447429c31a01d49260ddd6b4ee03

 src/main/java/com/gitblit/servlet/BranchGraphServlet.java

@@ -36,11 +36,14 @@
import java.util.TreeSet;



import javax.imageio.ImageIO;

import com.google.inject.Inject;

import com.google.inject.Singleton;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;



import org.eclipse.jgit.lib.ObjectId;

import org.eclipse.jgit.lib.Ref;

import org.eclipse.jgit.lib.Repository;

import org.eclipse.jgit.revplot.AbstractPlotRenderer;

@@ -49,12 +52,13 @@
import org.eclipse.jgit.revplot.PlotLane;

import org.eclipse.jgit.revplot.PlotWalk;

import org.eclipse.jgit.revwalk.RevCommit;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;



import com.gitblit.Constants;

import com.gitblit.IStoredSettings;

import com.gitblit.Keys;

import com.gitblit.manager.IRepositoryManager;

import com.gitblit.manager.IRuntimeManager;

import com.gitblit.utils.JGitUtils;

import com.gitblit.utils.StringUtils;



@@ -64,6 +68,7 @@
 * @author James Moger

 *

 */

@Singleton

public class BranchGraphServlet extends HttpServlet {



   private static final long serialVersionUID = 1L;

@@ -75,23 +80,26 @@


   private static final int RIGHT_PAD = 2;



   private final Logger log = LoggerFactory.getLogger(getClass());



   private final Stroke[] strokeCache;



   private final IStoredSettings settings;

   private IStoredSettings settings;



   private final IRepositoryManager repositoryManager;

   private IRepositoryManager repositoryManager;



   @Inject

   public BranchGraphServlet(

         IRuntimeManager runtimeManager,

         IStoredSettings settings,

         IRepositoryManager repositoryManager) {



      super();

      this.settings = runtimeManager.getSettings();

      this.settings = settings;

      this.repositoryManager = repositoryManager;



      strokeCache = new Stroke[4];

      for (int i = 1; i < strokeCache.length; i++)

      for (int i = 1; i < strokeCache.length; i++) {

         strokeCache[i] = new BasicStroke(i);

      }

   }



   /**

@@ -115,6 +123,9 @@
   @Override

   protected long getLastModified(HttpServletRequest req) {

      String repository = req.getParameter("r");

      if (StringUtils.isEmpty(repository)) {

         return 0;

      }

      String objectId = req.getParameter("h");

      Repository r = null;

      try {

@@ -122,8 +133,15 @@
         if (StringUtils.isEmpty(objectId)) {

            objectId = JGitUtils.getHEADRef(r);

         }

         ObjectId id = r.resolve(objectId);

         if (id == null) {

            return 0;

         }

         RevCommit commit = JGitUtils.getCommit(r, objectId);

         return JGitUtils.getCommitDate(commit).getTime();

      } catch (Exception e) {

         log.error("Failed to determine last modified", e);

         return 0;

      } finally {

         if (r != null) {

            r.close();

@@ -139,17 +157,33 @@
      PlotWalk rw = null;

      try {

         String repository = request.getParameter("r");

         if (StringUtils.isEmpty(repository)) {

            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);

            response.getWriter().append("Bad request");

            return;

         }

         String objectId = request.getParameter("h");

         String length = request.getParameter("l");



         r = repositoryManager.getRepository(repository);

         if (r == null) {

            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);

            response.getWriter().append("Bad request");

            return;

         }



         rw = new PlotWalk(r);

         if (StringUtils.isEmpty(objectId)) {

            objectId = JGitUtils.getHEADRef(r);

         }



         rw.markStart(rw.lookupCommit(r.resolve(objectId)));

         ObjectId id = r.resolve(objectId);

         if (id ==  null) {

            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);

            response.getWriter().append("Bad request");

            return;

         }

         rw.markStart(rw.lookupCommit(id));



         // default to the items-per-page setting, unless specified

         int maxCommits = settings.getInteger(Keys.web.itemsPerPage, 50);