# _*_ coding: utf-8 _*_

"""
python_coroutine.py by xianhu
"""

import asyncio
import aiohttp
import threading


# 生产者、消费者例子
def consumer():         # 定义消费者，由于有yeild关键词，此消费者为一个生成器
    print("[Consumer] Init Consumer ......")
    r = "init ok"       # 初始化返回结果，并在启动消费者时，返回给生产者
    while True:
        n = yield r     # 消费者通过yield关键词接收生产者产生的消息，同时返回结果给生产者
        print(f"[Consumer] conusme n = {n}, r = {r}")
        r = f"consume {n} OK"


def produce(c):         # 定义生产者，此时的 c 为一个生成器
    print("[Producer] Init Producer ......")
    r = c.send(None)    # 启动消费者生成器，同时第一次接收返回结果
    print(f"[Producer] Start Consumer, return {r}")
    for n in range(1, 6):
        print(f"[Producer] While, Producing {n} ......")
        r = c.send(n)   # 向消费者发送消息，同时准备接收结果。此时会切换到消费者执行
        print(f"[Producer] Consumer return: {r}")
    c.close()           # 关闭消费者生成器
    print("[Producer] Close Producer ......")

# produce(consumer())


# 异步IO例子：适配Python3.4，使用asyncio库
@asyncio.coroutine
def hello(index):                   # 通过装饰器asyncio.coroutine定义协程
    print(f'Hello world! index={index}, thread={threading.currentThread()}')
    yield from asyncio.sleep(1)     # 模拟IO任务
    print(
        f'Hello again! index={index}, thread={threading.currentThread()}'
    ) @ asyncio.coroutine

loop = asyncio.get_event_loop()     # 得到一个事件循环模型
tasks = [hello(1), hello(2)]        # 初始化任务列表
loop.run_until_complete(asyncio.wait(tasks))    # 执行任务
loop.close()                        # 关闭事件循环列表


# 异步IO例子：适配Python3.5，使用async和await关键字
async def hello1(index):            # 通过关键字async定义协程
    print(f'Hello world! index={index}, thread={threading.currentThread()}')
    await asyncio.sleep(1)          # 模拟IO任务
    print(f'Hello again! index={index}, thread={threading.currentThread()}')

loop = asyncio.get_event_loop()     # 得到一个事件循环模型
tasks = [hello1(1), hello1(2)]      # 初始化任务列表
loop.run_until_complete(asyncio.wait(tasks))    # 执行任务
loop.close()                        # 关闭事件循环列表


# aiohttp 实例
async def get(url):
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as resp:
            print(url, resp.status)
            print(url, await resp.text())

loop = asyncio.get_event_loop()     # 得到一个事件循环模型
tasks = [                           # 初始化任务列表
    get("http://zhushou.360.cn/detail/index/soft_id/3283370"),
    get("http://zhushou.360.cn/detail/index/soft_id/3264775"),
    get("http://zhushou.360.cn/detail/index/soft_id/705490")
]
loop.run_until_complete(asyncio.wait(tasks))    # 执行任务
loop.close()                        # 关闭事件循环列表