Python中協(xié)程用法代碼詳解

import time
def A():
  while True:
    print("----我是A函數(shù)---")
    yield
    time.sleep(0.5)
def B(c):
  while True:
    print("----我是B函數(shù)---")
    next(c)
    time.sleep(0.5)
if __name__ == '__main__':
  a = A()
  B(a)

# 結(jié)果如下：
# ----我是B函數(shù)---
# ----我是A函數(shù)---
# ----我是B函數(shù)---
# ----我是A函數(shù)---
# ----我是B函數(shù)---
# ----我是A函數(shù)---
# ----我是B函數(shù)---
# ----我是A函數(shù)---
# ----我是B函數(shù)---
# ----我是A函數(shù)---
# ......

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : xiaoke
from greenlet import greenlet
import time


def test1():
  while True:
    print("---我是A函數(shù)--")
    gr2.switch()
    time.sleep(0.5)


def test2():
  while True:
    print("---我是B函數(shù)--")
    gr1.switch()
    time.sleep(0.5)


def main():
  # 切換到gr1中運行
  gr1.switch()


if __name__ == '__main__':
  gr1 = greenlet(test1)
  gr2 = greenlet(test2)
  main()

# 結(jié)果如下：
# ---我是A函數(shù)--
# ---我是B函數(shù)--
# ---我是A函數(shù)--
# ---我是B函數(shù)--
# ---我是A函數(shù)--
# ---我是B函數(shù)--
# ---我是A函數(shù)--
# ---我是B函數(shù)--
# ......

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : xiaoke
import gevent


def task1(n):
  for i in range(n):
    print('----task1-----')

    gevent.sleep(1)
    # time.sleep(1) # time.sleep沒有讓gevent感知到等待


def task2(n):
  for i in range(n):
    print('----task2-----')

    gevent.sleep(1)
    # time.sleep(1)


def main():
  g1 = gevent.spawn(task1, 5)
  g2 = gevent.spawn(task2, 5)

  g1.join()
  g2.join()


if __name__ == "__main__":
  main()


# 結(jié)果如下：
# ----task1-----
# ----task2-----
# ----task1-----
# ----task2-----
# ----task1-----
# ----task2-----
# ----task1-----
# ----task2-----
# ----task1-----
# ----task2-----

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : xiaoke

import urllib.request # py3

import gevent
from gevent import monkey
# 猴子補丁，將標(biāo)準(zhǔn)庫的涉及IO操作方法替換成gevent
monkey.patch_all() 

# 協(xié)程的任務(wù)函數(shù)
def my_download(url):
  print('GET %s' % url)

  response = urllib.request.urlopen(url)
  data = response.read()

  print('下載 %d bytes from %s' % (len(data), url))


def main():
  g1 = gevent.spawn(my_download, 'http://www.google.cn')
  g2 = gevent.spawn(my_download, 'http://www.qq.com')
  g3 = gevent.spawn(my_download, 'http://www.baidu.com')

  gevent.joinall([g1, g2, g3]) # 等待指定的協(xié)程結(jié)束


if __name__ == "__main__":
  main()

# 結(jié)果如下：
# GET http://www.google.cn
# GET http://www.qq.com
# GET http://www.baidu.com
# 下載 102221 bytes from http://www.baidu.com
# 下載 52297 bytes from http://www.qq.com
# 下載 3213 bytes from http://www.google.cn

#從上能夠看到是先獲取baidu的相關(guān)信息，然后依次是qq
#google，但是收到數(shù)據(jù)的先后順序不一定與發(fā)送順序相同，
#這也就體現(xiàn)出了異步，即不確定什么時候會收到數(shù)據(jù)，順序不一定.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : xiaoke
import socket
import gevent
from gevent import monkey
#猴子補丁，將標(biāo)準(zhǔn)庫的涉及IO操作方法替換成gevent
monkey.patch_all()


# 需要為客戶端提供服務(wù)
def do_service(connect_socket):
  while True:
    # tcp recv() 只會返回接收到的數(shù)據(jù)
    recv_data = connect_socket.recv(1024)

    # if recv_data == b'':
    if len(recv_data) == 0:
      # 發(fā)送方關(guān)閉tcp的連接,recv()不會阻塞，而是直接返回''
      # print('client %s close' % str(client_addr))
      # s.getpeername()  s.getsockname()
      print('client %s close' % str(connect_socket.getpeername()))
      break
    print('recv: %s' % recv_data.decode('gbk'))

def main():

  listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # 設(shè)置允許復(fù)用地址,當(dāng)建立連接之后服務(wù)器先關(guān)閉，設(shè)置地址復(fù)用
  listen_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

  my_addr = ('192.168.105.125', 8080)
  listen_socket.bind(my_addr)

  listen_socket.listen(5) # 設(shè)置套接字成監(jiān)聽,5表示一個己連接隊列長度
  print('listening...')

  while True:
    # 接受連接請求，創(chuàng)建連接套接字，用于客戶端連通信
    connect_socket, client_addr = listen_socket.accept() # accept默認會引起阻塞
    # 新創(chuàng)建連接用的socket, 客戶端的地址
    # print(connect_socket)
    print(client_addr)

    # 每當(dāng)來新的客戶端連接，創(chuàng)建協(xié)程，由協(xié)程和客戶端通信
    coroutine_do_service = gevent.spawn(do_service, connect_socket)


if __name__ == "__main__":
  main()