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
60
61
62
63
|
# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors
# This software is distributed under the terms and conditions of the 'Apache-2.0'
# license which can be found in the file 'LICENSE' in this package distribution
# or at 'http://www.apache.org/licenses/LICENSE-2.0'.
import oslo_messaging
from oslo_log import log as logging
import time
from moon_utilities.api import APIList
from moon_utilities import configuration
from oslo_config import cfg
from moon_orchestrator.api.generic import Status, Logs
from moon_orchestrator.api.containers import Containers
from moon_orchestrator.api.slaves import Slaves
TOPIC = "orchestrator"
LOG = logging.getLogger("moon.orchestrator.messenger")
CONF = cfg.CONF
class Server:
def __init__(self, containers, docker_manager, slaves):
cfg.CONF.transport_url = self.__get_transport_url()
self.CONTAINERS = containers
self.transport = oslo_messaging.get_transport(cfg.CONF)
self.target = oslo_messaging.Target(topic=TOPIC, server='server1')
LOG.info("Starting MQ server with topic: {}".format(TOPIC))
self.docker_manager = docker_manager
for _container in containers:
Status._container = containers[_container]
self.endpoints = [
APIList((Status, Logs, Containers)),
Status(),
Logs(),
Containers(self.docker_manager),
Slaves(slaves)
]
self.server = oslo_messaging.get_rpc_server(self.transport, self.target, self.endpoints,
executor='threading',
access_policy=oslo_messaging.DefaultRPCAccessPolicy)
@staticmethod
def __get_transport_url():
messenger = configuration.get_configuration(configuration.MESSENGER)["messenger"]
return messenger['url']
def run(self):
try:
self.server.start()
while True:
time.sleep(1)
except KeyboardInterrupt:
LOG.warning("Stopping server by crtl+c (please be patient, closing connections...)")
except SystemExit:
LOG.warning("Stopping server (please be patient, closing connections...)")
except Exception as e:
LOG.error("Exception occurred: {}".format(e))
self.server.stop()
self.server.wait()
|