MULTITHREADING - PRODUCER AND CONSUMER WITH QUEUE

最新推荐文章于 2024-04-03 18:04:12 发布

seamanj

最新推荐文章于 2024-04-03 18:04:12 发布

阅读量390

点赞数

分类专栏： python

python 专栏收录该内容

24 篇文章

订阅专栏

本文通过Python的Queue模块实现了一个生产者消费者模型。在这个模型中，生产者线程不断地生成随机数并放入队列，而消费者线程则不断地从队列中取出这些数值。该模型展示了如何有效地在多线程环境中同步数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Queue

In this chapter, we'll implement another version of Producer and Consumer code with Queue(see Condition objects with producer and consumer).

In the following example, the Consumer and Producer threads runs indefinitely while checking the status of the queue. The Producer thread is responsible for putting items into the queue if it is not full while the Consumer thread consumes items if there are any.

import threading
import time
import logging
import random
import Queue

logging.basicConfig(level=logging.DEBUG,
                    format='(%(threadName)-9s) %(message)s',)

BUF_SIZE = 10
q = Queue.Queue(BUF_SIZE)

class ProducerThread(threading.Thread):
    def __init__(self, group=None, target=None, name=None,
                 args=(), kwargs=None, verbose=None):
        super(ProducerThread,self).__init__()
        self.target = target
        self.name = name

    def run(self):
        while True:
            if not q.full():
                item = random.randint(1,10)
                q.put(item)
                logging.debug('Putting ' + str(item)  
                              + ' : ' + str(q.qsize()) + ' items in queue')
                time.sleep(random.random())
        return

class ConsumerThread(threading.Thread):
    def __init__(self, group=None, target=None, name=None,
                 args=(), kwargs=None, verbose=None):
        super(ConsumerThread,self).__init__()
        self.target = target
        self.name = name
        return

    def run(self):
        while True:
            if not q.empty():
                item = q.get()
                logging.debug('Getting ' + str(item) 
                              + ' : ' + str(q.qsize()) + ' items in queue')
                time.sleep(random.random())
        return

if __name__ == '__main__':
    
    p = ProducerThread(name='producer')
    c = ConsumerThread(name='consumer')

    p.start()
    time.sleep(2)
    c.start()
    time.sleep(2)

Output:(producer ) Putting 2 : 1 items in queue(producer ) Putting 10 : 2 items in queue(producer ) Putting 6 : 3 items in queue(producer ) Putting 7 : 4 items in queue(producer ) Putting 1 : 5 items in queue(consumer ) Getting 2 : 4 items in queue(consumer ) Getting 10 : 3 items in queue(producer ) Putting 1 : 4 items in queue(producer ) Putting 8 : 5 items in queue(consumer ) Getting 6 : 4 items in queue(producer ) Putting 10 : 5 items in queue...