distributed-tasks/producer_consumer_queue.py at master · tj-experimental/distributed-tasks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import threading
import queue

import time
import random

MAX_ITEMS = 10
QUEUE = queue.Queue(MAX_ITEMS)


class ProducerThread(threading.Thread):

    def run(self):
        numbers = range(MAX_ITEMS)
        global QUEUE

        while True:
            number = random.choice(numbers)
            QUEUE.put(number)
            print(f"Produced {number}")
            time.sleep(random.random())


class ConsumerThread(threading.Thread):

    def run(self):
        global QUEUE
        while True:
            number = QUEUE.get()
            QUEUE.task_done()
            print(f"Consumer 1: Consumed number {number}")
            time.sleep(random.random())


class Consumer2Thread(threading.Thread):

    def run(self):
        global QUEUE
        while True:
            number = QUEUE.get()
            QUEUE.task_done()
            print(f"Consumer 2: Consumed number {number}")
            time.sleep(random.random())


def main(*args):
    producer = ProducerThread()
    producer.start()

    consumer1 = ConsumerThread()
    consumer1.start()

    consumer2 = Consumer2Thread()
    consumer2.start()


if __name__ == '__main__':
    main()