Python之Http多线程/多进程高并发高稳定服务器构建

xingyun86 2018-12-24 2252

Python是一门不错的脚本语言,用途非常广泛,也非常简单易懂,异常处理也非常完善。近日使用了Python的HTTP服务器,发现并发支持并不完备,也不够稳定。于是看了下SimpleHTTPServer的实现,修改出多进程和多线程的支持。

#!coding:utf-8
#!/usr/bin/env python
# coding=utf-8
# -*- coding: utf-8 -*-
import os
import sys
import socket
import SocketServer
import BaseHTTPServer
from SimpleHTTPServer import SimpleHTTPRequestHandler
from modules.sigs import CSigs
class NOBHTTPRequestHandler (SimpleHTTPRequestHandler):
    def end_headers(self):
        self.send_header("Access-Control-Allow-Origin", "*")
        self.send_header("Access-Control-Allow-Methods",
                         "POST, GET, OPTIONS, PUT, DELETE")
        self.send_header("Access-Control-Allow-Headers",
                         "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
        SimpleHTTPRequestHandler.end_headers(self)
    def do_OPTIONS(self):
        self.send_response(200)
        self.end_headers()
class ForkingHTTPServerOnlyLinux(SocketServer.ForkingTCPServer):
    # Allow reuse address
    allow_reuse_address = 1
    def server_bind(self):
        """Override server_bind to store the server name."""
        SocketServer.TCPServer.server_bind(self)
        host, port = self.socket.getsockname()[:2]
        self.server_name = socket.getfqdn(host)
        self.server_port = port
def StartHttpServerOnlyLinux(HandlerClass=SimpleHTTPRequestHandler, ServerClass=ForkingHTTPServerOnlyLinux, ProtocolVersion="HTTP/1.0"):
    """
    HTTP request handler class.
    This runs an HTTP server on port 8000 (or the first command line
    argument).
    ProtocolVersion -- "HTTP/1.0","HTTP/1.1","HTTP/2.0"
    """
    if sys.argv[1:]:
        port = int(sys.argv[1])
    else:
        port = 8000
    if sys.argv[2:]:
        root = sys.argv[2]
    else:
        root = os.getcwd()
    # root path is exist
    if not os.path.isdir(root):
        try:
            os.makedirs(root)
        except Exception:
            pass
    # root path is exist again
    if os.path.isdir(root):
        os.chdir(root)
        server_address = ("", port)
        HandlerClass.protocol_version = ProtocolVersion
        httpd = ServerClass(server_address, HandlerClass)
        sa = httpd.socket.getsockname()
        print "Serving HTTP on", sa[0], "port", sa[1],  "root", root, "..."
        httpd.serve_forever()
class ThreadingHTTPServer(SocketServer.ThreadingTCPServer):
    # Allow reuse address
    allow_reuse_address = 1
    def server_bind(self):
        """Override server_bind to store the server name."""
        SocketServer.TCPServer.server_bind(self)
        host, port = self.socket.getsockname()[:2]
        self.server_name = socket.getfqdn(host)
        self.server_port = port
def StartHttpServer(HandlerClass=SimpleHTTPRequestHandler, ServerClass=ThreadingHTTPServer, ProtocolVersion="HTTP/1.0"):
    """
    HTTP request handler class.
    This runs an HTTP server on port 8000 (or the first command line
    argument).
    ProtocolVersion -- "HTTP/1.0","HTTP/1.1","HTTP/2.0"
    """
    if sys.argv[1:]:
        port = int(sys.argv[1])
    else:
        port = 8000
    if sys.argv[2:]:
        root = sys.argv[2]
    else:
        root = os.getcwd()
    # root path is exist
    if not os.path.isdir(root):
        try:
            os.makedirs(root)
        except Exception:
            pass
    # root path is exist again
    if os.path.isdir(root):
        os.chdir(root)
        server_address = ("", port)
        HandlerClass.protocol_version = ProtocolVersion
        httpd = ServerClass(server_address, HandlerClass)
        sa = httpd.socket.getsockname()
        print "Serving HTTP on", sa[0], "port", sa[1],  "root", root, "..."
        httpd.serve_forever()
'''
自定义主函数
'''
def main():
    # 注册CTRL+C的信号处理
    CSigs.reg_sig()
    while(True):
        try:
            # Multi-processes service only on linux
            # StartHttpServerOnlyLinux()
            # Multi-threads service on all the systems
            StartHttpServer(NOBHTTPRequestHandler)
        except Exception, e:
            print("Exception:" + str(e))
            pass
if __name__ == "__main__":
    main()


上传的附件:
×
打赏作者
最新回复 (0)
只看楼主
全部楼主
返回