aboutsummaryrefslogtreecommitdiffstats
path: root/moonv4/moon_orchestrator/moon_orchestrator/security_function.py
blob: 1b33ef60aa0090bd6a51b5545e0308e38d7bff0b (plain)
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
# 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 os
import time
from oslo_config import cfg
from oslo_log import log as logging
import oslo_messaging
from moon_orchestrator.dockers import DockerBase

LOG = logging.getLogger(__name__)
CONF = cfg.CONF
DOMAIN = "moon_orchestrator"

__CWD__ = os.path.dirname(os.path.abspath(__file__))
# TODO (dthom): select the right template folder
TEMPLATES_FOLDER = os.path.join(__CWD__, "..", "conf", "dockers")


class SecurityFunction(DockerBase):

    id = "moon_function"
    __build = """RUN mkdir -p /etc/moon/
COPY conf /etc/moon/
ADD dist/{py_pkg}.tar.gz /root
WORKDIR /root/{py_pkg}
RUN pip3 install -r requirements.txt
RUN pip3 install .
"""

    def __init__(self, intra_extension_id, conf_file="", docker=None, network_config=None):
        self.id = "moon_pdp_{}".format(intra_extension_id)
        super(SecurityFunction, self).__init__(
            name="moon_secfunction",
            run_cmd=["python3", "-m", "moon_secfunction", intra_extension_id],
            conf_file=conf_file,
            docker=docker,
            network_config=network_config,
            build_cmd=self.__build,
            id=self.id,
            tag=CONF.security_function.container
        )
        # note(asteroide): time to let the new docker boot
        time.sleep(3)
        self.get_status()

    def get_status(self):
        transport = oslo_messaging.get_transport(CONF)
        target = oslo_messaging.Target(topic=self.id, version='1.0')
        client = oslo_messaging.RPCClient(transport, target)
        LOG.info("Calling Status on {}".format(self.id))
        ret = client.call({"component_id": self.id}, 'get_status', args=None)
        LOG.info(ret)
        return ret