有時候啼辣，我們希望提前結(jié)束線程，但安全可靠地停止線程御滩，并不是一件容易的事情鸥拧，如果立即停止線程，會使共享的數(shù)據(jù)結(jié)構(gòu)處于不一致的狀態(tài)削解，如目前已經(jīng)廢棄使用的Thread類的stop方法(它會使線程在拋出java.lang.ThreadDeath之后終止線程富弦，即使是在執(zhí)行synchronized方法的時候)。更好的做法是執(zhí)行完終止處理氛驮，再終止線程腕柜，即Two-phase Termination，兩階段終止模式矫废。
該模式有兩個角色：
- Terminator盏缤，終止者，負(fù)責(zé)接收終止請求蓖扑，執(zhí)行終止處理唉铜，處理完成后再終止自己。
- TerminationRequester：終止請求發(fā)出者律杠，用來向Terminator發(fā)出終止請求潭流。
該模式示例代碼如下：
Terminator：
public class CounterIncrement extends Thread {
private volatile boolean terminated = false;
private int counter = 0;
private Random random = new Random(System.currentTimeMillis());
@Override
public void run() {
try {
while (!terminated) {
System.out.println(Thread.currentThread().getName()+" "+counter++);
Thread.sleep(random.nextInt(1000));
}
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
this.clean();
}
}
private void clean() {
System.out.println("do some clean work for the second phase,current counter "+counter);
}
public void close() {
this.terminated = true;
this.interrupt();
}
}
TerminationRequester：
public class CounterTest {
public static void main(String[] args) throws InterruptedException {
CounterIncrement counterIncrement = new CounterIncrement();
counterIncrement.start();
Thread.sleep(15_000L);
//主動清理
counterIncrement.close();
}
}
這段代碼可以看出實現(xiàn)兩階段終止模式必須注意的是：
使用線程停止標(biāo)志和interrupt方法竞惋，兩者缺一不可
public void close() {
this.terminated = true;
this.interrupt();
}
這里使用了terminated作為線程停止標(biāo)志，變量采用volatile修飾灰嫉，避免了使用顯式鎖的開銷拆宛，又保證了內(nèi)存可見性。線程run方法會檢查terminated屬性熬甫，如果屬性為true胰挑，就停止線程，但線程可能調(diào)用了阻塞方法椿肩，處于wait狀態(tài)瞻颂，任務(wù)也就可能永遠(yuǎn)不會檢查terminated標(biāo)志；線程也有可能處于sleep()狀態(tài)郑象，等sleep時間過后再執(zhí)行終止?fàn)顟B(tài)贡这，程序的響應(yīng)性就下降了。你可以把方法改成如下運(yùn)行厂榛，線程停止明顯變慢了許多：
public void close() {
terminated = true;
}
模擬客戶端或者服務(wù)端都可能終止服務(wù)的例子
public class AppServer extends Thread {
private static final int DEFAULT_PORT = 12722;
private final static ExecutorService executor = Executors.newFixedThreadPool(10);
private int port;
private volatile boolean start = true;
private List<ClientHandler> clientHandlers = new ArrayList<>();
private ServerSocket server;
public AppServer() {
this(DEFAULT_PORT);
}
public AppServer(int port) {
this.port = port;
}
@Override
public void run() {
try {
server = new ServerSocket(port);
while (start) {
Socket client = server.accept();
ClientHandler clientHandler = new ClientHandler(client);
executor.submit(clientHandler);
this.clientHandlers.add(clientHandler);
}
} catch (IOException e) {
//throw new RuntimeException();
} finally {
this.dispose();
}
}
public void dispose() {
System.out.println("dispose");
this.clientHandlers.stream().forEach(ClientHandler::stop);
this.executor.shutdown();
}
public void shutdown() throws IOException {
this.start = false;
this.interrupt();
this.server.close();
}
}
public class ClientHandler implements Runnable {
private final Socket socket;
private volatile boolean running = true;
public ClientHandler(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
try (InputStream inputStream = socket.getInputStream();
OutputStream outputStream = socket.getOutputStream();
BufferedReader br = new BufferedReader(new InputStreamReader(inputStream));
PrintWriter printWriter = new PrintWriter(outputStream)) {
while (running) {
String message = br.readLine();
if (message == null) {
break;
}
System.out.println("Come from client >" + message);
printWriter.write("echo " + message+"\n");
printWriter.flush();
}
} catch (IOException e) {
//自動關(guān)閉的時候 將running
this.running = false;
}finally {
this.stop();
}
}
public void stop() {
if (!running) {
return;
}
this.running = false;
try {
this.socket.close();
} catch (IOException e) {
}
}
}
public class AppServerClient {
public static void main(String[] args) throws InterruptedException, IOException {
AppServer server = new AppServer(12135);
server.start();
Thread.sleep(20_000L);
server.shutdown();
}
}
mac telnet模擬客戶端輸入
bogon:~ kpioneer$ telnet localhost 12135
Trying ::1...
Connected to localhost.
Escape character is '^]'.
hello
echo hello
I love you
echo I love you
Connection closed by foreign host.
服務(wù)端輸出：
Come from client >hello
Come from client >I love you
dispose
總結(jié)：
可以看到盖矫，在子類使用兩階段終止模式時，其只需要實現(xiàn)各自所需要執(zhí)行的任務(wù)击奶，并且更新當(dāng)前任務(wù)的數(shù)量即可辈双。在某些情況下，當(dāng)前任務(wù)的數(shù)量也可以不進(jìn)行更新柜砾，比如在進(jìn)行終止時湃望，不關(guān)心當(dāng)前剩余多少任務(wù)需要執(zhí)行。
