天天看点

基于python的socket编程实现服务发现

问题与背景

做一个边云协作的项目,需要在边端自主发现云端,并且注册,边端使用python实现,借助这个机会积累一下python的socket编程相关的内容,并且总结服务发现的解决方案,注意,这个方案只适合内网中使用。

知识积累

#创建Tcp/Ip套接字
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) #流式Socket
#创建Udp/Ip套接字
s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM) #数据报式Socket
socket.AF_UNIX  #只能够用于单一的Unix系统进程间通信
socket.AF_INET6  #只能够用于IPv6通信
socket.SOCK_RAW  #原始套接字,可以处理ICMP、ARP等网络报文,其它的不行
socket.SOCK_SEQPACKET  #可靠的连续数据包服务      

具体的api的使用,参考资料的第三篇文章,比较详细,不再粘贴复制了。

服务发现解决方案

import socket

s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)

PORT = 1060

s.bind(('', PORT))

print('Listening for broadcast at ', s.getsockname())

while True:
    data, address = s.recvfrom(65535)

    print('Server received from {}:{}'.format(address, data.decode('utf-8')))
    break      
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)

PORT = 1060

network = '255.255.255.255'

s.sendto('Client broadcast message!'.encode('utf-8'), (network, PORT))      

判断socket连接状态解决方案

import socket
 
def isNetChainOK(testserver=('www.baidu.com', 443)):
    so = socket.socket()
    so.settimeout(3)
    try:
        status = so.connect_ex(testserver)
        if status == 0:
            so.close()
            print("网络正常!")
        else:
            print("网络异常,请检查网络!")
    except Exception:
        print("网络异常,请检查网络!")
 
if __name__ == '__main__':
    isNetChainOK()