Python SockerServer Thread
'''
Created on 2019-10-16
SOCKETSERVER
ThreadedTCPServer:多线程接收客户端信息
@author: heidu004
'''
#!/usr/bin/python3
# -*-coding:utf-8 -*-
#BaseRequestHandlerclass
#StreamRequestHandler
from socketserver import TCPServer, StreamRequestHandler,ThreadingMixIn
import time
import threading
import ctypes
import inspect
ADDR = ('127.0.0.1', 1234)
BUFSIZ = 1024
__exitFlag__ = False
g_conn_pool = [] # 连接池
threadList = ["Thread-1", "Thread-2", "Thread-3"]
# 创建 StreamRequestHandler 类的子类
class MyRequestHandler(StreamRequestHandler):
def setup(self):
#self.request.sendall("连接服务器成功!".encode(encoding='utf8'))
# client_address 属性的值为客户端的主机端口元组
print('{}'.format(self.client_address)+'接入')#客户端地址
# 加入连接池
g_conn_pool.append(self.request)
# 重写 handle 方法，该方法在父类中什么都不做
# 当客户端主动连接服务器成功后，自动运行此方法
def handle(self):
while True:
try:
# request.recv 方法接收客户端发来的消息
data = self.request.recv(BUFSIZ)
if not data:
break
msg = time.strftime("%Y-%m-%d %X") #获取结构化事件戳
print("[%s][%s]>>%s" %(msg, self.client_address,data.decode('utf-8')))
#self.request.sendall("已接收到相关指令".encode(encoding='utf8'))
except Exception:
print("客户端已断开{}".format(self.client_address))
# 意外掉线
self.remove()
break
for con_pool in g_conn_pool:#向所有客户端转发
if con_pool != self.request:
con_pool.sendall('[{}][{}] {}'.format(time.strftime("%Y-%m-%d %X"),self.client_address, data.decode()).encode())
else:
con_pool.sendall('[{}] [以发送]'.format(time.strftime("%Y-%m-%d %X")).encode())
# request.sendall 方法发送消息给客户端
#self.request.sendall('[{}] {}'.format(time.ctime(), data.decode()).encode())
def finish(self):
print("清除了这个客户端。")
def remove(self):
print("有一个客户端掉线了。")
g_conn_pool.remove(self.request)
class ThreadedTCPServer(ThreadingMixIn, TCPServer):
pass
def _async_raise(tid, exctype):
"""引发异常，根据需要执行清理"""
tid = ctypes.c_long(tid)
if not inspect.isclass(exctype):
exctype = type(exctype)
res = ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, ctypes.py_object(exctype))
if res == 0:
raise ValueError("无效的线程ID")
elif res != 1:
ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)
raise SystemError("PyThreadState_SetAsyncExc 无效")
def stop_thread(thread):
_async_raise(thread.ident, SystemExit)
def main():
# 主线程逻辑
cmd=''
while True:
cmd = input('>>')
print(cmd)
if cmd == 'HELP':
print('输入COUNT:查看当前在线人数')
print('输入INPUT:给指定客户端发送消息')
print('输入CLOSE:关闭服务端')
elif cmd == 'COUNT':
print("--------------------------")
print("当前在线人数：", len(g_conn_pool))
elif cmd == 'INPUT':
if len(g_conn_pool) >0:
print("--------------------------")
index, msg = input("请输入“索引,消息”的形式：>>").split(",")
g_conn_pool[int(index)].sendall(msg.encode(encoding='utf8'))
else:
print('没有客户端连接')
elif cmd == 'INPUTS':
if len(g_conn_pool) >0:
print("--------------------------")
__exitFlag__=True
else:
print('没有客户端连接')
elif cmd == 'CLOSE':
server.shutdown()#关闭socket的功能
print("关闭socket的功能")
stop_thread(server_thread)#关闭关闭server_thread的功能
print("关闭关闭server_thread的功能")
print('程序已经关闭')
exit()
else:
print('输入正确指令查看HELP')
if __name__ == '__main__':
msg = time.strftime("%Y-%m-%d %X") #获取结构化事件戳
print('---------------------------------------------------------------------------')
print('| _____ &&&&_> > ')
print('| \/,---< &&&&&&\ \' 输入HELP:查看帮助 ')
print('| ( )c~c ~~@~@ )-- &&\ \' 输入COUNT:查看当前在线人数 ')
print('| C >/ \< |&/ 输入INPUT:给指定客户端发送消息 ')
print('| \_O/ - 哇塞 <<_ * - *_>> 输入CLOSE:关闭服务端 ')
print('| ,- >o<-. / ____ _/' )
print('| / \/ \ / /\ _) _)')
print('| / /| | |\ \ / / ) |')
print('| \ \| | |/ / \ \ / | [%s]'%(msg))
print('| \_\ | |_/ \ \_ | [%s]-缓存[%s]'%(ADDR,BUFSIZ))
print('| /_/`___|_\ /_/\____| ')
print('| | | | \ \| SOCKETSERVER_ThreadedTCPServer')
print('| | | | `. ) 2019-10-20')
print('| | | | / / heidu004')
print('| |__|_|_ /_/| 服务器永远等待客户端的连接')
print('| (____)_) |\_\_ ')
print('-----------------------------------------------------------------------------')
server = ThreadedTCPServer(ADDR, MyRequestHandler)
# 新开一个线程运行服务端
server_thread = threading.Thread(target=server.serve_forever)# 服务器永远等待客户端的连接
server_thread.daemon = True
server_thread.start()
main()