pythonマルチスレッド生産者消費者

1781 ワード

import threading
import time
import logging
import random
import Queue

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

BUF_SIZE = 10
queue = Queue.Queue(BUF_SIZE)

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

    def run(self):
        while True:
            if not self.queue.full():
                item = random.randint(1,10)
                self.queue.put(item)
                logging.debug('Putting ' + str(item)  
                              + ' : ' + str(self.queue.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, queue):
        super(ConsumerThread,self).__init__()
        self.target = target
        self.name = name
        self.queue = queue
        return

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

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

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