From: Mathieu Baudier <mbaudier@argeo.org>
Date: Tue, 17 Sep 2013 08:47:14 +0000 (+0000)
Subject: Introduce mirrors
X-Git-Tag: argeo-slc-2.1.7~313
X-Git-Url: http://git.argeo.org/?a=commitdiff_plain;ds=sidebyside;h=ae7dd82088345d6f6f6825e194fd9c312035cc4d;p=gpl%2Fargeo-slc.git

Introduce mirrors

git-svn-id: https://svn.argeo.org/slc/trunk@6487 4cfe0d0a-d680-48aa-b62c-e0a02a3f76cc
---

diff --git a/runtime/org.argeo.slc.repo/src/main/java/org/argeo/slc/repo/osgi/OsgiFactoryImpl.java b/runtime/org.argeo.slc.repo/src/main/java/org/argeo/slc/repo/osgi/OsgiFactoryImpl.java
index 3381eb892..ab4f9775c 100644
--- a/runtime/org.argeo.slc.repo/src/main/java/org/argeo/slc/repo/osgi/OsgiFactoryImpl.java
+++ b/runtime/org.argeo.slc.repo/src/main/java/org/argeo/slc/repo/osgi/OsgiFactoryImpl.java
@@ -1,11 +1,14 @@
 package org.argeo.slc.repo.osgi;
 
+import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import javax.jcr.Node;
 import javax.jcr.Repository;
@@ -22,6 +25,7 @@ import org.argeo.slc.jcr.SlcTypes;
 import org.argeo.slc.repo.NodeIndexer;
 import org.argeo.slc.repo.OsgiFactory;
 
+/** Default implementation of {@link OsgiFactory}. */
 public class OsgiFactoryImpl implements OsgiFactory {
 	private final static Log log = LogFactory.getLog(OsgiFactoryImpl.class);
 
@@ -31,6 +35,9 @@ public class OsgiFactoryImpl implements OsgiFactory {
 
 	private List<NodeIndexer> nodeIndexers = new ArrayList<NodeIndexer>();
 
+	/** key is URI prefix, value list of base URLs */
+	private Map<String, List<String>> mirrors = new HashMap<String, List<String>>();
+
 	public void init() {
 		if (workspace == null)
 			throw new SlcException("A workspace must be specified");
@@ -77,19 +84,49 @@ public class OsgiFactoryImpl implements OsgiFactory {
 
 	public Node getDist(Session distSession, String uri)
 			throws RepositoryException {
-
 		String distPath = '/' + JcrUtils.urlAsPath(uri);
 
-		// TODO manage mirrors
+		// already retrieved
+		if (distSession.itemExists(distPath))
+			return distSession.getNode(distPath);
+
+		// find mirror
+		List<String> urlBases = null;
+		String uriPrefix = null;
+		uriPrefixes: for (String uriPref : mirrors.keySet()) {
+			if (uri.startsWith(uriPref)) {
+				if (mirrors.get(uriPref).size() > 0) {
+					urlBases = mirrors.get(uriPref);
+					uriPrefix = uriPref;
+					break uriPrefixes;
+				}
+			}
+		}
+		if (urlBases == null)
+			try {
+				return loadUrlToPath(uri, distSession, distPath);
+			} catch (FileNotFoundException e) {
+				throw new SlcException("Cannot download " + uri, e);
+			}
+
+		// try to download
+		for (String urlBase : urlBases) {
+			String relativePath = uri.substring(uriPrefix.length());
+			String url = urlBase + relativePath;
+			try {
+				return loadUrlToPath(url, distSession, distPath);
+			} catch (FileNotFoundException e) {
+				if (log.isDebugEnabled())
+					log.debug("Cannot download" + url
+							+ ", trying another mirror");
+			}
+		}
 
-		if (!distSession.itemExists(distPath))
-			loadUrlToPath(uri, distSession, distPath);
-		Node distNode = distSession.getNode(distPath);
-		return distNode;
+		throw new SlcException("Could not download " + uri);
 	}
 
-	protected void loadUrlToPath(String url, Session distSession, String path)
-			throws RepositoryException {
+	protected Node loadUrlToPath(String url, Session distSession, String path)
+			throws RepositoryException, FileNotFoundException {
 		if (log.isDebugEnabled())
 			log.debug("Downloading " + url + "...");
 
@@ -106,9 +143,11 @@ public class OsgiFactoryImpl implements OsgiFactory {
 					SlcTypes.SLC_PROXIED);
 			JcrUtils.urlToAddressProperties(origin, url);
 			distSession.save();
-
+			return fileNode;
 		} catch (MalformedURLException e) {
 			throw new SlcException("URL " + url + " not valid.", e);
+		} catch (FileNotFoundException e) {
+			throw e;
 		} catch (IOException e) {
 			throw new SlcException("Cannot load " + url + " to " + path, e);
 		}
@@ -131,4 +170,8 @@ public class OsgiFactoryImpl implements OsgiFactory {
 		this.nodeIndexers = nodeIndexers;
 	}
 
+	public void setMirrors(Map<String, List<String>> mirrors) {
+		this.mirrors = mirrors;
+	}
+
 }