1 17 package org.apache.geronimo.console.databasemanager.wizard; 18 19 import java.io.BufferedOutputStream ; 20 import java.io.File ; 21 import java.io.FileOutputStream ; 22 import java.io.IOException ; 23 import java.io.InputStream ; 24 import java.io.OutputStream ; 25 import java.io.Serializable ; 26 import java.net.MalformedURLException ; 27 import java.net.URL ; 28 import java.net.URLConnection ; 29 import java.util.ArrayList ; 30 import java.util.Collections ; 31 import java.util.HashSet ; 32 import java.util.Iterator ; 33 import java.util.List ; 34 import java.util.Properties ; 35 import java.util.Random ; 36 import java.util.Set ; 37 import java.util.jar.JarEntry ; 38 import java.util.jar.JarFile ; 39 import org.apache.commons.logging.Log; 40 import org.apache.commons.logging.LogFactory; 41 import org.apache.geronimo.kernel.repository.Artifact; 42 import org.apache.geronimo.kernel.repository.FileWriteMonitor; 43 import org.apache.geronimo.kernel.repository.WriteableRepository; 44 45 51 public class DriverDownloader { 52 private final static Log log = LogFactory.getLog(DriverDownloader.class); 53 Random random; 54 55 public Properties readDriverFile(URL url) { 56 try { 57 InputStream in = url.openStream(); 58 Properties props = new Properties (); 59 props.load(in); 60 in.close(); 61 return props; 62 } catch (IOException e) { 63 log.error("Unable to download driver properties", e); 64 return null; 65 } 66 } 67 68 public DriverInfo[] loadDriverInfo(URL driverInfoFile) { 69 List list = new ArrayList (); 70 Properties props = readDriverFile(driverInfoFile); 71 if(props == null) { 72 return new DriverInfo[0]; 73 } 74 Set drivers = new HashSet (); 75 for (Iterator it = props.keySet().iterator(); it.hasNext();) { 76 String key = (String ) it.next(); 77 if(!key.startsWith("driver.")) { 78 continue; 79 } 80 int pos = key.indexOf('.', 7); 81 if(pos > -1) { 82 drivers.add(key.substring(7, pos)); 83 } 84 } 85 List urls = new ArrayList (); 86 for (Iterator it = drivers.iterator(); it.hasNext();) { 87 String driver = (String ) it.next(); 88 String name = props.getProperty("driver."+driver+".name"); 89 String repository = props.getProperty("driver."+driver+".repository"); 90 String unzip = props.getProperty("driver."+driver+".unzip"); 91 urls.clear(); 92 int index = 1; 93 while(true) { 94 String url = props.getProperty("driver."+driver+".url."+index); 95 if(url != null) { 96 ++index; 97 try { 98 urls.add(new URL (url)); 99 } catch (MalformedURLException e) { 100 log.error("Unable to process URL from driver list", e); 101 } 102 } else { 103 break; 104 } 105 } 106 if(name != null && repository != null && urls.size() > 0) { 107 DriverInfo info = new DriverInfo(name, repository); 108 info.setUnzipPath(unzip); 109 info.setUrls((URL []) urls.toArray(new URL [urls.size()])); 110 list.add(info); 111 } 112 } 113 Collections.sort(list); 114 return (DriverInfo[]) list.toArray(new DriverInfo[list.size()]); 115 } 116 117 120 public void loadDriver(WriteableRepository repo, DriverInfo driver, FileWriteMonitor monitor) throws IOException { 121 int urlIndex = 0; 122 if (driver.urls.length > 1) { 123 if (random == null) { 124 random = new Random (); 125 } 126 urlIndex = random.nextInt(driver.urls.length); 127 } 128 URL url = driver.urls[urlIndex]; 129 InputStream in; 130 String uri = driver.getRepositoryURI(); 131 if (driver.unzipPath != null) { 132 byte[] buf = new byte[1024]; 133 int size; 134 int total = 0; 135 int threshold = 10240; 136 URLConnection uc = url.openConnection(); 137 int filesize = uc.getContentLength(); 138 InputStream net = uc.getInputStream(); 139 JarFile jar = null; 140 File download = null; 141 try { 142 download = File.createTempFile("geronimo-driver-download", ".zip"); 143 OutputStream out = new BufferedOutputStream (new FileOutputStream (download)); 144 if (monitor != null) { 145 monitor.writeStarted("Download driver archive to " + download, filesize); 146 } 147 try { 148 while ((size = net.read(buf)) > -1) { 149 out.write(buf, 0, size); 150 if (monitor != null) { 151 total += size; 152 if (total > threshold) { 153 monitor.writeProgress(total); 154 threshold += 10240; 155 } 156 } 157 } 158 out.flush(); 159 out.close(); 160 } finally { 161 if (monitor != null) { 162 monitor.writeComplete(total); 163 } 164 } 165 jar = new JarFile (download); 166 JarEntry entry = jar.getJarEntry(driver.unzipPath); 167 if (entry == null) { 168 log.error("Cannot extract driver JAR " + driver.unzipPath + " from download file " + url); 169 } else { 170 in = jar.getInputStream(entry); 171 repo.copyToRepository(in, (int)entry.getSize(), Artifact.create(uri), monitor); 172 } 173 } finally { 174 if (jar != null) try { 175 jar.close(); 176 } catch (IOException e) { 177 log.error("Unable to close JAR file", e); 178 } 179 if (download != null) { 180 download.delete(); 181 } 182 } 183 } else { 184 URLConnection con = url.openConnection(); 185 in = con.getInputStream(); 186 repo.copyToRepository(in, con.getContentLength(), Artifact.create(uri), monitor); 187 } 188 } 189 190 public static class DriverInfo implements Comparable , Serializable { 191 private final static long serialVersionUID = -1202452382992975449L; 192 193 private String name; 194 private String repositoryURI; 195 private String unzipPath; 196 private URL [] urls; 197 198 public DriverInfo(String name, String repositoryURI) { 199 this.name = name; 200 this.repositoryURI = repositoryURI; 201 } 202 203 public String getName() { 204 return name; 205 } 206 207 public void setName(String name) { 208 this.name = name; 209 } 210 211 public String getRepositoryURI() { 212 return repositoryURI; 213 } 214 215 public void setRepositoryURI(String repositoryURI) { 216 this.repositoryURI = repositoryURI; 217 } 218 219 public String getUnzipPath() { 220 return unzipPath; 221 } 222 223 public void setUnzipPath(String unzipPath) { 224 this.unzipPath = unzipPath; 225 } 226 227 public URL [] getUrls() { 228 return urls; 229 } 230 231 public void setUrls(URL [] urls) { 232 this.urls = urls; 233 } 234 235 public int compareTo(Object o) { 236 return name.compareTo(((DriverInfo)o).name); 237 } 238 } 239 } 240 | Popular Tags |