org.nlab.smtp.pool.SmtpConnectionPool Maven / Gradle / Ivy

Go to download
package org.nlab.smtp.pool;

import org.apache.commons.pool2.impl.AbandonedConfig;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.nlab.smtp.transport.connection.ClosableSmtpConnection;
import org.nlab.smtp.transport.factory.SmtpConnectionFactory;

import javax.mail.Session;

/**
 * Created by nlabrot on 30/04/15.
 */
public class SmtpConnectionPool extends GenericObjectPool {

  public SmtpConnectionPool(SmtpConnectionFactory factory) {
    super(factory);
  }

  public SmtpConnectionPool(SmtpConnectionFactory factory, GenericObjectPoolConfig config) {
    super(factory, config);
  }

  public SmtpConnectionPool(SmtpConnectionFactory factory, GenericObjectPoolConfig config, AbandonedConfig abandonedConfig) {
    super(factory, config, abandonedConfig);
  }

  @Override
  public ClosableSmtpConnection borrowObject() throws Exception {
    ClosableSmtpConnection object = super.borrowObject();
    if (object instanceof ObjectPoolAware) {
      ((ObjectPoolAware) object).setObjectPool(this);
    }
    return object;
  }

  @Override
  public ClosableSmtpConnection borrowObject(long borrowMaxWaitMillis) throws Exception {
    ClosableSmtpConnection object = super.borrowObject(borrowMaxWaitMillis);
    if (object instanceof ObjectPoolAware) {
      ((ObjectPoolAware) object).setObjectPool(this);
    }
    return object;
  }

  public Session getSession() {
    return ((SmtpConnectionFactory) getFactory()).getSession();
  }


}