plugins/svn4idea/src/org/jetbrains/idea/svn/svnkit/lowLevel/SvnIdeaRepositoryPoolManager.java

/*
 * Copyright 2000-2012 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jetbrains.idea.svn.svnkit.lowLevel;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.util.Pair;
import com.intellij.util.ThrowableConsumer;
import com.intellij.util.ThrowableConvertor;
import org.tmatesoft.svn.core.ISVNCanceller;
import org.tmatesoft.svn.core.SVNCancelException;
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.*;
import org.tmatesoft.svn.core.wc.ISVNRepositoryPool;
import org.tmatesoft.svn.util.ISVNDebugLog;
import org.tmatesoft.svn.util.SVNDebugLog;

/**
 * @author Irina.Chernushina
 * @since 30.07.2012
 */
public class SvnIdeaRepositoryPoolManager implements ISVNRepositoryPool, ISVNSession {
  private final static ApplicationLevelNumberConnectionsGuardImpl ourGuard;

  private final ISVNDebugLog myLog;
  private final QuicklyDisposableISVNCanceller myCanceller;

  private final SvnRepositoryPool myPool;
  private volatile boolean myKeepConnection;
  private final QuicklyDisposableISVNTunnelProvider myTunnelProvider;
  private final QuicklyDisposableISVNConnectionListener myListener;
  private QuicklyDisposableISVNAuthenticationManager myAuthManager;
  private ThrowableConvertor<SVNURL, SVNRepository, SVNException> myCreator;

  static {
    ourGuard = new ApplicationLevelNumberConnectionsGuardImpl();
  }

  public SvnIdeaRepositoryPoolManager(final boolean keepConnection, final ISVNAuthenticationManager authManager,
                                      final ISVNTunnelProvider tunnelProvider) {
    this(keepConnection, authManager, tunnelProvider, -1, -1);
  }

  public SvnIdeaRepositoryPoolManager(final boolean keepConnection,
                                      final ISVNAuthenticationManager authManager,
                                      final ISVNTunnelProvider tunnelProvider, final int maxCached, final int maxConcurrent) {
    myKeepConnection = keepConnection;
    myTunnelProvider = new QuicklyDisposableISVNTunnelProvider(tunnelProvider);
    myAuthManager = new QuicklyDisposableISVNAuthenticationManager(authManager);
    myLog = new ProxySvnLog(SVNDebugLog.getDefaultLog());
    myCanceller = new QuicklyDisposableISVNCanceller(new MyCanceller());

    final ThrowableConvertor<SVNURL, SVNRepository, SVNException> creator = new ThrowableConvertor<SVNURL, SVNRepository, SVNException>() {
      @Override
      public SVNRepository convert(SVNURL svnurl) throws SVNException {
        final SVNRepository repos = myCreator != null ? myCreator.convert(svnurl) : SVNRepositoryFactory.create(svnurl, SvnIdeaRepositoryPoolManager.this);
        repos.setAuthenticationManager(myAuthManager);
        repos.setTunnelProvider(myTunnelProvider);
        repos.setDebugLog(myLog);
        repos.setCanceller(myCanceller);
        if (myKeepConnection) {
          repos.addConnectionListener(myListener);
        }
        return repos;
      }
    };

    final ThrowableConsumer<Pair<SVNURL, SVNRepository>, SVNException> adjuster = new ThrowableConsumer<Pair<SVNURL, SVNRepository>, SVNException>() {
      @Override
      public void consume(Pair<SVNURL, SVNRepository> pair) throws SVNException {
        SVNRepository repository = pair.getSecond();
        SVNURL url = pair.getFirst();
        repository.setLocation(url, false);
        repository.addConnectionListener(myListener);
        repository.setAuthenticationManager(myAuthManager);
        repository.setTunnelProvider(myTunnelProvider);
        repository.setDebugLog(myLog);
        repository.setCanceller(myCanceller);
      }
    };

    if (keepConnection) {
      CachingSvnRepositoryPool pool = new CachingSvnRepositoryPool(creator, maxCached, maxConcurrent, adjuster, ourGuard);
      myPool = pool;
      ISVNConnectionListener listener = new ISVNConnectionListener() {
        @Override
        public void connectionOpened(SVNRepository repository) {
          ourGuard.connectionOpened();
        }

        @Override
        public void connectionClosed(SVNRepository repository) {
          repository.removeConnectionListener(myListener);
          myPool.returnRepo(repository);
          ourGuard.connectionClosed();
        }
      };
      myListener = new QuicklyDisposableISVNConnectionListener(listener);
      ourGuard.addRepositoryPool(pool);
    }
    else {
      myListener = null;
      myPool = new NoKeepConnectionPool(creator);
    }
  }

  private static class MyCanceller implements ISVNCanceller {
    @Override
    public void checkCancelled() throws SVNCancelException {
      final ProgressManager pm = ProgressManager.getInstance();
      final ProgressIndicator pi = pm.getProgressIndicator();
      if (pi != null) {
        if (pi.isCanceled()) throw new SVNCancelException();
      }
      ProgressIndicator indicator = ProgressManager.getInstance().getProgressIndicator();
      if (indicator != null && indicator.isCanceled()) {
        throw new SVNCancelException();
      }
    }
  }

  @Override
  public void setAuthenticationManager(ISVNAuthenticationManager authManager) {
    myAuthManager = new QuicklyDisposableISVNAuthenticationManager(authManager);
  }

  @Override
  public void setCanceller(ISVNCanceller canceller) {
    // we didn't prepare for such use
    throw new UnsupportedOperationException();
  }

  @Override
  public void setDebugLog(ISVNDebugLog log) {
    // we didn't prepare for such use
    throw new UnsupportedOperationException();
  }

  @Override
  public SVNRepository createRepository(SVNURL url, boolean mayReuse) throws SVNException {
    return myPool.getRepo(url, mayReuse);
  }

  @Override
  public void shutdownConnections(boolean shutdownAll) {
    dispose();
  }

  /**
   * Disposes this pool.
   *
   * @since 1.2.0
   */
  @Override
  public void dispose() {
    if (myKeepConnection) {
      ourGuard.removeRepositoryPool((CachingSvnRepositoryPool)myPool);
    }
    myKeepConnection = false;
    if (myListener != null) {
      myListener.dispose();
    }
    myAuthManager.dispose();
    myTunnelProvider.dispose();
    myPool.dispose();
  }

  @Override
  public boolean keepConnection(SVNRepository repository) {
    return myKeepConnection;
  }

  @Override
  public void saveCommitMessage(SVNRepository repository, long revision, String message) {
  }

  @Override
  public String getCommitMessage(SVNRepository repository, long revision) {
    return null;
  }

  @Override
  public boolean hasCommitMessage(SVNRepository repository, long revision) {
    return false;
  }

  public void setCreator(ThrowableConvertor<SVNURL, SVNRepository, SVNException> creator) {
    assert ApplicationManager.getApplication().isUnitTestMode();
    myCreator = creator;
  }

  public SvnRepositoryPool getPool() {
    assert ApplicationManager.getApplication().isUnitTestMode();
    return myPool;
  }

  public static ApplicationLevelNumberConnectionsGuardImpl getOurGuard() {
    assert ApplicationManager.getApplication().isUnitTestMode();
    return ourGuard;
  }
}