← Back to branch summary

~cprov/+junk/adt-proxy-service

« back to all changes in this revision

Viewing changes to adt-proxy-service.py

  • Committer: Celso Providelo
  • Date: 2015-03-03 01:12:04 UTC
  • Revision ID: celso.providelo@canonical.com-20150303011204-sbfa8s62vjp3x10z
Adjust according lp:adt-cloud-worker

Show diffs side-by-side

added added

removed removed

Lines of Context:
adt-proxy-service.py
30
30
 
31
31
    args = parser.parse_args()
32
32
    routing_key = args.tag
 
33
    n_requests = args.requests
33
34
 
34
35
    # Load configuration options.
35
36
    config = configparser.ConfigParser()
36
37
    config.read(args.conf)
37
38
    amqp_uris = config.get('amqp', 'uris').split()
38
39
 
 
40
 
 
41
    # Bogus script based producer ...
 
42
    arch, platform = routing_key.split('.')
 
43
    payload = {
 
44
        'request_id': kombu.uuid(),
 
45
        'arch': arch,
 
46
        'platform': platform,
 
47
        'package_name': 'libpng',
 
48
        'nova_flavor': 'm1.medium',
 
49
        'nova_image': ('ubuntu-released/ubuntu-trusty-14.04-{}-server-' 
 
50
                       '20150209.1-disk1.img'.format(arch)),
 
51
    }
 
52
 
39
53
    adt_exchange = kombu.Exchange("adt.exchange", type="topic")
40
54
    queue = kombu.Queue(
41
55
        'adt.requests.{}'.format(routing_key),
45
59
    # available works the request is preserved.
46
60
    declare = [adt_exchange, queue]
47
61
 
48
 
    # Bogus script based producer ...
49
 
    arch, platform = routing_key.split('.')
50
 
    payload = {
51
 
        'arch': arch,
52
 
        'platform': platform,
53
 
    }
54
 
 
55
62
    with kombu.Connection(amqp_uris) as conn:
56
63
        with producers[conn].acquire(block=True) as producer:
57
64
 
63
70
                producer, producer.publish,
64
71
                errback=errback, max_retries=3)
65
72
 
66
 
            for i in range(args.requests):
 
73
            for i in range(n_requests):
67
74
                payload['index'] = i
68
75
                publish(
69
76
                    payload,