python websocket例程_python 实现websocket

#!/usr/bin python#-*- coding:UTF-8 -*-

importredisimporttime, threading, sched, json, socket, base64, hashlib,logging,traceback

allkv_json= "" #发送给前端的数据

conn_list = [] #处于在线的socket链接

#单例

defsingleton(cls):

instances={}def getinstatce(*args, **kwargs):if cls not ininstances:

instances[cls]= cls(*args, **kwargs)returninstances[cls]returngetinstatce

@singletonclassRedisUtils:def __init__(self, port, db):

self.redis_object= redis.Redis('127.0.0.1', port, db)

self.allKV=[]defget_all_kv(self):

keys=self.redis_object.keys()

self.allKV= [key for key in keys if self.redis_object.ttl(key) is notNone]#for key in keys:

#self.allKV[key] = 'online' if (time.time() - int(self.redis_object.get(key))) < 10 else 'off_line'

classSchedTask:

allkv=[]

@classmethoddefgetHostOnlineStatus(cls):globalallkv_json

ru= RedisUtils(6379, 1)

ru.get_all_kv()

cls.allkv=ru.allKV

allkv_json=json.dumps(cls.allkv)printallkv_json

@classmethoddefgetHostOnlineStatusTask(cls):

threading.Thread(target=cls.getHostOnlineStatus).start()

@classmethoddefrun(cls, timedelay):whileTrue:

s=sched.scheduler(time.time, time.sleep)

s.enter(timedelay,1, cls.getHostOnlineStatusTask, ())

s.run()classWebsocketUtils(threading.Thread):

MAGIC_STRING= '258EAFA5-E914-47DA-95CA-C5AB0DC85B11'HANDSHAKE_STRING= "HTTP/1.1 101 Switching Protocols\r\n"\"Upgrade:WebSocket\r\n"\"Connection: Upgrade\r\n"\"Sec-WebSocket-Accept: {1}\r\n"\"WebSocket-Location: ws://{2}/chat\r\n"\"WebSocket-Protocol:chat\r\n\r\n"

def __init__(self, host, port):if notisinstance(host, str):raise KeyError("The host must be a string like \'127.0.0.1\'")else:

self.host=hostif notisinstance(port, int):raise KeyError('The port must be a integer')else:

self.port=porttry:

self.sock=socket.socket(socket.AF_INET, socket.SOCK_STREAM)

self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,1)

self.sock.bind((host, port))print '========================================================================================='

printhost,port

self.sock.listen(100)except:printtraceback.format_exc()print 'start socket error'super(WebsocketUtils, self).__init__()#前端握手

defhandshake(self, conn):

headers={}

shake= conn.recv(1024)printshakeif notlen(shake):print('len error')returnFalse

header, data= shake.split('\r\n\r\n', 1)for line in header.split('\r\n')[1:]:

key, value= line.split(':', 1)

headers[key]=valueif 'Sec-WebSocket-Key' not inheaders:print('this is not websocket, client close.')printheaders

conn.close()returnFalse

sec_key= headers['Sec-WebSocket-Key']

res_key= base64.b64encode(hashlib.sha1(sec_key +WebsocketUtils.MAGIC_STRING).digest())

str_handshke= WebsocketUtils.HANDSHAKE_STRING.replace('{1}', res_key).replace('{2}',

self.host+ ":" +str(self.port))printstr_handshke

conn.send(str_handshke)

i=0defsend_msg(self):

WebsocketUtils.i+=1

globalconn_listprint 'send msg' +str(WebsocketUtils.i)for conn inconn_list:try:

conn.send('%c%c%s' % (0x81, len(allkv_json), allkv_json))except:print 'send msg error'conn.close()

conn_list.remove(conn)#定时给前端发信息

defsched_send_msg(self):whileTrue:

s=sched.scheduler(time.time,time.sleep)

s.enter(2,1,self.send_msg,())

s.run()defrun(self):#另开一个线程给各个conn发消息

globalconn_list

t=threading.Thread(target=self.sched_send_msg)

t.start()whileTrue:print 'wait link'

try:printself.sock

self.conn, addr=self.sock.accept()print 'link ok'self.handshake(self.conn)print 'handshake ok'conn_list.append(self.conn)except:printtraceback.format_exc()print 'error'time.sleep(3)if __name__ == '__main__':#websocket线程

websocket_utils = WebsocketUtils('0.0.0.0', 9000)#websocket_utils.setDaemon(True) # 把当前进程设置为守护进程，主线程执行完毕，子线程均停止

websocket_utils.start()#定时获取主机在线信息

SchedTask.run(5)