项目上有个需求是,客户可以同时选择5个接口去查询外部数据,这5个外部接口独立,还要考虑最大并发量,于是本人考虑用Semaphore和线程池去解决这个问题。
Semaphore:负责协调各个线程, 以保证它们能够正确、合理的使用公共资源。也是操作系统中用于控制进程同步互斥的量。以一个停车场是运作为例。为了简单起见,假设停车场只有三个车位,一开始三个车位都是空的。这时如果同时来了五辆车,看门人允许其中三辆不受阻碍的进入,然后放下车拦,剩下的车则必须在入口等待,此后来的车也都不得不在入口处等待。这时,有一辆车离开停车场,看门人得知后,打开车拦,放入一辆,如果又离开两辆,则又可以放入两辆,如此往复。在这个停车场系统中,车位是公共资源,每辆车好比一个线程,看门人起的就是信号量的作用。
本文写个demo,希望用最简单的代码说明问题。
第一步:定义一个注解,因为处理类可能有很多方法,自定义一个注解,可以执行被标记为注解的方法
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface QueryMethod {
String value();
}
第二步:写个父类,主要是定义线程处理的逻辑以及定义Semaphore基本参数
import java.lang.reflect.Method;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
public class QueryManager {
public static final int MAX_OCCURS = 50;
private static Semaphore semp = new Semaphore(MAX_OCCURS);
void dorun() throws InterruptedException{
ExecutorService exec = Executors.newFixedThreadPool(5);
for(Method m : this.getClass().getMethods()){
QueryMethod a = m.getAnnotation(QueryMethod.class);
if(a != null){
exec.execute(new MethodInvokeRunner(semp, this, m, new Object[]{}));
}
}
exec.shutdown();
while(!exec.awaitTermination(3, TimeUnit.MINUTES)){
continue;
}
}
private class MethodInvokeRunner implements Runnable{
private Object obj;
private Method m;
private Object[] args;
private Semaphore semp;
MethodInvokeRunner(Semaphore semp, Object obj, Method m, Object[] args){
this.semp = semp;
this.obj = obj;
this.m = m;
this.args = args;
}
@Override
public void run() {
try {
semp.acquire();
m.invoke(obj, args);
} catch (Exception e1) {
e1.printStackTrace();
} finally {
semp.release();
}
}
}
}
第三步:定义子类,写具体的业务处理方法,主要为5个接口的查询方法
/**
* 具体业务逻辑
* @author ****
* @date *****
*/
public class QueryManagerDoRun extends QueryManager{
@QueryMethod(value = "0010")
public void query1(){
System.out.println("0010-开始");
System.out.println("查询接口1的逻辑");
System.out.println("0010-结束");
}
@QueryMethod(value = "0020")
public void query2(){
System.out.println("0020-开始");
System.out.println("查询接口2的逻辑");
System.out.println("0020-结束");
}
@QueryMethod(value = "0030")
public void query3(){
System.out.println("0030-开始");
System.out.println("查询接口3的逻辑");
System.out.println("0030-结束");
}
@QueryMethod(value = "0040")
public void query4(){
System.out.println("0040-开始");
System.out.println("查询接口4的逻辑");
System.out.println("0040-结束");
}
@QueryMethod(value = "0050")
public void query5(){
System.out.println("0050-开始");
System.out.println("查询接口5的逻辑");
System.out.println("0050-结束");
}
}