package hjk.java;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import javax.swing.JOptionPane;

public class ServerPortListener implements Runnable{
 /**
  * @keywords 自動化管理的　伺服器埠多執行緒監聽　工具類 　
  * @status 可用 　除錯中
  * @future 下一步功能擴充套件：　連線數管理
  * @author Karl Jinkey 

[email protected] 2010-05-20
  * @sample 使用樣例：　
  *    ServerPortListener listener = new ServerPortListener(serverPort);
    Thread threadListener = new Thread(listener);
    threadListener.setDaemon(true); 
    threadListener.start();//啟動一個監聽器執行緒
    threadListener.join();//阻塞呼叫程序，等待客戶端連線
    socket = listener.getSocket();//獲得傳入的連線
    if(socket==null){
     //TODO:　錯誤處理程式碼
    }else{
     //TODO: 進一步的Socket處理
    }
  */
 private static Integer objCount = 0;
 private static Map<Short, ServerSocket> serverMap = Collections.synchronizedMap(new HashMap<Short, ServerSocket>());
 private static Set<Socket> socketSet = Collections.synchronizedSet(new HashSet<Socket>());
 
 private ServerSocket server = null;
 private Socket socket = null;
 private short serverPort;
 public InfoMap lastError = new InfoMap();
 public ServerPortListener(short serverPort) {
  System.out.println(objCount);
  synchronized (objCount) {
   objCount++;
  }
  this.serverPort = serverPort;
 } 
 public Socket getSocket() {
  return socket;
 }

 public void setSocket(Socket socket) {
  this.socket = socket;
 }

 public short getServerPort() {
  return serverPort;
  
 }

 public void setServerPort(short serverPort) {
  this.serverPort = serverPort;
 }

 @Override
 public void run() {
  while(true){
   try {
    server = serverMap.get(serverPort);
    if(server!=null ){
     if(!server.isClosed()) break;//埠服務存在，跳出迴圈
     serverMap.remove(serverPort);
    }
    server = new ServerSocket(serverPort);
    serverMap.put(serverPort, server);
    server.setReuseAddress(true);
    break;
   } catch (IOException e) {
    //lastError.msg ="建立服務埠["+serverPort+"]時發生異常："+e.toString();
    if(server!=null){
     try{
      server.close();
     }catch(IOException e2){
      lastError.msg ="關閉ServerSocket時發生異常："+e2.toString();      
     }
    }
    int res = JOptionPane.showConfirmDialog(null,
      "建立服務埠["+serverPort+"]時發生異常："+e.toString(), "錯誤提示：",JOptionPane.YES_NO_OPTION);
    if(res!=JOptionPane.YES_OPTION) {     
     return;
    }
    //System.gc();
    continue;
   }
   
  }
  
  while (true) {    
   try{
    //if(server.isBound()){//測試繫結狀態：
     System.out.println("監聽埠："+serverPort);
     socket = server.accept();
     socketSet.add(socket);
     System.out.println("監聽成功，啟動服務，客戶端IP："+socket.getRemoteSocketAddress().toString());
     break;//監聽到一個連線請求，跳出迴圈
//    }else{
//     server.close();
//     server = new ServerSocket(serverPort);
//     server.setReuseAddress(true);
//    }     
   }catch(Exception e){
    lastError.msg ="監聽器異常終止："+e.toString();
   }
  }
 }
 //釋放資源：
 private void freeAll(){
  if(serverMap.size()!=0){
   Set<Short> set = serverMap.keySet();
   for(Short key:set){
    System.out.println("ser.close():"+objCount);
    ServerSocket ser = serverMap.get(key);
    if(ser!=null&&!ser.isClosed()){
     try{
      ser.close();
     }catch(IOException e){}
    }    
   }
  }
  if(socketSet.size()!=0){
   
   for(Socket so:socketSet){
    if(so!=null&&!so.isClosed()){
     try{
      so.close();
     }catch(IOException e){}
    }    
   }
  }  
 }
 @Override
 protected void finalize() throws Throwable {
  synchronized (objCount) {
   //System.out.println("objCount:"+objCount);
   objCount--;
   if(objCount==0){
    freeAll();
    super.finalize();
    return;
   }
  }
  if(socket!=null){
   if(!socket.isClosed()){
    try{
     socket.close();
    }catch(IOException e){}
   }
   socketSet.remove(socket);
  }
  super.finalize();
 }
}