一个非常简单,非常短小的线程池,请高手指点

[abcdhy]

一个非常简单,非常短小的线程池,请高手指点一下!!


最近写了一个 HTTP 代理服务器,发现访问网页时建立的连接很多,消耗的线程也非常的多,对于系统是

一个不小的开销.而且这些线程存在的时间都很短,99%以上的线程存在的时间都在毫秒的等级,相对来说

线程的建立的注销就占了绝大部分的CPU时间.
因此,在网上搜了一下线程池的资料,发现里面的东西不是太大太复杂,就是写得太烂,根本没有一点专业

水准.
没办法,只好自己简单的学习了一下 wait()/notify()机制,写了一个很小的线程池.
代码如下(一共2个类):
/*
*一个简单的线程池 ThreadPool .java
*/
public class ThreadPool {
//以下是配置信息,可以更改
static int MAX_THREAD = 1000; //未使用
static int MIN_THREAD = 14;

static int id = 1; //线程 ID 号,主要用于监视线程的工作情况

static private ThreadPool pool = new ThreadPool();
static public ThreadPool getThreadPool() {
return pool;
}

Stack<WorkThread> stack = new Stack<WorkThread>(MIN_THREAD);
private ThreadPool() {
}

synchronized public boolean putWorkThread(WorkThread wt) {
if(stack.size()<MIN_THREAD){
stack.push(wt);
return true;
} else {
return false;
}
}

synchronized public WorkThread getWorkThread() {
WorkThread wt = null;
if(stack.isEmpty()) {
wt = new WorkThread(this);
new Thread(wt,"线程ID:"+id).start();
id++;
} else {
wt = stack.pop();
}
return wt;
}
}
--------------------------------------------------------------------------
/*
*工作线程类 WorkThread.java
*/
public class WorkThread implements Runnable {
Object lock = new Object();
Runnable runner = null;
ThreadPool pool = null;

public WorkThread(ThreadPool pool) {
this.pool = pool;
}

public void start(Runnable r) {
runner = r;
synchronized(lock) {
lock.notify();
}
}

public void run() {
while(true) {
if(runner != null) {
runner.run();
runner = null; //及时回收资源
}
if(pool.putWorkThread(this)) {
System.out.println (Thread.currentThread().getName()+" 被回收!");
synchronized(lock) {
try {
lock.wait();
} catch (InterruptedException ie) {
System.out.println ("停止线程时出现异常");
}
}
} else {
System.out.println (Thread.currentThread().getName()+" 被丢弃!");
break;
}
}
}
}

使用方法:
Runnable r = ......;
new Thread(r).start();
这是你以前的代码,使用线程池只需要将后一句变成
ThreadPool.getThreadPool().getWorkThread().start(r);
就可以了,其他的代码不用任何更改.

这个线程池还有两个功能没有实现:
1.最大线程数的控制
2.空闲线程等待一定时间后自动注销
但这个线程池也有一个好处--可以随时改变最小线程的数量.

由于学 Java 不太懂,其中可能有不太完善的地方,还请各位高手指点一下,谢谢!!

[abcdhy]

自己写的一个 Stack 类,这里仅仅我只是为了性能.
大可使用 java.util 下的相应类代替.

具体代码如下:
/* Stack.java
*一个简单的堆栈类,主要是为了快速的删除元素和取得大小
*/
public class Stack<E> {

private Object[] values = null;
private int pos = 0;

public Stack() {
this(10);
}

public Stack(int length) {
values = new Object[length];
pos = 0;
}

public E push(E item) {
if(pos < values.length) {
values[pos] = item;
pos++;
} else {
Object[] temp = new Object[values.length*2];
System.arraycopy(values,0,temp,0,values.length);
values = temp;
values[pos] = item;
pos++;
}
return item;
}

public E pop() {
E temp = null;
if(pos > 0) {
temp = (E)values[--pos];
values[pos] = null;
}
return temp;
}

public E peek() {
E temp = null;
if(pos > 0) {
temp = (E)values[pos-1];
values[pos] = null;
}
return temp;
}

public boolean isEmpty() {
return (pos<1);
}

public int size() {
return pos;
}
}

[zedk928]

支持啊

[过客猫2022]

ding!

[abcdhy]

呵呵,今天仔细的研究了一下,发现一个 bug ,现已修正,代码如下:

/*
*一个简单的线程池
*/
public class ThreadPool {
//以下是配置信息,可以更改
static int MAX_THREAD = 1000; //未使用
static int MIN_THREAD = 14;

static int id = 1; //线程 ID 号,主要用于监视线程的工作情况

static private ThreadPool pool = new ThreadPool();
static public ThreadPool getThreadPool() {
return pool;
}

Stack<WorkThread> stack = new Stack<WorkThread>(MIN_THREAD);
private ThreadPool() {
}

synchronized public boolean putWorkThread(WorkThread wt) {
if(stack.size()<MIN_THREAD){
stack.push(wt);
return true;
} else {
return false;
}
}

synchronized public WorkThread getWorkThread() {
if(stack.isEmpty()) {
Thread thread = new Thread(new WorkThread(this),"线程ID:"+id);
thread.start();
try {
thread.join();
} catch (InterruptedException ie) {
System.out.println ("建立线程时出现错误");
}
id++;
}
return stack.pop();
}
}

[abcdhy]

为避免死锁,最后代码如下,请大家多多测试,帮我找找问题:

/*
*一个简单的线程池
*/
public class ThreadPool {
//以下是配置信息,可以更改
static int MAX_THREAD = 1000; //未使用
static int MIN_THREAD = 14;

static int id = 1; //线程 ID 号,主要用于监视线程的工作情况

static private ThreadPool pool = new ThreadPool();
static public ThreadPool getThreadPool() {
return pool;
}

Stack<WorkThread> stack = new Stack<WorkThread>(MIN_THREAD);
private ThreadPool() {
}

synchronized public boolean putWorkThread(WorkThread wt) {
if(stack.size()<MIN_THREAD){
stack.push(wt);
return true;
} else {
return false;
}
}

public WorkThread getWorkThread() {
if(stack.isEmpty()) {
Thread thread = new Thread(new WorkThread(this),"线程ID:"+id);
thread.start();
try {
thread.join();
} catch (InterruptedException ie) {
System.out.println ("建立线程时出现错误");
}
id++;
}
WorkThread wt = null;
synchronized(stack) {
wt = stack.pop();
}
return wt;
}
}

[Ji秋风]

顶~

[chenyansong]

学习！

[longwing]

帮顶！顺便加标签。

[kaymo]

没玩过 嘿嘿

[fdm_sea]

落后了,已经用上1.5的语法了

[evilzydar]

MARK

[unat]

mark

[cqgaoke]

up