X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=src%2Fnepi%2Fresources%2Flinux%2Fccn%2Fccncontent.py;h=da8c2ab062bbd380bef7d695fc5bd45dcb661d0b;hb=6285ca51026efb69642eea9dfc7c480e722d84a9;hp=d5500118e0292a23cb5c06b9c791b7274e8268f2;hpb=4c64cfa394d6210febf4c384276820093b104106;p=nepi.git diff --git a/src/nepi/resources/linux/ccn/ccncontent.py b/src/nepi/resources/linux/ccn/ccncontent.py index d5500118..da8c2ab0 100644 --- a/src/nepi/resources/linux/ccn/ccncontent.py +++ b/src/nepi/resources/linux/ccn/ccncontent.py @@ -3,9 +3,8 @@ # 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. +# it under the terms of the GNU General Public License version 2 as +# published by the Free Software Foundation; # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -18,113 +17,133 @@ # Author: Alina Quereilhac from nepi.execution.attribute import Attribute, Flags, Types -from nepi.execution.trace import Trace, TraceAttr -from nepi.execution.resource import ResourceManager, clsinit_copy, ResourceState, \ +from nepi.execution.resource import clsinit_copy, ResourceState, \ ResourceAction from nepi.resources.linux.application import LinuxApplication from nepi.resources.linux.ccn.ccnr import LinuxCCNR -from nepi.util.timefuncs import strfnow, strfdiff +from nepi.util.timefuncs import tnow import os -reschedule_delay = "0.5s" - @clsinit_copy class LinuxCCNContent(LinuxApplication): - _rtype = "LinuxCCNContent" + _rtype = "linux::CCNContent" @classmethod def _register_attributes(cls): content_name = Attribute("contentName", "The name of the content to publish (e.g. ccn:/VIDEO) ", - flags = Flags.ExecReadOnly) + flags = Flags.Design) content = Attribute("content", "The content to publish. It can be a path to a file or plain text ", - flags = Flags.ExecReadOnly) + flags = Flags.Design) + + scope = Attribute("scope", + "Use the given scope on the start-write request (if -r specified). " + "scope can be 1 (local), 2 (neighborhood), or 3 (unlimited). " + "Note that a scope of 3 is encoded as the absence of any scope in the interest. ", + type = Types.Integer, + default = 1, + flags = Flags.Design) cls._register_attribute(content_name) cls._register_attribute(content) + cls._register_attribute(scope) def __init__(self, ec, guid): super(LinuxCCNContent, self).__init__(ec, guid) self._home = "content-%s" % self.guid - self._published = False @property def ccnr(self): - ccnr = self.get_connected(LinuxCCNR.rtype()) + ccnr = self.get_connected(LinuxCCNR.get_rtype()) if ccnr: return ccnr[0] return None + @property + def ccnd(self): + if self.ccnr: return self.ccnr.ccnd + return None + @property def node(self): if self.ccnr: return self.ccnr.node return None - def deploy(self): - if not self.get("command"): - self.set("command", self._default_command) - - if not self.get("env"): - self.set("env", self._default_environment) + def do_deploy(self): + if not self.ccnr or self.ccnr.state < ResourceState.READY: + self.debug("---- RESCHEDULING DEPLOY ---- node state %s " % self.node.state ) + + # ccnr needs to wait until ccnd is deployed and running + self.ec.schedule(self.reschedule_delay, self.deploy) + else: + if not self.get("command"): + self.set("command", self._start_command) + + if not self.get("env"): + self.set("env", self._environment) - if not self.get("stdin"): # set content to stdin, so the content will be # uploaded during provision self.set("stdin", self.get("content")) - # Wait until associated ccnd is provisioned - ccnr = self.ccnr + command = self.get("command") - if not ccnr or ccnr.state < ResourceState.READY: - # ccnr needs to wait until ccnd is deployed and running - self.ec.schedule(reschedule_delay, self.deploy) - else: - # Invoke the actual deployment - super(LinuxCCNContent, self).deploy() - - # As soon as the ccnr is running we can push the content - # to the repository ( we don't want to lose time launching - # writting the content to the repository later on ) - if self._state == ResourceState.READY: - self._start_in_background() - self._published = True - - def start(self): - # CCNR should already be started by now. - # Nothing to do but to set the state to STARTED - if self._published: - self._start_time = strfnow() - self._state = ResourceState.STARTED + self.info("Deploying command '%s' " % command) + + self.do_discover() + self.do_provision() + + self.set_ready() + + def upload_start_command(self): + command = self.get("command") + env = self.get("env") + + self.info("Uploading command '%s'" % command) + + # We want to make sure the content is published + # before the experiment starts. + # Run the command as a bash script in the background, + # in the host ( but wait until the command has + # finished to continue ) + env = self.replace_paths(env) + command = self.replace_paths(command) + + (out, err), proc = self.execute_command(command, + env, blocking = True) + + if proc.poll(): + msg = "Failed to execute command" + self.error(msg, out, err) + raise RuntimeError(msg) + + def do_start(self): + if self.state == ResourceState.READY: + command = self.get("command") + self.info("Starting command '%s'" % command) + + self.set_started() else: - msg = "Failed to execute command '%s'" % command + msg = " Failed to execute command '%s'" % command self.error(msg, out, err) - self._state = ResourceState.FAILED - raise RuntimeError, msg + raise RuntimeError(msg) @property - def state(self): - state = super(LinuxCCNContent, self).state - if self._state in [ResourceState.FINISHED, ResourceState.FAILED]: - self._published = False - - if self._state == ResourceState.READY: - # CCND is really deployed only when ccn daemon is running - if not self._published: - return ResourceState.PROVISIONED - - return self._state + def _start_command(self): + command = ["ccnseqwriter"] + command.append("-r %s" % self.get("contentName")) + command.append("-s %d" % self.get("scope")) + command.append("< %s" % os.path.join(self.app_home, 'stdin')) - @property - def _default_command(self): - return "ccnseqwriter -r %s " % self.get("contentName") + command = " ".join(command) + return command @property - def _default_environment(self): - env = "PATH=$PATH:${EXP_HOME}/ccnx/bin " - return env - + def _environment(self): + return self.ccnd.path + def valid_connection(self, guid): # TODO: Validate! return True