com.github.yingzhuo.fastdfs.springboot.domain.conn.ConnectionFactory Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fastdfs-spring-boot-starter Show documentation

The newest version!

package com.github.yingzhuo.fastdfs.springboot.domain.conn;

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;

/**
 * 连接工厂
 *
 * @author tobato
 * @author 应卓
 */
public class ConnectionFactory extends BaseKeyedPooledObjectFactory {

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

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

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

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

    /**
     * 创建连接
     */
    @Override
    public Connection create(InetSocketAddress address) {
        return new ConnectionImpl(address, soTimeout, connectTimeout, charset);
    }

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

    /**
     * 从池中移出
     */
    @Override
    public void destroyObject(InetSocketAddress key, PooledObject p) throws Exception {
        p.getObject().close();
    }

    /**
     * 验证池中对象是否可用
     */
    @Override
    public boolean validateObject(InetSocketAddress key, PooledObject p) {
        return p.getObject().isValid();
    }

    // --------------------------------------------------------------------------------------------------------------

    private static class PooledConnection extends DefaultPooledObject {
        public PooledConnection(Connection object) {
            super(object);
        }
    }

}