• 欢迎访问开心洋葱网站,在线教程,推荐使用最新版火狐浏览器和Chrome浏览器访问本网站,欢迎加入开心洋葱 QQ群
  • 为方便开心洋葱网用户,开心洋葱官网已经开启复制功能!
  • 欢迎访问开心洋葱网站,手机也能访问哦~欢迎加入开心洋葱多维思维学习平台 QQ群
  • 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏开心洋葱吧~~~~~~~~~~~~~!
  • 由于近期流量激增,小站的ECS没能经的起亲们的访问,本站依然没有盈利,如果各位看如果觉着文字不错,还请看官给小站打个赏~~~~~~~~~~~~~!

python编写的多线程Socket服务器模块

python 水墨上仙 1141次浏览

python编写的多线程Socket服务器模块

#!/usr/bin/env python
# -*- coding: utf8 -*-
#
# [代码名字: Threaded Server]
# [代码分类: Python Core, socket, threading]
# [代码描述: Simple example of Python's socket and threading modules]
# [代码文档: http://docs.python.org/library/socket.html, http://docs.python.org/library/threading.html]
# [代码作者: Happyonion75271.com  <happyonion@gmail.com>]
# [代码协议: GPL]
import sys
import socket
import threading
import time
QUIT = False
class ClientThread( threading.Thread ):
    ''' 
    Class that implements the client threads in this server
    '''
    def __init__( self, client_sock ):
        '''
        Initialize the object, save the socket that this thread will use.
        '''
        threading.Thread.__init__( self )
        self.client = client_sock
    def run( self ):
        ''' 
        Thread's main loop. Once this function returns, the thread is finished 
        and dies. 
        '''
        #
        # Need to declare QUIT as global, since the method can change it
        #
        global QUIT
        done = False
        cmd = self.readline()
        #
        # Read data from the socket and process it
        #
        while not done:
            if 'quit' == cmd :
                self.writeline( 'Ok, bye' )
                QUIT = True
                done = True
            elif 'bye' == cmd:
                self.writeline( 'Ok, bye' )
                done = True
            else:
                self.writeline( self.name )
            cmd = self.readline()
        #
        # Make sure the socket is closed once we're done with it
        #
        self.client.close()
        return
    def readline( self ):
        ''' 
        Helper function, reads up to 1024 chars from the socket, and returns 
        them as a string, all letters in lowercase, and without any end of line 
        markers '''
        result = self.client.recv( 1024 )
        if( None != result ):
            result = result.strip().lower()
        return result
    def writeline( self, text ):
        ''' 
        Helper function, writes teh given string to the socket, with an end of 
        line marker appended at the end 
        '''
        self.client.send( text.strip() + '\n' )
class Server:
    ''' 
    Server class. Opens up a socket and listens for incoming connections.
    Every time a new connection arrives, it creates a new ClientThread thread
    object and defers the processing of the connection to it. 
    '''
    def __init__( self ):
        self.sock = None
        self.thread_list = []
    def run( self ):
        '''
        Server main loop. 
        Creates the server (incoming) socket, and listens on it of incoming
        connections. Once an incomming connection is deteceted, creates a 
        ClientThread to handle it, and goes back to listening mode.
        '''
        all_good = False
        try_count = 0
        #
        # Attempt to open the socket
        #
        while not all_good:
            if 3 < try_count:
                #
                # Tried more than 3 times, without success... Maybe the port
                # is in use by another program
                #
                sys.exit( 1 )
            try:
                #
                # Create the socket
                #
                self.sock = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
                #
                # Bind it to the interface and port we want to listen on
                #
                self.sock.bind( ( '127.0.0.1', 5050 ) )
                #
                # Listen for incoming connections. This server can handle up to
                # 5 simultaneous connections
                #
                self.sock.listen( 5 )
                all_good = True
                break
            except socket.error, err:
                #
                # Could not bind on the interface and port, wait for 10 seconds
                #
                print 'Socket connection error... Waiting 10 seconds to retry.'
                del self.sock
                time.sleep( 10 )
                try_count += 1
        print "Server is listening for incoming connections."
        print "Try to connect through the command line, with:"
        print "telnet localhost 5050"
        print "and then type whatever you want."
        print
        print "typing 'bye' finishes the thread, but not the server ",
        print "(eg. you can quit telnet, run it again and get a different ",
        print "thread name"
        print "typing 'quit' finishes the server"
        try:
            #
            # NOTE - No need to declare QUIT as global, since the method never 
            #    changes its value
            #
            while not QUIT:
                try:
                    #
                    # Wait for half a second for incoming connections
                    #
                    self.sock.settimeout( 0.500 )
                    client = self.sock.accept()[0]
                except socket.timeout:
                    #
                    # No connection detected, sleep for one second, then check
                    # if the global QUIT flag has been set
                    #
                    time.sleep( 1 )
                    if QUIT:
                        print "Received quit command. Shutting down..."
                        break
                    continue
                #
                # Create the ClientThread object and let it handle the incoming
                # connection
                #
                new_thread = ClientThread( client )
                print 'Incoming Connection. Started thread ',
                print new_thread.getName()
                self.thread_list.append( new_thread )
                new_thread.start()
                #
                # Go over the list of threads, remove those that have finished
                # (their run method has finished running) and wait for them 
                # to fully finish
                #
                for thread in self.thread_list:
                    if not thread.isAlive():
                        self.thread_list.remove( thread )
                        thread.join()
        except KeyboardInterrupt:
            print 'Ctrl+C pressed... Shutting Down'
        except Exception, err:
            print 'Exception caught: %s\nClosing...' % err
        #
        # Clear the list of threads, giving each thread 1 second to finish
        # NOTE: There is no guarantee that the thread has finished in the
        #    given time. You should always check if the thread isAlive() after
        #    calling join() with a timeout paramenter to detect if the thread
        #    did finish in the requested time
        #
        for thread in self.thread_list:
            thread.join( 1.0 )
        #
        # Close the socket once we're done with it
        #
        self.sock.close()
if "__main__" == __name__:
    server = Server()
    server.run()
    print "Terminated"

 


开心洋葱 , 版权所有丨如未注明 , 均为原创丨未经授权请勿修改 , 转载请注明python编写的多线程Socket服务器模块
喜欢 (0)
加载中……