adding scheduled slave startup

git-svn-id: https://hudson.dev.java.net/svn/hudson/trunk/hudson/main@14446 71c3de6d-444a-0410-be80-ed276b4c234a

core/src/main/java/hudson/slaves/RetentionStrategy.java

@@ -184,5 +184,6 @@ public abstract class RetentionStrategy<T extends Computer> implements Describab
 
     static {
         LIST.load(Demand.class);
+        LIST.load(SimpleScheduledRetentionStrategy.class);
     }
 }

core/src/main/java/hudson/slaves/SimpleScheduledRetentionStrategy.java

+package hudson.slaves;
+
+import java.util.logging.Logger;
+import java.util.logging.Level;
+import java.util.Calendar;
+import java.util.GregorianCalendar;
+import java.util.Date;
+import java.io.IOException;
+import java.io.ObjectStreamException;
+import java.io.InvalidObjectException;
+
+import javax.servlet.ServletException;
+
+import hudson.model.Descriptor;
+import hudson.scheduler.CronTabList;
+import hudson.util.FormFieldValidator;
+import static hudson.Util.fixNull;
+import hudson.Util;
+import org.kohsuke.stapler.DataBoundConstructor;
+import org.kohsuke.stapler.StaplerRequest;
+import org.kohsuke.stapler.StaplerResponse;
+import antlr.ANTLRException;
+
+/**
+ * Created by IntelliJ IDEA.
+ *
+ * @author connollys
+ * @since Jan 13, 2009 1:17:34 PM
+ */
+public class SimpleScheduledRetentionStrategy extends RetentionStrategy<SlaveComputer> {
+
+    private static final Logger LOGGER = Logger.getLogger(SimpleScheduledRetentionStrategy.class.getName());
+
+    private final String startTimeSpec;
+    private transient CronTabList tabs;
+    private transient Calendar lastChecked;
+    private final long upTimeMins;
+
+    @DataBoundConstructor
+    public SimpleScheduledRetentionStrategy(String startTimeSpec, long upTimeMins) throws ANTLRException {
+        this.startTimeSpec = startTimeSpec;
+        this.tabs = CronTabList.create(startTimeSpec);
+        this.lastChecked = new GregorianCalendar();
+        this.upTimeMins = upTimeMins;
+        this.lastChecked.add(Calendar.MINUTE, -1);
+    }
+
+    protected Object readResolve() throws ObjectStreamException {
+        try {
+            tabs = CronTabList.create(startTimeSpec);
+            lastChecked = new GregorianCalendar();
+            this.lastChecked.add(Calendar.MINUTE, -1);
+        } catch (ANTLRException e) {
+            InvalidObjectException x = new InvalidObjectException(e.getMessage());
+            x.initCause(e);
+            throw x;
+        }
+        return this;
+    }
+
+    public long check(SlaveComputer c) {
+        if (c.isOffline()) {
+            Calendar startUp = null;
+            while (new Date().getTime() - lastChecked.getTimeInMillis() > 1000) {
+                LOGGER.log(Level.FINE, "scheduled start checking {0}", lastChecked);
+                if (tabs.check(lastChecked)) {
+                    // need to ensure that we don't fire up after the upTime has been passed
+                    // if we have an aggressive upTime, or a slow check
+                    startUp = (Calendar) lastChecked.clone();
+                }
+                lastChecked.add(Calendar.MINUTE, 1);
+            }
+            if (startUp != null) {
+                if (new Date().getTime() - startUp.getTimeInMillis() > 1000 * 60L * upTimeMins) {
+                    LOGGER.log(Level.INFO, "Missed scheduled startup");
+                } else {
+                    LOGGER.log(Level.INFO, "Launching computer {0} per schedule", new Object[]{c.getName()});
+                    if (c.isLaunchSupported()) {
+                        c.connect(true);
+                    }
+                }
+            }
+            return upTimeMins;
+        } else {
+            if (new Date().getTime() - lastChecked.getTimeInMillis() > 1000 * 60L * upTimeMins) {
+                LOGGER.log(Level.INFO, "Disconnecting computer {0} as it has finished its scheduled uptime",
+                        new Object[]{c.getName()});
+                c.disconnect();
+            }
+            return 1;
+        }
+    }
+
+    public static final DescriptorImpl DESCRIPTOR = new DescriptorImpl();
+
+    public Descriptor<RetentionStrategy<?>> getDescriptor() {
+        return DESCRIPTOR;
+    }
+
+    private static class DescriptorImpl extends Descriptor<RetentionStrategy<?>> {
+
+        /**
+         * Constructs a new DescriptorImpl.
+         */
+        public DescriptorImpl() {
+            super(Demand.class);
+        }
+
+        public String getDisplayName() {
+            return Messages.SimpleScheduledRetentionStrategy_displayName();
+        }
+
+        /**
+         * Performs syntax check.
+         */
+        public void doCheck(StaplerRequest req, StaplerResponse rsp) throws IOException, ServletException {
+            // false==No permission needed for this syntax check
+            new FormFieldValidator(req, rsp, false) {
+                @Override
+                protected void check() throws IOException, ServletException {
+                    try {
+                        String msg = CronTabList.create(fixNull(request.getParameter("value"))).checkSanity();
+                        if (msg != null) {
+                            warning(msg);
+                        } else {
+                            ok();
+                        }
+                    } catch (ANTLRException e) {
+                        error(e.getMessage());
+                    }
+                }
+            }.process();
+        }
+    }
+
+    static {
+        LIST.add(DESCRIPTOR);
+    }
+
+}

core/src/main/resources/hudson/slaves/Messages.properties

@@ -5,4 +5,5 @@ JNLPLauncher.displayName=Launch slave agents via JNLP
 ComputerLauncher.unexpectedError=Unexpected error in launching a slave. This is probably a bug in Hudson
 ComputerLauncher.abortedLaunch=Launching slave process aborted.
 CommandLauncher.NoLaunchCommand=No launch command specified
-DumbSlave.displayName=Dumb Slave
\ No newline at end of file
+DumbSlave.displayName=Dumb Slave
+SimpleScheduledRetentionStrategy.displayName=Take this slave on-line according to a schedule
\ No newline at end of file

core/src/main/resources/hudson/slaves/SimpleScheduledRetentionStrategy/config.jelly

+<j:jelly xmlns:j="jelly:core" xmlns:st="jelly:stapler" xmlns:d="jelly:define" xmlns:l="/lib/layout"
+         xmlns:t="/lib/hudson" xmlns:f="/lib/form">
+    <f:entry title="${%Startup Schedule}" help="/help/project-config/timer-format.html">
+        <f:textarea name="slave.startTimeSpec" checkUrl="'${rootURL}/slaves/SimpleScheduledRetentionStrategy/check?value='+escape(this.value)"
+                    value="${s.startTimeSpec}"/>
+    </f:entry>
+    <f:entry title="${%Scheduled Uptime (mins)}">
+        <f:textarea name="slave.upTimeMins" value="${s.upTimeMins}"/>
+    </f:entry>
+</j:jelly>
\ No newline at end of file