cn.novelweb.tool.upload.fastdfs.pool.PooledConnectionFactory Maven / Gradle / Ivy

Go to download
package cn.novelweb.tool.upload.fastdfs.pool;

import cn.novelweb.tool.upload.fastdfs.conn.Connection;
import cn.novelweb.tool.upload.fastdfs.conn.SocketConnection;
import cn.novelweb.tool.upload.fastdfs.utils.Log;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.pool2.BaseKeyedPooledObjectFactory;
import org.apache.commons.pool2.PooledObject;
import org.apache.commons.pool2.impl.DefaultPooledObject;

import java.net.InetSocketAddress;
import java.nio.charset.Charset;

/**
 * 定义了被池化的对象的创建，初始化，激活，钝化以及销毁功能

 * 2020-02-03 17:25
 *
 * @author Dai Yuanchuan
 **/
@Slf4j
public class PooledConnectionFactory extends BaseKeyedPooledObjectFactory {

    /**
     * 默认字符集
     */
    private static final String DEFAULT_CHARSET_NAME = "UTF-8";

    /**
     * 设置默认字符集
     */
    private String charsetName = DEFAULT_CHARSET_NAME;

    /**
     * 读取时间
     */
    private int soTimeout;

    /**
     * 连接超时时间
     */
    private int connectTimeout;

    /**
     * 字符集
     */
    private Charset charset;

    public PooledConnectionFactory(int soTimeout, int connectTimeout, Charset charset) {
        this.soTimeout = soTimeout;
        this.connectTimeout = connectTimeout;
        this.charset = charset;
    }

    public PooledConnectionFactory(int soTimeout, int connectTimeout) {
        this(soTimeout, connectTimeout, null);
    }

    /**
     * 创建连接
     */
    @Override
    public Connection create(InetSocketAddress address) throws Exception {
        // 初始化字符集
        if (null == charset) {
            charset = Charset.forName(charsetName);
        }
        SocketConnection connection = new SocketConnection(address, soTimeout, connectTimeout, charset);
        Log.debug("新建连接[{}]", address);
        return connection;
    }

    /**
     * 将对象池化pooledObject
     */
    @Override
    public PooledObject wrap(Connection conn) {
        return new DefaultPooledObject(conn);
    }

    @Override
    public void destroyObject(InetSocketAddress key, PooledObject p) throws Exception {
        p.getObject().close();
        Log.debug("关闭连接[{}]", key);
    }

    /**
     * 验证连接是否可用
     *
     * @param key 连接对应的Key
     * @param p   池化的对象
     */
    @Override
    public boolean validateObject(InetSocketAddress key, PooledObject p) {
        boolean flag = p.getObject().isValid();
        Log.debug("验证连接是否可用[{}],验证结果[{}]", key, flag);
        return flag;
    }

    public int getSoTimeout() {
        return soTimeout;
    }

    public void setSoTimeout(int soTimeout) {
        this.soTimeout = soTimeout;
    }

    public int getConnectTimeout() {
        return connectTimeout;
    }

    public void setConnectTimeout(int connectTimeout) {
        this.connectTimeout = connectTimeout;
    }

    public Charset getCharset() {
        return charset;
    }

    public void setCharsetName(String charsetName) {
        this.charsetName = charsetName;
    }
}