-
Notifications
You must be signed in to change notification settings - Fork 0
/
aqueue.py
45 lines (37 loc) · 1.03 KB
/
aqueue.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""
IF:
asyncio ImportError: cannot import name 'Full'
THEN:
Check if your project or current working directory includes a file called queue.py and rename it.
"""
import asyncio
import logging
logging.basicConfig(level=logging.DEBUG)
logging.getLogger('asyncio').setLevel(logging.DEBUG)
async def produce(queue, num):
for i in range(num):
await queue.put(i) # 队列中加入
await asyncio.sleep(1)
await queue.put(None)
async def consume(queue):
while True:
res = await queue.get() # 取数据
if res is None:
break
print('consume:{}'.format(res))
def main():
loop = asyncio.get_event_loop()
loop.set_debug(True)
queue = asyncio.Queue(loop=loop)
try:
loop.run_until_complete(asyncio.gather(
produce(queue, 5),
consume(queue)
))
except Exception as e:
print(str(e))
finally:
loop.stop()
loop.close()
if __name__ == '__main__':
main()