Python實現(xiàn)多線程/多進程的TCP服務(wù)器

#!C:\Python3.6.5\python.exe
# -*- coding: gbk -*-

import socket
import threading

class WSGIServer(object):
 def __init__(self, port):
 """初始化對象"""
 # 創(chuàng)建套接字
 self.tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
 # 解決程序端口占用問題
 self.tcp_server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
 # 綁定本地ip地址
 self.tcp_server_socket.bind(("", port))
 # 將套接字變?yōu)楸O(jiān)聽套接字，最大連接數(shù)量為100
 self.tcp_server_socket.listen(100)

 def run_forever(self):
 """設(shè)備連接"""
 while True:
 # 1.等待設(shè)備連接(通過ip地址和端口建立tcp連接)
 # 如果有設(shè)備連接，則會生成用于設(shè)備和服務(wù)器通訊的套接字：new_socket
 # 會獲取到設(shè)備的ip地址和端口
 new_socket, client_addr = self.tcp_server_socket.accept()
 print("設(shè)備{0}已連接".format(client_addr))

 # 2.創(chuàng)建線程處理設(shè)備的需求
 t1 = threading.Thread(target=self.service_machine, args=(new_socket, client_addr))
 t1.start()

 def service_machine(self, new_socket, client_addr):
 """業(yè)務(wù)處理"""
 while True:
 # 3.接收設(shè)備發(fā)送的數(shù)據(jù)，單次最大1024字節(jié)，按‘gbk'格式解碼
 receive_data = new_socket.recv(1024).decode("gbk")
 # 4.如果設(shè)備發(fā)送的數(shù)據(jù)不為空
 if receive_data:
 # 4.1 打印接收的數(shù)據(jù)，這里可以將設(shè)備發(fā)送的數(shù)據(jù)寫入到文件中
 # 獲取設(shè)備的ID信息
 print(receive_data)
 if receive_data[0:6] == "report":
  response = "SET OK:" + receive_data
 else:
  receive_data = receive_data[6:].split(",")[0]
  # 拼接響應(yīng)數(shù)據(jù)
  response = "alarm=" + receive_data + ",Switch:clear"
 print(response)
 # 4.2 返回原數(shù)據(jù)作為應(yīng)答，按‘utf-8'格式編碼
 new_socket.send(response.encode("utf-8"))
 # 5.當設(shè)備斷開連接時，會收到空的字節(jié)數(shù)據(jù)，判斷設(shè)備已斷開連接
 else:
 print('設(shè)備{0}斷開連接...'.format(client_addr))
 break

 # 關(guān)閉套接字
 new_socket.close()


def main(port):
 """創(chuàng)建一個WEB服務(wù)器"""
 wsgi_server = WSGIServer(port)
 print("服務(wù)器已開啟")
 wsgi_server.run_forever()



if __name__ == '__main__':
 port = 8125 # 指定端口
 main(8125)