kind-rabbit/task_creator.py at master · mgdenno/kind-rabbit · 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
59
import sys
import json
import pika

messages = [
    {
        "simulationId": 1,
        "runTimeMs": 1000
    },
    {
        "simulationId": 2,
        "runTimeMs": 1000
    },
    {
        "simulationId": 3,
        "runTimeMs": 10000
    },
    {
        "simulationId": 4,
        "runTimeMs": 10000
    },
    {
        "simulationId": 5,
        "runTimeMs": 5000
    },
    {
        "simulationId": 6,
        "runTimeMs": 5000
    },
    {
        "simulationId": 7,
        "runTimeMs": 1000
    },
    {
        "simulationId": 8,
        "runTimeMs": 1000
    }
]

connection = pika.BlockingConnection(
    pika.ConnectionParameters(
        host='localhost',
        credentials=pika.PlainCredentials("admin", "queue")
    )
)
channel = connection.channel()

for message in messages:
    channel.queue_declare(queue='task_queue', durable=True)
    channel.basic_publish(
        exchange='',
        routing_key='task_queue',
        body=json.dumps(message),
        properties=pika.BasicProperties(
            delivery_mode=pika.spec.PERSISTENT_DELIVERY_MODE
        ))
    print(f"[x] Sent simulationId: {message['simulationId']}")

connection.close()