106 lines
2.7 KiB
Python
106 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2013 Mirantis, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import logging
|
|
import six
|
|
import functools
|
|
|
|
import amqp.exceptions as amqp_exceptions
|
|
from kombu import Connection
|
|
from kombu import Exchange
|
|
from oslo_serialization import jsonutils
|
|
from kombu import Queue
|
|
|
|
from nailgun.settings import settings
|
|
from nailgun.rpc import utils
|
|
|
|
|
|
logger = logging.getLogger('receiverd')
|
|
|
|
|
|
creds = (
|
|
("userid", "guest"),
|
|
("password", "guest"),
|
|
("hostname", "localhost"),
|
|
("port", "5672"),
|
|
)
|
|
|
|
conn_str = 'amqp://{0}:{1}@{2}:{3}//'.format(
|
|
*[settings.RABBITMQ.get(*cred) for cred in creds]
|
|
)
|
|
|
|
naily_exchange = Exchange(
|
|
'naily',
|
|
'topic',
|
|
durable=True
|
|
)
|
|
|
|
naily_queue = Queue(
|
|
'naily',
|
|
exchange=naily_exchange,
|
|
routing_key='naily'
|
|
)
|
|
|
|
naily_service_exchange = Exchange(
|
|
'naily_service',
|
|
'fanout',
|
|
durable=False,
|
|
auto_delete=True
|
|
)
|
|
|
|
naily_service_queue = Queue(
|
|
'naily_service',
|
|
exchange=naily_service_exchange
|
|
)
|
|
|
|
nailgun_exchange = Exchange(
|
|
'nailgun',
|
|
'topic',
|
|
durable=True
|
|
)
|
|
|
|
nailgun_queue = Queue(
|
|
'nailgun',
|
|
exchange=nailgun_exchange,
|
|
routing_key='nailgun'
|
|
)
|
|
|
|
|
|
def cast(name, message, service=False):
|
|
logger.debug(
|
|
"RPC cast to orchestrator:\n{0}".format(
|
|
jsonutils.dumps(message, indent=4)
|
|
)
|
|
)
|
|
use_queue = naily_queue if not service else naily_service_queue
|
|
use_exchange = naily_exchange if not service else naily_service_exchange
|
|
with Connection(conn_str) as conn:
|
|
with conn.Producer(serializer='json', compression='gzip') as producer:
|
|
publish = functools.partial(
|
|
producer.publish, message,
|
|
exchange=use_exchange, routing_key=name, declare=[use_queue]
|
|
)
|
|
try:
|
|
publish()
|
|
except amqp_exceptions.PreconditionFailed as e:
|
|
logger.warning(six.text_type(e))
|
|
# (dshulyak) we should drop both exchanges/queues in order
|
|
# for astute to be able to recover temporary queues
|
|
utils.delete_entities(
|
|
conn, naily_service_exchange, naily_service_queue,
|
|
naily_exchange, naily_queue)
|
|
publish()
|