python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > 生产消费者模型中task_done()的作用

关于生产消费者模型中task_done()的具体作用

作者:MicoJack Honey

这篇文章主要介绍了关于生产消费者模型中task_done()的具体作用,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

直奔主题

tase_done()的作用:

只有消费者把队列所有的数据处理完毕,queue.join()才会停止阻塞

代码解释

#关于tase_done()的作用:只有消费者把队列所有的数据处理完毕,queue.join()才会停止阻塞
import time
from queue import Queue
from threading import Thread
q = Queue()

def produce():
    for i in range(10):
        q.put(i)
        print('生产:',i)
    print('生产任务完毕!')
    q.join()
    print(produce.__name__,'函数结束!')

def consumer():
    for i in range(10):
        print('消费:', q.get())
        q.task_done()
        # if i == 4:
        #     print('休息1s...')
        #     time.sleep(1)#sleep作用:查看生产者是否阻塞
    print(consumer.__name__,'函数结束!')



pro = Thread(target=produce)
con = Thread(target=consumer)

pro.start()
con.start()

con.join()
print('消费者任务完成')
pro.join()
print('生产者任务完成')

生产: 0
生产: 1
生产: 2
生产: 3
生产: 4
生产: 5
消费: 0
消费: 1
消费: 2
消费: 3
消费: 4
消费: 5
生产: 6
生产: 7
生产:8
消费: 6
消费: 7
消费: 8
生产: 9
生产任务完毕!
消费: 9
consumer 函数结束!
produce 函数结束!
消费者任务完成
生产者任务完成

说明:生产任务已经完毕为什么没有直接produce函数结束?

说明这里产生了阻塞,产生阻塞的原因是因为消费者没有处理完生产者所有任务。

tase_done()的作用:只有消费者把队列所有的数据处理完毕,queue.join()才会停止阻塞所以才会有produce函数结束!然后生产任务结束。

对比解释看效果

注释掉q.task_done()和q.join()

#关于tase_done()的作用:只有消费者把队列所有的数据处理完毕,queue.join()才会停止阻塞
import time
from queue import Queue
from threading import Thread
q = Queue()

def produce():
    for i in range(10):
        q.put(i)
    print('生产任务完毕!')
    # q.join()
    # if i == 4:
        #     print('休息1s...')
        #     time.sleep(1)#sleep作用:查看生产者是否阻塞
    print(produce.__name__,'函数结束!')

def consumer():
    for i in range(10):
        print('消费:', q.get())
        # q.task_done()
    print(consumer.__name__,'函数结束!')



pro = Thread(target=produce)
con = Thread(target=consumer)

pro.start()
con.start()

con.join()
print('消费者任务完成')
pro.join()
print('生产者任务完成')

生产: 0
生产: 1
生产: 2
生产: 3
生产: 4
生产: 5
生产: 6
生产: 7
生产: 8
生产: 9
消费: 0
消费: 1
生产任务完毕!
produce 函数结束!
消费: 2
消费: 3
消费: 4
消费: 5
消费: 6
消费: 7
消费: 8
消费: 9
consumer 函数结束!
消费者任务完成
生产者任务完成

说明:不带task_done()和join()情况下生产任务结束,produce函数立刻结束,并没有等待消费者处理完所有任务立即结束。

仅注释掉q.task_done()**

#关于tase_done()的作用:只有消费者把队列所有的数据处理完毕,queue.join()才会停止阻塞
import time
from queue import Queue
from threading import Thread
q = Queue()

def produce():
    for i in range(10):
        q.put(i)
    print('生产任务完毕!')
    q.join()
    print(produce.__name__,'函数结束!')

def consumer():
    for i in range(10):
        print('消费:', q.get())
        # q.task_done()
        # if i == 4:
        #     print('休息1s...')
        #     time.sleep(1)#sleep作用:查看生产者是否阻塞
    print(consumer.__name__,'函数结束!')



pro = Thread(target=produce)
con = Thread(target=consumer)

pro.start()
con.start()

con.join()
print('消费者任务完成')
pro.join()
print('生产者任务完成')

生产: 0
生产: 1
生产: 2
生产: 3
生产: 4
生产: 5
生产: 6
生产: 7
生产: 8
生产: 9
生产任务完毕!
消费: 0
消费: 1
消费: 2
消费: 3
消费: 4
消费: 5
消费: 6
消费: 7
消费: 8
消费: 9
consumer 函数结束!
消费者任务完成
。。。阻塞。。。

说明:消费者已经把任务完成了,但是produce函数却一直阻塞没有结束。

总结

通过以上分析可以得出结论:

task_done()的作用:消费者处理一个任务,它就通知生产者我处理了一个任务。

为什么要告诉生产者消费者处理了一个任务呢?

这也是为什么我进行注释对比的原因

生产者如果被消费者的task_done()告知没有把生产者交给的任务处理完毕,那么生产者就会一直阻塞(生气…)。

大家可以把sleep()函数的注释删掉看看,并且设置任意一个sleep()时间观察现象。

如果只有q.join()而没有task_done(),消费者处理完所有任务,生产者任然阻塞,所以q.join()一般和q.task_done()配合使用。

最后

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文