svnrepositorypool.java

来自「web版的SVN客户端」· Java 代码 · 共 151 行

/*
 * Copyright (c) 2004, 2005 Polarion Software, All rights reserved.
 * Email: [e-mail]
 *
 * This program and the accompanying materials are made available under the
 * terms of the Apache License, Version 2.0 (the "License"). You may not use
 * this file except in compliance with the License. Copy of the License is
 * located in the file LICENSE.txt in the project distribution. You may also
 * obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
 *
 *
 * POLARION SOFTWARE MAKES NO REPRESENTATIONS OR WARRANTIES
 * ABOUT THE SUITABILITY OF THE SOFTWARE, EITHER EXPRESSED OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO THE IMPLIED WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT. POLARION SOFTWARE
 * SHALL NOT BE LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT
 * OF USING, MODIFYING OR DISTRIBUTING THIS SOFTWARE OR ITS DERIVATIVES.
 */
package org.polarion.svnwebclient.data.javasvn;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.polarion.svncommons.commentscache.CommentsCache;
import org.polarion.svncommons.commentscache.CommentsCacheException;
import org.polarion.svncommons.commentscache.SVNAuthenticationManager;
import org.polarion.svnwebclient.authorization.UserCredentials;
import org.polarion.svnwebclient.data.DataProviderException;
import org.tmatesoft.svn.core.SVNException;
import org.tmatesoft.svn.core.SVNURL;
import org.tmatesoft.svn.core.auth.ISVNAuthenticationManager;
import org.tmatesoft.svn.core.io.SVNRepository;
import org.tmatesoft.svn.core.io.SVNRepositoryFactory;

/**
 * 
 * @author <A HREF="mailto:[e-mail]">Polarion Software </A>
 */
public class SVNRepositoryPool {
    private static  Map instances = Collections.synchronizedMap(new HashMap()); 
    
    protected  long size;
    protected  String url;
    protected  boolean initialized;
    protected String id;
    protected List repositories = new ArrayList();
    protected  int activeUsersCount;
    
    public static synchronized void init(String id, long size, String url, String userName, String password) {
    	SVNRepositoryPool instance = (SVNRepositoryPool) instances.get(id);
    	if (instance == null) {         
            SVNRepositoryPool pool = null;
            try {
                pool = new SVNRepositoryPool(id, size, url, userName, password);
                pool.activeUsersCount = 1;
                instances.put(id, pool);
            } catch (DataProviderException e) { 
                e.printStackTrace(); 
            }
        } else {
        	instance.activeUsersCount ++;
    	}
    }

    public static synchronized SVNRepositoryPool getInstance(String id) throws DataProviderException {
        SVNRepositoryPool pool = (SVNRepositoryPool) instances.get(id);
        if (pool == null) {
                throw new DataProviderException(SVNRepositoryPool.class.getName()
                        + " must be initialized before first usage");	
        }
        return pool;
    }

    private SVNRepositoryPool(String id, long size, String url, String userName, String password) throws DataProviderException {
        this.id = id;
        this.initialized = true;
        this.url = url;
        this.size = size;
                        
        try {
            for (int i = 0; i < size; i++) {
                SVNRepository repository = SVNRepositoryFactory.create(SVNURL.parseURIDecoded(url), 
                		CommentsCache.getInstance(id, userName, password));        
                this.repositories.add(repository);
            }
        } catch (SVNException e) {
            throw new DataProviderException(e);
        } catch (CommentsCacheException e) {
            throw new DataProviderException(e);
        }
    }
   
    public synchronized void releaseRepository(SVNRepository repository) {
        if (repository != null) {
            this.repositories.add(repository);
            this.notifyAll();
        }
    }

    public void shutdown() {
 	   this.activeUsersCount --;
 	   if (this.activeUsersCount == 0) {
 	        for (Iterator i = this.repositories.iterator(); i.hasNext();) {
 	            SVNRepository repository = (SVNRepository) i.next();
 	            try {
 	                repository.closeSession();
 	            } catch (SVNException e) {
 	            }
 	        }
 	   }	   
    } 
        
    public static void terminate() {
    	Iterator it = instances.entrySet().iterator();
    	while (it.hasNext()) { 
    		Map.Entry entry = (Map.Entry) it.next();
    		String id = (String)entry.getKey();
    		SVNRepositoryPool pool = (SVNRepositoryPool) entry.getValue();
    		for (Iterator i = pool.repositories.iterator(); i.hasNext();) {
              SVNRepository repository = (SVNRepository) i.next();
              try {
                  repository.closeSession();
              } catch (SVNException e) {
              }
              instances.remove(id);          
    		}
    	}
    }
    
    public String getRepositoryName() {        
        return this.url;
    }

	public synchronized SVNRepository getRepository(UserCredentials credentials) {
	    while (this.repositories.size() == 0) {
	        try {
	            this.wait();
	        } catch (InterruptedException e) {
	        }
	    }
	
	    SVNRepository ret = (SVNRepository) this.repositories.remove(0);
	    ISVNAuthenticationManager authManager = new SVNAuthenticationManager(credentials.getUsername(), credentials.getPassword());
	    ret.setAuthenticationManager(authManager);        
	    return ret;
	}
}