X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=src%2Fnepi%2Fresources%2Fomf%2Fchannel.py;h=2a44b0d98853c6aa6e1987163f785bd38dbd11f0;hb=e91aa23d2c2b31dbe31d12c6ba26edf8db4f977e;hp=fa8df03b9ee1aa2636d65c05baac8667d984fcdd;hpb=fc0434af3f42156b268e0e201f6b2083385c9dc5;p=nepi.git diff --git a/src/nepi/resources/omf/channel.py b/src/nepi/resources/omf/channel.py index fa8df03b..2a44b0d9 100644 --- a/src/nepi/resources/omf/channel.py +++ b/src/nepi/resources/omf/channel.py @@ -1,31 +1,33 @@ -""" - NEPI, a framework to manage network experiments - Copyright (C) 2013 INRIA - - This program is free software: you can redistribute it and/or modify - it under the terms of the GNU General Public License as published by - the Free Software Foundation, either version 3 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. - - You should have received a copy of the GNU General Public License - along with this program. If not, see . - -""" - -from nepi.execution.resource import ResourceManager, clsinit, ResourceState +# +# NEPI, a framework to manage network experiments +# Copyright (C) 2013 INRIA +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . +# +# Author: Alina Quereilhac +# Julien Tribino + +from nepi.execution.resource import ResourceManager, clsinit_copy, ResourceState, \ + reschedule_delay from nepi.execution.attribute import Attribute, Flags +from nepi.resources.omf.omf_resource import ResourceGateway, OMFResource from nepi.resources.omf.omf_api import OMFAPIFactory -reschedule_delay = "0.5s" -@clsinit -class OMFChannel(ResourceManager): +@clsinit_copy +class OMFChannel(OMFResource): """ .. class:: Class Args : @@ -44,21 +46,13 @@ class OMFChannel(ResourceManager): _rtype = "OMFChannel" _authorized_connections = ["OMFWifiInterface", "OMFNode"] - @classmethod def _register_attributes(cls): """Register the attributes of an OMF channel + """ channel = Attribute("channel", "Name of the application") - xmppSlice = Attribute("xmppSlice","Name of the slice", flags = Flags.Credential) - xmppHost = Attribute("xmppHost", "Xmpp Server",flags = Flags.Credential) - xmppPort = Attribute("xmppPort", "Xmpp Port",flags = Flags.Credential) - xmppPassword = Attribute("xmppPassword", "Xmpp Port",flags = Flags.Credential) cls._register_attribute(channel) - cls._register_attribute(xmppSlice) - cls._register_attribute(xmppHost) - cls._register_attribute(xmppPort) - cls._register_attribute(xmppPassword) def __init__(self, ec, guid): """ @@ -76,8 +70,13 @@ class OMFChannel(ResourceManager): self._omf_api = None + @property + def exp_id(self): + return self.ec.exp_id + def valid_connection(self, guid): - """Check if the connection with the guid in parameter is possible. Only meaningful connections are allowed. + """ Check if the connection with the guid in parameter is possible. + Only meaningful connections are allowed. :param guid: Guid of the current RM :type guid: int @@ -85,12 +84,17 @@ class OMFChannel(ResourceManager): """ rm = self.ec.get_resource(guid) + if rm.rtype() in self._authorized_connections: - msg = "Connection between %s %s and %s %s accepted" % (self.rtype(), self._guid, rm.rtype(), guid) + msg = "Connection between %s %s and %s %s accepted" % ( + self.rtype(), self._guid, rm.rtype(), guid) self.debug(msg) return True - msg = "Connection between %s %s and %s %s refused" % (self.rtype(), self._guid, rm.rtype(), guid) + + msg = "Connection between %s %s and %s %s refused" % ( + self.rtype(), self._guid, rm.rtype(), guid) self.debug(msg) + return False def _get_target(self, conn_set): @@ -103,17 +107,19 @@ class OMFChannel(ResourceManager): :return: self._nodes_guid """ + res = [] for elt in conn_set: rm_iface = self.ec.get_resource(elt) for conn in rm_iface.connections: rm_node = self.ec.get_resource(conn) if rm_node.rtype() == "OMFNode" and rm_node.get('hostname'): - if rm_iface.state < ResourceState.READY or rm_node.state < ResourceState.READY: + if rm_iface.state < ResourceState.PROVISIONED or \ + rm_node.state < ResourceState.READY: return "reschedule" couple = [rm_node.get('hostname'), rm_iface.get('alias')] #print couple - self._nodes_guid.append(couple) - return self._nodes_guid + res.append(couple) + return res def discover(self): """ Discover the availables channels @@ -128,58 +134,71 @@ class OMFChannel(ResourceManager): pass def deploy(self): - """Deploy the RM. It means : Get the xmpp client and send messages using OMF 5.4 protocol to configure the channel - It becomes DEPLOYED after sending messages to configure the channel + """ Deploy the RM. It means : Get the xmpp client and send messages + using OMF 5.4 protocol to configure the channel. + It becomes DEPLOYED after sending messages to configure the channel """ if not self._omf_api : self._omf_api = OMFAPIFactory.get_api(self.get('xmppSlice'), - self.get('xmppHost'), self.get('xmppPort'), self.get('xmppPassword')) - - - if self.get('channel'): - set_nodes = self._get_target(self._connections) - if set_nodes == "reschedule" : - self.ec.schedule(reschedule_delay, self.deploy) - return - print set_nodes - try: - for couple in set_nodes: - #print "Couple node/alias : " + couple[0] + " , " + couple[1] - attrval = self.get('channel') - attrname = "net/%s/%s" % (couple[1], 'channel') - self._omf_api.configure(couple[0], attrname, attrval) - except AttributeError: - self._state = ResourceState.FAILED - msg = "Credentials are not initialzed. XMPP Connections impossible" - self.debug(msg) - raise - else : + self.get('xmppHost'), self.get('xmppPort'), + self.get('xmppPassword'), exp_id = self.exp_id) + + if not self._omf_api : + msg = "Credentials are not initialzed. XMPP Connections impossible" + self.error(msg) + self.fail() + return + + if not self.get('channel'): msg = "Channel's value is not initialized" self.error(msg) + self.fail() + raise + + self._nodes_guid = self._get_target(self._connections) + if self._nodes_guid == "reschedule" : + self.ec.schedule("2s", self.deploy) + return False + + try: + for couple in self._nodes_guid: + #print "Couple node/alias : " + couple[0] + " , " + couple[1] + attrval = self.get('channel') + attrname = "net/%s/%s" % (couple[1], 'channel') + self._omf_api.configure(couple[0], attrname, attrval) + except AttributeError: + msg = "Credentials are not initialzed. XMPP Connections impossible" + self.error(msg) + self.fail() + raise super(OMFChannel, self).deploy() def start(self): - """Start the RM. It means nothing special for a channel for now - It becomes STARTED as soon as this method starts. + """ Start the RM. It means nothing special for a channel for now + It becomes STARTED as soon as this method starts. """ super(OMFChannel, self).start() def stop(self): - """Stop the RM. It means nothing special for a channel for now - It becomes STOPPED as soon as this method is called + """ Stop the RM. It means nothing special for a channel for now + It becomes STOPPED as soon as this method is called """ super(OMFChannel, self).stop() + self.set_finished() def release(self): - """Clean the RM at the end of the experiment and release the API + """ Clean the RM at the end of the experiment and release the API """ if self._omf_api : OMFAPIFactory.release_api(self.get('xmppSlice'), - self.get('xmppHost'), self.get('xmppPort'), self.get('xmppPassword')) + self.get('xmppHost'), self.get('xmppPort'), + self.get('xmppPassword'), exp_id = self.exp_id) + + super(OMFChannel, self).release()