concurrence examples
#server
#chat.py
from concurrence import dispatch, Tasklet, Message
from concurrence.io import BufferedStream, Socket, Server
class MSG_WRITE_LINE(Message): pass
class MSG_QUIT(Message): pass
class MSG_LINE_READ(Message): pass
connected_clients = set() #set of currently connected clients (tasks)
def handle(client_socket):
"""handles a single client connected to the chat server"""
stream = BufferedStream(client_socket)
client_task = Tasklet.current() #this is the current task as started by server
connected_clients.add(client_task)
def writer():
for msg, args, kwargs in Tasklet.receive():
if msg.match(MSG_WRITE_LINE):
stream.writer.write_bytes(args[0] + '\n')
stream.writer.flush()
def reader():
for line in stream.reader.read_lines():
line = line.strip()
if line == 'quit':
MSG_QUIT.send(client_task)()
else:
MSG_LINE_READ.send(client_task)(line)
reader_task = Tasklet.new(reader)()
writer_task = Tasklet.new(writer)()
MSG_WRITE_LINE.send(writer_task)("type 'quit' to exit..")
for msg, args, kwargs in Tasklet.receive():
print args[0]
if msg.match(MSG_QUIT):
break
elif msg.match(MSG_LINE_READ):
#a line was recv from our client, multicast it to the other clients
for task in connected_clients:
if task != client_task: #don't echo the line back to myself
MSG_WRITE_LINE.send(task)(args[0])
elif msg.match(MSG_WRITE_LINE):
MSG_WRITE_LINE.send(writer_task)(args[0])
connected_clients.remove(client_task)
reader_task.kill()
writer_task.kill()
client_socket.close()
def server():
"""accepts connections on a socket, and dispatches
new tasks for handling the incoming requests"""
print 'listening for connections on port 9010'
Server.serve(('localhost', 9010), handle)
if __name__ == '__main__':
dispatch(server)
#client
#client.py
# coding: UTF8
from concurrence.core import dispatch, Message, Tasklet
from concurrence.io.buffered import BufferedStream
from concurrence.io.socket import Socket
class MSG_SEND(Message): pass
class MSG_QUIT(Message): pass
class MSG_LINE_READ(Message): pass
def client():
client_task = Tasklet.current()
client_id = id(client_task)
sock = Socket.connect(("localhost", 9010), 5)
stream = BufferedStream(sock)
def writer():
for msg, args, _ in Tasklet.receive():
try:
if msg.match(MSG_SEND):
data = args[0]
print "SEND:", data
stream.writer.write_bytes(data + '\n')
stream.writer.flush()
except Exception, e:
print "WREX:", e
break
MSG_QUIT.send(client_task)()
def reader():
while True:
try:
for line in stream.reader.read_lines():
line = line.strip()
MSG_LINE_READ.send(client_task)(line)
except Exception, e:
print "RDEX:", e
break
MSG_QUIT.send(client_task)()
reader_task = Tasklet.new(reader)()
writer_task = Tasklet.new(writer)()
def send(data):
MSG_SEND.send(client_task)(data)
def process(data):
print "RECV:", data
def play_game():
while True:
send("hello")
Tasklet.sleep(5)
Tasklet.later(3, play_game)()
for msg, args, _ in Tasklet.receive():
if msg.match(MSG_LINE_READ):
process(args[0])
elif msg.match(MSG_SEND):
MSG_SEND.send(writer_task)(args[0])
print "close"
reader_task.kill()
writer_task.kill()
sock.close()
if __name__ == '__main__':
dispatch(client)