Semaphoore 示例

1.第一步，实现一个数据库连接类，注意这里是简写版，需要实现很多方法和导入很多类

package DBPoolSemaphoore;

import com.mysql.jdbc.*;

/**
 *类说明：数据库连接的平庸实现
 */
public class SqlConnectImpl implements Connection{
      /*拿一个数据库连接*/
    public static final Connection fetchConnection(){
        
        return new SqlConnectImpl();
    }

}

2.数据库连接池类

package DBPoolSemaphoore;

import java.util.LinkedList;
import java.util.Random;
import java.util.concurrent.Semaphore;

import com.mysql.jdbc.Connection;

public class DBPoolSemaphore {
 
    //连接池大小
    private final static int POOL_SIZE=10;
    
    //定义两个指示器，分别表示池子还有可用连接和已用连接 

    //注意这里不能定义一个来增删信号，必须定义两个信号否则有BUG，因为release方法会不断使信号量上升，构造方法中定义的10个只表示初始化是10个
    private final Semaphore useful,useless;
    
    //连接池的存放器集合
    private static LinkedList<Connection> pool = new LinkedList<>();
    
    //初始化连接池
    static{
        for ( int i = 0 ; i < POOL_SIZE;i++){
            pool.addLast(SqlConnectImpl.fetchConnection());
        }
    }
    
    //构造函数中把useful信号和useless信号初始化，
    //既然useful表示可用就给10个信号
    //而useless表示正在使用的连接就赋值为0
    public DBPoolSemaphore(){
        this.useful=new Semaphore(10);
        this.useless=new Semaphore(0);
        
    }
    
    //归还信号，也就是归还连接
    public void returnConnect(Connection connection) throws InterruptedException{
        if ( connection !=null){
            System.out.println("当前有"+useful.getQueueLength()+"个线程等待数据库连接！"+
                    "可用连接数："+useful.availablePermits());
            //从归还信号中去掉一个，表示放回了一个连接
            useless.acquire();
            //锁住池子，因为是归还连接，所以把传进来的连接放入一个进池子
            synchronized (pool) {
                pool.addLast(connection);
            }
            //记得把可用信号增加一个，也就是释放一个，表示可用的连接+1
            useful.release();
        }
    }
    
    //锁住池子，拿信号，也就代表拿连接，拿了之后删掉并返回
    public Connection takeConnect() throws InterruptedException{
        //useful.acquire()方法表示从可用信号中拿掉一个
        //当没有的时候这个方法会被阻塞从而进入等待状态，而当被唤醒的时候说明已经有连接可用拿

       //要提前判断有没有信号，没有就阻塞，所以这一步要放在方法的第一步
        useful.acquire(); 
        //当acquire方法中有信号的时候就开始拿连接
        Connection  connection;
        synchronized (pool) {
            connection = pool.removeFirst();
        }
        return connection;
    } 
}

3.测试类

package DBPoolSemaphoore;

import java.util.Random;

import com.mysql.jdbc.Connection;

public class SemaphoreTest {

    private static DBPoolSemaphore dbPool = new DBPoolSemaphore();
    
    //数据库连接线程
    private static class BusiThread extends Thread{
        @Override
        public void run() { 
            long start = System.currentTimeMillis();
            try {
                //拿连接
                Connection connect = dbPool.takeConnect();
                System.out.println("Thread_"+Thread.currentThread().getId()
                        +"_获取数据库连接共耗时【"+(System.currentTimeMillis()-start)+"】ms.");
                
                System.out.println("查询数据完成，归还连接！");
                //归还连接
                dbPool.returnConnect(connect);
            } catch (InterruptedException e) {
            }
            
        }
    }
    
    public static void main(String[] args) {
        //定义50个线程抢连接
        for (int i = 0; i < 50; i++) {
            Thread thread = new BusiThread();
            thread.start();
        }
    }
}