#!/usr/bin/env python
#*-* coding:utf-8 *-*
import socket
import select
import sys
import signal
class ChatServer():
  def __init__(self,host,port,timeout=10,backlog=5):
    #記錄連接的客戶端數(shù)量
    self.clients =0
    #存儲連接的客戶端socket和地址對應(yīng)的字典
    self.clientmap={}
    #存儲連接的客戶端socket
    self.outputs = []
    #建立socket
    self.server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    self.server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    self.server.bind((host,port))
    self.server.listen(backlog)
    #增加信號處理
    signal.signal(signal.SIGINT,self.sighandler) 
  def sighandler(self):
    sys.stdout.write("Shutdown Server......\n")
    #向已經(jīng)連接客戶端發(fā)送關(guān)系信息，并主動關(guān)閉socket
    for output in self.outputs:
      output.send("Shutdown Server")
      output.close()
    #關(guān)閉listen
    self.server.close()
    sys.stdout.flush()
  #主函數(shù)，用來啟動服務(wù)器
  def run(self):
    #需要監(jiān)聽的可讀對象
    inputs=[self.server]
    
    runing=True
    #添加監(jiān)聽主循環(huán)
    while runing:
      try:
        readable,writeable,exceptional = select.select(inputs,self.outputs,[])
        #此處會被select模塊阻塞，只有當(dāng)監(jiān)聽的三個參數(shù)發(fā)生變化時，select才會返回
      except select.error,e:
        break
      #當(dāng)返回的readable中含有本地socket的信息時，表示有客戶端正在請求連接
      if self.server in readable:
        #接受客戶端連接請求
        client,addr=self.server.accept()
        sys.stdout.write("New Connection from %s\n"%str(addr))
        sys.stdout.flush()
        #更新服務(wù)器上客戶端連接情況
        #1，數(shù)量加1
        #2，self.outputs增加一列
        #3，self.clientmap增加一對
        #4, 給input添加可讀監(jiān)控
        self.clients += 1
        self.outputs.append(client)
        self.clientmap[client]=addr
        inputs.append(client)
      
      #readable中含有已經(jīng)添加的客戶端socket，并且可讀
      #說明 1,客戶端有數(shù)據(jù)發(fā)送過來或者 2,客戶端請求關(guān)閉
      elif len(readable) != 0:
        #1, 取出這個列表中的socket
        csock=readable[0]
        #2, 根據(jù)這個socket，在事先存放的clientmap中，去除客戶端的地址，端口的詳細信息
        host,port = self.clientmap[csock]
        #3,取數(shù)據(jù), 或接受關(guān)閉請求，并處理
        #注意，這個操作是阻塞的，但是由于數(shù)據(jù)是在本地緩存之后，所以速度會非常快
        try:
          data = csock.recv(1024).strip()
          for cs in self.outputs:
            if cs != csock:
              cs.send("%s\n"%data)
        except socket.error,e:
          self.clients -= 1
          inputs.remove(csock)
          self.outputs.remove(csock)
          del self.clientmap[csock]
      #print self.outputs
    self.server.close()
        
if __name__ == "__main__":
  chat=ChatServer("",8008)
  chat.run()