Python使用UDP广播实现服务器自动发现

2016-08-30 董付国 Python小屋 Python小屋

 机房管理软件或者教学软件一般都由服务器和客户端组成,当客户端启动之后会自动搜索服务器,这是如何实现的呢?下面通过一段代码来演示其中一种比较高效的方法。

客户端代码:

import socket
import time

def findServer():
    #创建socket对象
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    #绑定socket
    sock.bind(('', 5000))
    while True:
        #接收信息
        data, addr = sock.recvfrom(1024)
        #服务器广播信息
        if data.decode() == 'ServerIP':
            #查看服务器IP
            print(addr[0])
        #休息一秒后
        time.sleep(1)

findServer()

服务端代码:

import socket
import time

def findServer():
    #检查是否已有服务器运行
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.bind(('', 5000))
    for i in range(3):
        data, addr = sock.recvfrom(1024)
        if data.decode() == 'ServerIP':
            sock.close()
            return True
        time.sleep(0.5)
    sock.close()
    return False

def sendServerIP():
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    while True:
        #获取本机IP
        IP = socket.gethostbyname(socket.gethostname())
        #255表示广播地址
        IP = IP[:IP.rindex('.')]+'.255'
        #发送信息
        sock.sendto('ServerIP'.encode(), (IP, 5000))
        time.sleep(1)


already = findServer()
#如果已有服务器运行,则不再启动
if not already:
    sendServerIP()


温馨提示:单击文章顶部作者名字旁边浅蓝色的“Python小屋”进入公众号,关注后可以查看更多内容!


欢迎转发给您的朋友,或许这正是Ta需要的知识!