欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

连接池之四

程序员文章站 2022-05-12 17:55:16
...

//简单连接池之四
public class RedisPool5 {

 public static void main(String[] args) {
  RedisPool5 pool = new RedisPool5(10,20,5);
  try {
   for(int i=0;i<280;i++) {
    Thread t1 = new TestThread2("t"+i, pool);
    t1.start();
    t1.join();
   }
  } catch (InterruptedException e) {
   e.printStackTrace();
  }
  pool.close();
 }

 
 //private Vector<JedisExt> list = new Vector<JedisExt>();
 //private List<JedisExt> list = new LinkedList<JedisExt>();
 private BlockingQueue<JedisExt> list = new LinkedBlockingQueue<JedisExt>();
 private int size; //当前正使用的连接数
 private int maxSize;
 private int minSize;
 
 public RedisPool5(int size,int maxSize,int minSize) {
  this.size = size;
  this.maxSize = maxSize;
  this.minSize = minSize;
  init();
 }
 
 private void init() {
  JedisExt je = null;
  try {
  for(int i=0;i<this.size;i++) {
   je = new JedisExt(initConnection());
     list.offer(je, 10, TimeUnit.SECONDS);
  }
  } catch (InterruptedException e) {
   e.printStackTrace();
  }
 }
 
 public JedisExt getConnection() {
  //先从闲置中获取,从list删除
  JedisExt j = null;
  try {
   for(int i=0;i<list.size();i++) {
    //j = list.get(i);
    j = list.take();
    if(!j.isused) {
     System.out.println("获取连接..");
     j.setIsused(true); //使用中
     return j;
    }
   }
  } catch (InterruptedException e) {
   e.printStackTrace();
  }
  //创建新的连接
  if(size<maxSize) {
   System.out.println("创建新连接..");
   size++;
   JedisExt je = new JedisExt(initConnection(), true);
   //list.add(je);
   return je;
  }
  //已到最大连接数返回空
  System.out.println("无连接..");
  return null;
 }
 
 //关闭所有连接
 public void close() {
  JedisExt je = null;
  try {
   for(int i=0;i<list.size();i++) {
    je = list.take();
    je.getJedis().close();
   }
  } catch (InterruptedException e) {
   e.printStackTrace();
  }
  list.clear();
  this.size = 0;
 }
 
 //释放连接
 public void release(JedisExt e) {
  e.setIsused(false);
  //list.add(e); //加入list
  try {
   list.offer(e, 10, TimeUnit.SECONDS);
  } catch (InterruptedException e1) {
   e1.printStackTrace();
  }
  //size++;
 }
 
 private class JedisExt extends Jedis{
  private boolean isused = false; //是否空闲
  private Jedis jedis;
  
  public JedisExt(Jedis j) {
   this.jedis = j;
  }
  
  public JedisExt(Jedis j, boolean isused) {
   this.jedis = j;
   this.isused = true;
  }

  public synchronized boolean isIsused() {
   return isused;
  }

  public synchronized void setIsused(boolean isused) {
   this.isused = isused;
  }
  
  public void close() {
   this.setIsused(false); //未使用
   //this.jedis.close();
  }

  public synchronized Jedis getJedis() {
   return jedis;
  }

 }
 
 //获取单个连接
 private static Jedis initConnection(){
  String host = "192.168.19.1";
     Jedis client = new Jedis(host, 6379);
     return client;
 }
 
 private static class TestThread2 extends Thread{
  private RedisPool5 pool;
  public TestThread2(String name, RedisPool5 pool) {
   super(name);
   this.pool = pool;
  }
  
  public void run() {
   try {
    Thread.sleep(new Random().nextInt(100));
   } catch (InterruptedException e1) {
    e1.printStackTrace();
   }
   JedisExt e = pool.getConnection();
   boolean b = false;
   if(e!=null) {
    b = e.isIsused();
   }
   System.out.println(e + " " + b);
   try {
    Thread.sleep(new Random().nextInt(600));
   } catch (InterruptedException e1) {
    e1.printStackTrace();
   }
   //释放连接
   if(e!=null)
   pool.release(e);
  }
 }
}

相关标签: 连接池