Queue.java 55.6 KB
Newer Older
K
kohsuke 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/*
 * The MIT License
 * 
 * Copyright (c) 2004-2009, Sun Microsystems, Inc., Kohsuke Kawaguchi, Stephen Connolly, Tom Huybrechts
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
K
kohsuke 已提交
24 25
package hudson.model;

26
import hudson.BulkChange;
27 28
import hudson.ExtensionList;
import hudson.ExtensionPoint;
29
import hudson.Util;
K
kohsuke 已提交
30
import hudson.XmlFile;
K
kohsuke 已提交
31 32
import hudson.init.Initializer;
import static hudson.init.InitMilestone.JOB_LOADED;
33 34
import static hudson.util.Iterators.reverse;

35 36
import hudson.cli.declarative.CLIMethod;
import hudson.cli.declarative.CLIResolver;
37
import hudson.model.queue.AbstractQueueTask;
38
import hudson.model.queue.FutureImpl;
39
import hudson.model.queue.QueueSorter;
40
import hudson.model.queue.QueueTaskDispatcher;
41
import hudson.model.queue.WorkUnit;
K
kohsuke 已提交
42
import hudson.model.Node.Mode;
43
import hudson.model.listeners.SaveableListener;
44
import hudson.model.queue.CauseOfBlockage;
K
kohsuke 已提交
45
import hudson.model.queue.FoldableAction;
46 47 48 49
import hudson.model.queue.CauseOfBlockage.BecauseLabelIsBusy;
import hudson.model.queue.CauseOfBlockage.BecauseNodeIsOffline;
import hudson.model.queue.CauseOfBlockage.BecauseLabelIsOffline;
import hudson.model.queue.CauseOfBlockage.BecauseNodeIsBusy;
50
import hudson.model.queue.WorkUnitContext;
51 52
import hudson.triggers.SafeTimerTask;
import hudson.triggers.Trigger;
K
kohsuke 已提交
53
import hudson.util.OneShotEvent;
54
import hudson.util.TimeUnit2;
55
import hudson.util.XStream2;
56 57
import hudson.util.ConsistentHash;
import hudson.util.ConsistentHash.Hash;
K
kohsuke 已提交
58

59 60 61 62 63
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
64
import java.lang.ref.WeakReference;
K
kohsuke 已提交
65
import java.util.ArrayList;
66
import java.util.Arrays;
K
kohsuke 已提交
67
import java.util.Calendar;
68
import java.util.Collection;
69
import java.util.Collections;
K
kohsuke 已提交
70 71 72 73 74
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
75
import java.util.NoSuchElementException;
K
kohsuke 已提交
76 77
import java.util.Set;
import java.util.TreeSet;
S
stephenconnolly 已提交
78
import java.util.Map.Entry;
79
import java.util.concurrent.atomic.AtomicInteger;
80
import java.util.concurrent.Future;
K
kohsuke 已提交
81 82 83
import java.util.logging.Level;
import java.util.logging.Logger;

K
kohsuke 已提交
84
import javax.management.timer.Timer;
85
import javax.servlet.ServletException;
K
kohsuke 已提交
86 87

import org.acegisecurity.AccessDeniedException;
88 89
import org.kohsuke.stapler.StaplerRequest;
import org.kohsuke.stapler.StaplerResponse;
K
kohsuke 已提交
90 91 92 93 94 95
import org.kohsuke.stapler.export.Exported;
import org.kohsuke.stapler.export.ExportedBean;

import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.converters.basic.AbstractSingleValueConverter;

K
kohsuke 已提交
96 97
/**
 * Build queue.
98 99
 *
 * <p>
100 101
 * This class implements the core scheduling logic. {@link Task} represents the executable
 * task that are placed in the queue. While in the queue, it's wrapped into {@link Item}
102
 * so that we can keep track of additional data used for deciding what to exeucte when.
103 104
 *
 * <p>
105 106 107 108 109 110
 * Items in queue goes through several stages, as depicted below:
 * <pre>
 * (enter) --> waitingList --+--> blockedProjects
 *                           |        ^
 *                           |        |
 *                           |        v
111
 *                           +--> buildables ---> pending ---> (executed)
112
 * </pre>
113 114
 *
 * <p>
115 116 117
 * In addition, at any stage, an item can be removed from the queue (for example, when the user
 * cancels a job in the queue.) See the corresponding field for their exact meanings.
 *
K
kohsuke 已提交
118 119
 * @author Kohsuke Kawaguchi
 */
120
@ExportedBean
121
public class Queue extends ResourceController implements Saveable {
K
kohsuke 已提交
122
    /**
123
     * Items that are waiting for its quiet period to pass.
124 125
     *
     * <p>
K
kohsuke 已提交
126 127 128
     * This consists of {@link Item}s that cannot be run yet
     * because its time has not yet come.
     */
129
    private final Set<WaitingItem> waitingList = new TreeSet<WaitingItem>();
K
kohsuke 已提交
130 131

    /**
132
     * {@link Task}s that can be built immediately
133 134 135
     * but blocked because another build is in progress,
     * required {@link Resource}s are not available, or otherwise blocked
     * by {@link Task#isBuildBlocked()}.
K
kohsuke 已提交
136
     */
137
    private final ItemList<BlockedItem> blockedProjects = new ItemList<BlockedItem>();
K
kohsuke 已提交
138 139

    /**
140
     * {@link Task}s that can be built immediately
K
kohsuke 已提交
141
     * that are waiting for available {@link Executor}.
142
     * This list is sorted in such a way that earlier items are built earlier.
K
kohsuke 已提交
143
     */
144
    private final ItemList<BuildableItem> buildables = new ItemList<BuildableItem>();
145

146
    /**
147 148
     * {@link Task}s that are being handed over to the executor, but execution
     * has not started yet.
149 150 151
     */
    private final ItemList<BuildableItem> pendings = new ItemList<BuildableItem>();

K
kohsuke 已提交
152 153
    /**
     * Data structure created for each idle {@link Executor}.
154
     * This is a job offer from the queue to an executor.
155 156
     *
     * <p>
K
kohsuke 已提交
157 158 159
     * An idle executor (that calls {@link Queue#pop()} creates
     * a new {@link JobOffer} and gets itself {@linkplain Queue#parked parked},
     * and we'll eventually hand out an {@link #item} to build.
K
kohsuke 已提交
160
     */
161
    public class JobOffer {
K
kohsuke 已提交
162
        public final Executor executor;
K
kohsuke 已提交
163 164 165 166 167

        /**
         * Used to wake up an executor, when it has an offered
         * {@link Project} to build.
         */
168
        private final OneShotEvent event = new OneShotEvent(Queue.this);
K
kohsuke 已提交
169

K
kohsuke 已提交
170 171 172 173
        /**
         * The project that this {@link Executor} is going to build.
         * (Or null, in which case event is used to trigger a queue maintenance.)
         */
K
kohsuke 已提交
174
        private BuildableItem item;
K
kohsuke 已提交
175

K
kohsuke 已提交
176
        private JobOffer(Executor executor) {
K
kohsuke 已提交
177 178 179
            this.executor = executor;
        }

180 181 182
        public void set(BuildableItem p) {
            assert this.item == null;
            this.item = p;
K
kohsuke 已提交
183 184 185
            event.signal();
        }

186 187 188 189
        /**
         * Verifies that the {@link Executor} represented by this object is capable of executing the given task.
         */
        public boolean canTake(Task task) {
190 191 192
            Node node = getNode();
            if (node==null)     return false;   // this executor is about to die

193 194
            if(node.canTake(task)!=null)
                return false;   // this node is not able to take the task
195

196 197 198
            for (QueueTaskDispatcher d : QueueTaskDispatcher.all())
                if (d.canTake(node,task)!=null)
                    return false;
199 200 201 202 203 204 205

            return isAvailable();
        }

        /**
         * Is this executor ready to accept some tasks?
         */
K
kohsuke 已提交
206
        public boolean isAvailable() {
207
            return item == null && !executor.getOwner().isOffline() && executor.getOwner().isAcceptingTasks();
K
kohsuke 已提交
208 209 210 211 212 213 214
        }

        public Node getNode() {
            return executor.getOwner().getNode();
        }

        public boolean isNotExclusive() {
215
            return getNode().getMode() == Mode.NORMAL;
K
kohsuke 已提交
216 217 218
        }
    }

S
stephenconnolly 已提交
219
    /**
K
kohsuke 已提交
220
     * The executors that are currently waiting for a job to run.
S
stephenconnolly 已提交
221
     */
K
kohsuke 已提交
222
    private final Map<Executor,JobOffer> parked = new HashMap<Executor,JobOffer>();
K
kohsuke 已提交
223

224
    private volatile transient LoadBalancer loadBalancer;
225

226
    private volatile transient QueueSorter sorter;
227

228 229
    public Queue(LoadBalancer loadBalancer) {
        this.loadBalancer =  loadBalancer.sanitize();
230 231 232 233 234
        // if all the executors are busy doing something, then the queue won't be maintained in
        // timely fashion, so use another thread to make sure it happens.
        new MaintainTask(this);
    }

235 236 237 238 239 240 241 242 243
    public LoadBalancer getLoadBalancer() {
        return loadBalancer;
    }

    public void setLoadBalancer(LoadBalancer loadBalancer) {
        if(loadBalancer==null)  throw new IllegalArgumentException();
        this.loadBalancer = loadBalancer;
    }

244 245 246 247 248 249 250 251
    public QueueSorter getSorter() {
        return sorter;
    }

    public void setSorter(QueueSorter sorter) {
        this.sorter = sorter;
    }

K
kohsuke 已提交
252 253 254 255 256
    /**
     * Loads the queue contents that was {@link #save() saved}.
     */
    public synchronized void load() {
        try {
K
TAB->WS  
kohsuke 已提交
257
            // first try the old format
K
kohsuke 已提交
258
            File queueFile = getQueueFile();
K
kohsuke 已提交
259 260
            if (queueFile.exists()) {
                BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(queueFile)));
K
TAB->WS  
kohsuke 已提交
261 262 263 264
                String line;
                while ((line = in.readLine()) != null) {
                    AbstractProject j = Hudson.getInstance().getItemByFullName(line, AbstractProject.class);
                    if (j != null)
M
mindless 已提交
265
                        j.scheduleBuild();
K
TAB->WS  
kohsuke 已提交
266 267 268 269 270 271 272
                }
                in.close();
                // discard the queue file now that we are done
                queueFile.delete();
            } else {
                queueFile = getXMLQueueFile();
                if (queueFile.exists()) {
273
                    List list = (List) new XmlFile(XSTREAM, queueFile).read();
K
kohsuke 已提交
274 275 276
                    int maxId = 0;
                    for (Object o : list) {
                        if (o instanceof Task) {
277
                            // backward compatibility
278
                            schedule((Task)o, 0);
K
kohsuke 已提交
279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
                        } else if (o instanceof Item) {
                            Item item = (Item)o;
                            if(item.task==null)
                                continue;   // botched persistence. throw this one away

                            maxId = Math.max(maxId, item.id);
                            if (item instanceof WaitingItem) {
                                waitingList.add((WaitingItem) item);
                            } else if (item instanceof BlockedItem) {
                                blockedProjects.put(item.task, (BlockedItem) item);
                            } else if (item instanceof BuildableItem) {
                                buildables.add((BuildableItem) item);
                            } else {
                                throw new IllegalStateException("Unknown item type! " + item);
                            }
                        } // this conveniently ignores null
K
TAB->WS  
kohsuke 已提交
295
                    }
K
kohsuke 已提交
296
                    WaitingItem.COUNTER.set(maxId);
297 298 299 300 301 302 303 304 305

                    // I just had an incident where all the executors are dead at AbstractProject._getRuns()
                    // because runs is null. Debugger revealed that this is caused by a MatrixConfiguration
                    // object that doesn't appear to be de-serialized properly.
                    // I don't know how this problem happened, but to diagnose this problem better
                    // when it happens again, save the old queue file for introspection.
                    File bk = new File(queueFile.getPath() + ".bak");
                    bk.delete();
                    queueFile.renameTo(bk);
K
TAB->WS  
kohsuke 已提交
306 307 308
                    queueFile.delete();
                }
            }
309
        } catch (IOException e) {
310
            LOGGER.log(Level.WARNING, "Failed to load the queue file " + getXMLQueueFile(), e);
K
kohsuke 已提交
311 312 313 314 315 316 317
        }
    }

    /**
     * Persists the queue contents to the disk.
     */
    public synchronized void save() {
318 319
        if(BulkChange.contains(this))  return;
        
K
kohsuke 已提交
320
        // write out the tasks on the queue
321
    	ArrayList<Queue.Item> items = new ArrayList<Queue.Item>();
K
kohsuke 已提交
322
    	for (Item item: getItems()) {
K
kohsuke 已提交
323
            if(item.task instanceof TransientTask)  continue;
324
    	    items.add(item);
K
kohsuke 已提交
325
    	}
K
kohsuke 已提交
326

K
kohsuke 已提交
327
        try {
328 329 330
            XmlFile queueFile = new XmlFile(XSTREAM, getXMLQueueFile());
            queueFile.write(items);
            SaveableListener.fireOnChange(this, queueFile);
331
        } catch (IOException e) {
332
            LOGGER.log(Level.WARNING, "Failed to write out the queue file " + getXMLQueueFile(), e);
K
kohsuke 已提交
333 334 335
        }
    }

336 337 338
    /**
     * Wipes out all the items currently in the queue, as if all of them are cancelled at once.
     */
339
    @CLIMethod(name="clear-queue")
340
    public synchronized void clear() {
341
        Hudson.getInstance().checkPermission(Hudson.ADMINISTER);
342 343
        for (WaitingItem i : waitingList)
            i.onCancelled();
344
        waitingList.clear();
345 346
        blockedProjects.cancelAll();
        buildables.cancelAll();
347 348 349
        scheduleMaintenance();
    }

K
kohsuke 已提交
350
    private File getQueueFile() {
K
TAB->WS  
kohsuke 已提交
351 352
        return new File(Hudson.getInstance().getRootDir(), "queue.txt");
    }
K
kohsuke 已提交
353

354
    /*package*/ File getXMLQueueFile() {
K
TAB->WS  
kohsuke 已提交
355 356
        return new File(Hudson.getInstance().getRootDir(), "queue.xml");
    }
K
kohsuke 已提交
357

358 359 360 361 362 363 364 365
    /**
     * @deprecated as of 1.311
     *      Use {@link #schedule(AbstractProject)}
     */
    public boolean add(AbstractProject p) {
        return schedule(p)!=null;
    }

K
kohsuke 已提交
366 367
    /**
     * Schedule a new build for this project.
368
     *
369 370 371
     * @return true if the project is actually added to the queue.
     *         false if the queue contained it and therefore the add()
     *         was noop
K
kohsuke 已提交
372
     */
373 374
    public WaitingItem schedule(AbstractProject p) {
        return schedule(p, p.getQuietPeriod());
375 376 377 378
    }

    /**
     * Schedules a new build with a custom quiet period.
379 380
     *
     * <p>
K
kohsuke 已提交
381 382
     * Left for backward compatibility with &lt;1.114.
     *
383
     * @since 1.105
384 385
     * @deprecated as of 1.311
     *      Use {@link #schedule(Task, int)}
386
     */
387 388
    public boolean add(AbstractProject p, int quietPeriod) {
        return schedule(p, quietPeriod)!=null;
K
kohsuke 已提交
389
    }
390 391 392 393

    /**
     * Schedules an execution of a task.
     *
394
     * @param actions
K
kohsuke 已提交
395 396 397
     *      These actions can be used for associating information scoped to a particular build, to
     *      the task being queued. Upon the start of the build, these {@link Action}s will be automatically
     *      added to the {@link Run} object, and hence avaialable to everyone.
398
     *      For the convenience of the caller, this list can contain null, and those will be silently ignored.
399 400 401 402 403 404 405 406 407 408
     * @since 1.311
     * @return
     *      null if this task is already in the queue and therefore the add operation was no-op.
     *      Otherwise indicates the {@link WaitingItem} object added, although the nature of the queue
     *      is that such {@link Item} only captures the state of the item at a particular moment,
     *      and by the time you inspect the object, some of its information can be already stale.
     *
     *      That said, one can still look at {@link WaitingItem#future}, {@link WaitingItem#id}, etc.
     */
    public synchronized WaitingItem schedule(Task p, int quietPeriod, List<Action> actions) {
409 410 411 412 413 414 415
        // remove nulls
        actions = new ArrayList<Action>(actions);
        for (Iterator<Action> itr = actions.iterator(); itr.hasNext();) {
            Action a =  itr.next();
            if (a==null)    itr.remove();
        }

K
kohsuke 已提交
416 417 418 419 420
    	for(QueueDecisionHandler h : QueueDecisionHandler.all())
    		if (!h.shouldSchedule(p, actions))
                return null;    // veto

        return scheduleInternal(p, quietPeriod, actions);
421
    }
422

423 424 425 426 427 428 429 430 431 432 433 434 435
    /**
     * Schedules an execution of a task.
     *
     * @since 1.311
     * @return
     *      null if this task is already in the queue and therefore the add operation was no-op.
     *      Otherwise indicates the {@link WaitingItem} object added, although the nature of the queue
     *      is that such {@link Item} only captures the state of the item at a particular moment,
     *      and by the time you inspect the object, some of its information can be already stale.
     *
     *      That said, one can still look at {@link WaitingItem#future}, {@link WaitingItem#id}, etc.
     */
    private synchronized WaitingItem scheduleInternal(Task p, int quietPeriod, List<Action> actions) {
K
kohsuke 已提交
436
        Calendar due = new GregorianCalendar();
437
    	due.add(Calendar.SECOND, quietPeriod);
438

439
        // Do we already have this task in the queue? Because if so, we won't schedule a new one.
440
    	List<Item> duplicatesInQueue = new ArrayList<Item>();
K
kohsuke 已提交
441
    	for(Item item : getItems(p)) {
442
    		boolean shouldScheduleItem = false;
K
kohsuke 已提交
443 444
    		for (QueueAction action: item.getActions(QueueAction.class)) {
                shouldScheduleItem |= action.shouldSchedule(actions);
445
    		}
K
kohsuke 已提交
446 447
    		for (QueueAction action: Util.filter(actions,QueueAction.class)) {
                shouldScheduleItem |= action.shouldSchedule(item.getActions());
448 449 450 451 452
    		}
    		if(!shouldScheduleItem) {
    			duplicatesInQueue.add(item);
    		}
    	}
453
    	if (duplicatesInQueue.isEmpty()) {
454
    		LOGGER.fine(p.getFullDisplayName() + " added to queue");
455

456
    		// put the item in the queue
K
kohsuke 已提交
457 458 459 460 461
            WaitingItem added = new WaitingItem(due,p,actions);
    		waitingList.add(added);
            scheduleMaintenance();   // let an executor know that a new item is in the queue.
            return added;
    	}
K
kohsuke 已提交
462

K
kohsuke 已提交
463
        LOGGER.fine(p.getFullDisplayName() + " is already in the queue");
464

K
kohsuke 已提交
465 466 467
        // but let the actions affect the existing stuff.
        for(Item item : duplicatesInQueue) {
            for(FoldableAction a : Util.filter(actions,FoldableAction.class)) {
468
                a.foldIntoExisting(item, p, actions);
K
kohsuke 已提交
469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494
            }
        }

        boolean queueUpdated = false;
        for(WaitingItem wi : Util.filter(duplicatesInQueue,WaitingItem.class)) {
            if(quietPeriod<=0) {
                // the user really wants to build now, and they mean NOW.
                // so let's pull in the timestamp if we can.
                if (wi.timestamp.before(due))
                    continue;
            } else {
                // otherwise we do the normal quiet period implementation
                if (wi.timestamp.after(due))
                    continue;
                // quiet period timer reset. start the period over again
            }

            // waitingList is sorted, so when we change a timestamp we need to maintain order
            waitingList.remove(wi);
            wi.timestamp = due;
            waitingList.add(wi);
            queueUpdated=true;
        }

        if (queueUpdated)   scheduleMaintenance();
        return null;
K
kohsuke 已提交
495
    }
496
    
497 498 499 500
    /**
     * @deprecated as of 1.311
     *      Use {@link #schedule(Task, int)} 
     */
501
    public synchronized boolean add(Task p, int quietPeriod) {
502
    	return schedule(p, quietPeriod)!=null;
503 504
    }

505 506 507 508 509 510
    public synchronized WaitingItem schedule(Task p, int quietPeriod) {
    	return schedule(p, quietPeriod, new Action[0]);
    }

    /**
     * @deprecated as of 1.311
511
     *      Use {@link #schedule(Task, int, Action...)} 
512
     */
513
    public synchronized boolean add(Task p, int quietPeriod, Action... actions) {
514 515 516
    	return schedule(p, quietPeriod, actions)!=null;
    }

517
    /**
K
kohsuke 已提交
518
     * Convenience wrapper method around {@link #schedule(Task, int, List)}
519
     */
520 521
    public synchronized WaitingItem schedule(Task p, int quietPeriod, Action... actions) {
    	return schedule(p, quietPeriod, Arrays.asList(actions));
522
    }
K
kohsuke 已提交
523

K
kohsuke 已提交
524
    /**
525
     * Cancels the item in the queue. If the item is scheduled more than once, cancels the first occurrence.
K
kohsuke 已提交
526
     *
527 528
     * @return true if the project was indeed in the queue and was removed.
     *         false if this was no-op.
K
kohsuke 已提交
529
     */
K
kohsuke 已提交
530
    public synchronized boolean cancel(Task p) {
K
kohsuke 已提交
531
        LOGGER.fine("Cancelling " + p.getFullDisplayName());
K
kohsuke 已提交
532 533
        for (Iterator<WaitingItem> itr = waitingList.iterator(); itr.hasNext();) {
            Item item = itr.next();
K
kohsuke 已提交
534
            if (item.task.equals(p)) {
K
kohsuke 已提交
535
                itr.remove();
536
                item.onCancelled();
K
kohsuke 已提交
537
                return true;
K
kohsuke 已提交
538 539
            }
        }
K
kohsuke 已提交
540
        // use bitwise-OR to make sure that both branches get evaluated all the time
541
        return blockedProjects.cancel(p)!=null | buildables.cancel(p)!=null;
K
kohsuke 已提交
542
    }
543 544 545
    
    public synchronized boolean cancel(Item item) {
        LOGGER.fine("Cancelling " + item.task.getFullDisplayName() + " item#" + item.id);
546 547 548 549 550
        // use bitwise-OR to make sure that all the branches get evaluated all the time
        boolean r = (item instanceof WaitingItem && waitingList.remove(item)) | blockedProjects.remove(item) | buildables.remove(item);
        if(r)
            item.onCancelled();
        return r;
551
    }
K
kohsuke 已提交
552 553

    public synchronized boolean isEmpty() {
554
        return waitingList.isEmpty() && blockedProjects.isEmpty() && buildables.isEmpty() && pendings.isEmpty();
K
kohsuke 已提交
555 556
    }

557
    private synchronized WaitingItem peek() {
558
        return waitingList.iterator().next();
K
kohsuke 已提交
559 560 561 562
    }

    /**
     * Gets a snapshot of items in the queue.
563 564 565
     *
     * Generally speaking the array is sorted such that the items that are most likely built sooner are
     * at the end.
K
kohsuke 已提交
566
     */
567
    @Exported(inline=true)
K
kohsuke 已提交
568
    public synchronized Item[] getItems() {
569
        Item[] r = new Item[waitingList.size() + blockedProjects.size() + buildables.size() + pendings.size()];
570 571
        waitingList.toArray(r);
        int idx = waitingList.size();
572 573
        for (BlockedItem p : blockedProjects.values())
            r[idx++] = p;
574
        for (BuildableItem p : reverse(buildables.values()))
575
            r[idx++] = p;
576 577
        for (BuildableItem p : reverse(pendings.values()))
            r[idx++] = p;
K
kohsuke 已提交
578 579
        return r;
    }
580 581 582 583 584
    
    public synchronized Item getItem(int id) {
    	for (Item item: waitingList) if (item.id == id) return item;
    	for (Item item: blockedProjects) if (item.id == id) return item;
    	for (Item item: buildables) if (item.id == id) return item;
585
        for (Item item: pendings) if (item.id == id) return item;
586 587
    	return null;
    }
K
kohsuke 已提交
588

K
kohsuke 已提交
589 590 591
    /**
     * Gets all the {@link BuildableItem}s that are waiting for an executor in the given {@link Computer}.
     */
592 593
    public synchronized List<BuildableItem> getBuildableItems(Computer c) {
        List<BuildableItem> result = new ArrayList<BuildableItem>();
594 595 596 597 598 599
        _getBuildableItems(c, buildables, result);
        _getBuildableItems(c, pendings, result);
        return result;
    }

    private void _getBuildableItems(Computer c, ItemList<BuildableItem> col, List<BuildableItem> result) {
600
        Node node = c.getNode();
601
        for (BuildableItem p : col.values()) {
602 603
            if (node.canTake(p.task) == null)
                result.add(p);
604 605 606
        }
    }

K
kohsuke 已提交
607
    /**
608
     * Gets the snapshot of all {@link BuildableItem}s.
K
kohsuke 已提交
609 610
     */
    public synchronized List<BuildableItem> getBuildableItems() {
611 612 613
        ArrayList<BuildableItem> r = new ArrayList<BuildableItem>(buildables.values());
        r.addAll(pendings.values());
        return r;
K
kohsuke 已提交
614 615
    }

616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
    /**
     * Gets the snapshot of all {@link BuildableItem}s.
     */
    public synchronized List<BuildableItem> getPendingItems() {
        return new ArrayList<BuildableItem>(pendings.values());
    }

    /**
     * Is the given task currently pending execution?
     */
    public synchronized boolean isPending(Task t) {
        for (BuildableItem i : pendings)
            if (i.task.equals(t))
                return true;
        return false;
    }

K
kohsuke 已提交
633 634 635 636 637 638 639 640
    /**
     * How many {@link BuildableItem}s are assigned for the given label?
     */
    public synchronized int countBuildableItemsFor(Label l) {
        int r = 0;
        for (BuildableItem bi : buildables.values())
            if(bi.task.getAssignedLabel()==l)
                r++;
641 642 643
        for (BuildableItem bi : pendings.values())
            if(bi.task.getAssignedLabel()==l)
                r++;
K
kohsuke 已提交
644 645 646
        return r;
    }

K
kohsuke 已提交
647 648 649 650 651
    /**
     * Gets the information about the queue item for the given project.
     *
     * @return null if the project is not in the queue.
     */
652 653
    public synchronized Item getItem(Task t) {
        BlockedItem bp = blockedProjects.get(t);
654
        if (bp!=null)
655 656
            return bp;
        BuildableItem bi = buildables.get(t);
657 658 659
        if(bi!=null)
            return bi;
        bi = pendings.get(t);
660
        if(bi!=null)
661 662
            return bi;

663
        for (Item item : waitingList) {
664
            if (item.task == t)
K
kohsuke 已提交
665 666 667 668 669
                return item;
        }
        return null;
    }

670 671 672 673 674 675 676 677 678
    /**
     * Gets the information about the queue item for the given project.
     *
     * @return null if the project is not in the queue.
     */
    public synchronized List<Item> getItems(Task t) {
    	List<Item> result =new ArrayList<Item>();
    	result.addAll(blockedProjects.getAll(t));
    	result.addAll(buildables.getAll(t));
679
        result.addAll(pendings.getAll(t));
680 681 682 683 684 685 686
        for (Item item : waitingList) {
            if (item.task == t)
                result.add(item);
        }
        return result;
    }

687 688
    /**
     * Left for backward compatibility.
689
     *
690 691
     * @see #getItem(Task)
    public synchronized Item getItem(AbstractProject p) {
692
        return getItem((Task) p);
693
    }
K
kohsuke 已提交
694
     */
695

K
kohsuke 已提交
696
    /**
K
kohsuke 已提交
697
     * Returns true if this queue contains the said project.
K
kohsuke 已提交
698
     */
699
    public synchronized boolean contains(Task t) {
700
        if (blockedProjects.containsKey(t) || buildables.containsKey(t) || pendings.containsKey(t))
K
kohsuke 已提交
701
            return true;
702
        for (Item item : waitingList) {
703
            if (item.task == t)
K
kohsuke 已提交
704 705 706 707 708 709 710
                return true;
        }
        return false;
    }

    /**
     * Called by the executor to fetch something to build next.
711
     * <p>
K
kohsuke 已提交
712 713
     * This method blocks until a next project becomes buildable.
     */
714
    public synchronized WorkUnit pop() throws InterruptedException {
K
kohsuke 已提交
715
        final Executor exec = Executor.currentExecutor();
716

K
kohsuke 已提交
717
        try {
718
            while (true) {
K
kohsuke 已提交
719 720 721
                final JobOffer offer = new JobOffer(exec);
                long sleep = -1;

722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737
                // consider myself parked
                assert !parked.containsKey(exec);
                parked.put(exec, offer);

                // reuse executor thread to do a queue maintenance.
                // at the end of this we get all the buildable jobs
                // in the buildables field.
                maintain();

                // allocate buildable jobs to executors
                Iterator<BuildableItem> itr = buildables.iterator();
                while (itr.hasNext()) {
                    BuildableItem p = itr.next();

                    // one last check to make sure this build is not blocked.
                    if (isBuildBlocked(p.task)) {
K
kohsuke 已提交
738
                        itr.remove();
739 740
                        blockedProjects.put(p.task,new BlockedItem(p));
                        continue;
K
kohsuke 已提交
741 742
                    }

743 744 745 746 747 748
                    JobOffer runner = loadBalancer.choose(p.task, new ApplicableJobOfferList(p.task));
                    if (runner == null)
                        // if we couldn't find the executor that fits,
                        // just leave it in the buildables list and
                        // check if we can execute other projects
                        continue;
K
kohsuke 已提交
749

750 751 752 753
                    assert runner.canTake(p.task);

                    // found a matching executor. use it.
                    runner.set(p);
754 755
                    itr.remove();
                    pendings.add(p);
756 757 758 759 760 761 762 763 764 765 766
                }

                // we went over all the buildable projects and awaken
                // all the executors that got work to do. now, go to sleep
                // until this thread is awakened. If this executor assigned a job to
                // itself above, the block method will return immediately.

                if (!waitingList.isEmpty()) {
                    // wait until the first item in the queue is due
                    sleep = peek().timestamp.getTimeInMillis() - new GregorianCalendar().getTimeInMillis();
                    if (sleep < 100) sleep = 100;    // avoid wait(0)
K
kohsuke 已提交
767 768
                }

769
                if (sleep == -1)
K
kohsuke 已提交
770 771 772 773
                    offer.event.block();
                else
                    offer.event.block(sleep);

774 775 776 777 778 779 780 781
                // retract the offer object
                assert parked.get(exec) == offer;
                parked.remove(exec);

                // am I woken up because I have a project to build?
                if (offer.item != null) {
                    // if so, just build it
                    LOGGER.fine("Pop returning " + offer.item + " for " + exec.getName());
782
                    pendings.remove(offer.item);
783 784 785

                    WorkUnitContext wuc = new WorkUnitContext(offer.item);
                    return wuc.createWorkUnit(offer.item.task);
K
kohsuke 已提交
786
                }
787
                // otherwise run a queue maintenance
K
kohsuke 已提交
788 789
            }
        } finally {
790 791 792 793 794 795 796 797 798 799
            // remove myself from the parked list
            JobOffer offer = parked.remove(exec);
            if (offer != null && offer.item != null) {
                // we are already assigned a project,
                // ask for someone else to build it.
                // note that while this thread is waiting for CPU
                // someone else can schedule this build again,
                // so check the contains method first.
                if (!contains(offer.item.task))
                    buildables.put(offer.item.task,offer.item);
K
kohsuke 已提交
800
            }
801 802 803 804 805 806

            // since this executor might have been chosen for
            // maintenance, schedule another one. Worst case
            // we'll just run a pointless maintenance, and that's
            // fine.
            scheduleMaintenance();
K
kohsuke 已提交
807 808 809 810
        }
    }

    /**
811
     * Represents a list of {@linkplain JobOffer#canTake(Task) applicable} {@link JobOffer}s
812
     * and provides various typical 
K
kohsuke 已提交
813
     */
814
    public final class ApplicableJobOfferList implements Iterable<JobOffer> {
815 816 817 818
        private final List<JobOffer> list;
        // laziy filled
        private Map<Node,List<JobOffer>> nodes;

819
        private ApplicableJobOfferList(Task task) {
820 821
            list = new ArrayList<JobOffer>(parked.size());
            for (JobOffer j : parked.values())
822
                if(j.canTake(task))
823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864
                    list.add(j);
        }

        /**
         * Returns all the {@linkplain JobOffer#isAvailable() available} {@link JobOffer}s.
         */
        public List<JobOffer> all() {
            return list;
        }

        public Iterator<JobOffer> iterator() {
            return list.iterator();
        }

        /**
         * List up all the {@link Node}s that have some available offers.
         */
        public Set<Node> nodes() {
            return byNodes().keySet();
        }

        /**
         * Gets a {@link JobOffer} for an executor of the given node, if any.
         * Otherwise null. 
         */
        public JobOffer _for(Node n) {
            List<JobOffer> r = byNodes().get(n);
            if(r==null) return null;
            return r.get(0);
        }

        public Map<Node,List<JobOffer>> byNodes() {
            if(nodes==null) {
                nodes = new HashMap<Node,List<JobOffer>>();
                for (JobOffer o : list) {
                    List<JobOffer> l = nodes.get(o.getNode());
                    if(l==null) nodes.put(o.getNode(),l=new ArrayList<JobOffer>());
                    l.add(o);
                }
            }
            return nodes;
        }
K
kohsuke 已提交
865 866 867 868
    }

    /**
     * Checks the queue and runs anything that can be run.
869 870
     *
     * <p>
K
kohsuke 已提交
871
     * When conditions are changed, this method should be invoked.
872
     * <p>
K
kohsuke 已提交
873 874 875 876 877 878 879
     * This wakes up one {@link Executor} so that it will maintain a queue.
     */
    public synchronized void scheduleMaintenance() {
        // this code assumes that after this method is called
        // no more executors will be offered job except by
        // the pop() code.
        for (Entry<Executor, JobOffer> av : parked.entrySet()) {
880
            if (av.getValue().item == null) {
K
kohsuke 已提交
881 882 883 884 885 886
                av.getValue().event.signal();
                return;
            }
        }
    }

887 888 889
    /**
     * Checks if the given task is blocked.
     */
890
    private boolean isBuildBlocked(Task t) {
891 892 893
        return t.isBuildBlocked() || !canRun(t.getResourceList());
    }

894
    /**
895 896
     * Make sure we don't queue two tasks of the same project to be built
     * unless that project allows concurrent builds.
897
     */
898 899 900 901 902 903 904
    private boolean allowNewBuildableTask(Task t) {
        try {
            if (t.isConcurrentBuild())
                return true;
        } catch (AbstractMethodError e) {
            // earlier versions don't have the "isConcurrentBuild" method, so fall back gracefully
        }
905
        return !buildables.containsKey(t) && !pendings.containsKey(t);
906
    }
907

K
kohsuke 已提交
908
    /**
K
kohsuke 已提交
909
     * Queue maintenance.
910
     * <p>
911
     * Move projects between {@link #waitingList}, {@link #blockedProjects}, and {@link #buildables}
K
kohsuke 已提交
912 913
     * appropriately.
     */
914
    public synchronized void maintain() {
915 916
        if (LOGGER.isLoggable(Level.FINE))
            LOGGER.fine("Queue maintenance started " + this);
917

918
        Iterator<BlockedItem> itr = blockedProjects.values().iterator();
919
        while (itr.hasNext()) {
920
            BlockedItem p = itr.next();
921
            if (!isBuildBlocked(p.task) && allowNewBuildableTask(p.task)) {
K
kohsuke 已提交
922
                // ready to be executed
K
kohsuke 已提交
923
                LOGGER.fine(p.task.getFullDisplayName() + " no longer blocked");
K
kohsuke 已提交
924
                itr.remove();
925
                makeBuildable(new BuildableItem(p));
K
kohsuke 已提交
926 927 928
            }
        }

929
        while (!waitingList.isEmpty()) {
930
            WaitingItem top = peek();
K
kohsuke 已提交
931

932
            if (!top.timestamp.before(new GregorianCalendar()))
K
kohsuke 已提交
933 934
                return; // finished moving all ready items from queue

935
            waitingList.remove(top);
936
            Task p = top.task;
937
            if (!isBuildBlocked(p) && allowNewBuildableTask(p)) {
K
kohsuke 已提交
938
                // ready to be executed immediately
K
kohsuke 已提交
939
                LOGGER.fine(p.getFullDisplayName() + " ready to build");
940
                makeBuildable(new BuildableItem(top));
K
kohsuke 已提交
941
            } else {
942
                // this can't be built now because another build is in progress
K
kohsuke 已提交
943
                // set this project aside.
K
kohsuke 已提交
944
                LOGGER.fine(p.getFullDisplayName() + " is blocked");
945
                blockedProjects.put(p,new BlockedItem(top));
K
kohsuke 已提交
946 947
            }
        }
948 949 950 951

        final QueueSorter s = sorter;
        if (s != null)
        	s.sortBuildableItems(buildables);
K
kohsuke 已提交
952 953
    }

954
    private void makeBuildable(BuildableItem p) {
955
        if(Hudson.FLYWEIGHT_SUPPORT && p.task instanceof FlyweightTask && (!Hudson.getInstance().isQuietingDown() || p.task instanceof NonBlockingTask)) {
956 957 958 959 960 961 962 963 964
            ConsistentHash<Node> hash = new ConsistentHash<Node>(new Hash<Node>() {
                public String hash(Node node) {
                    return node.getNodeName();
                }
            });
            Hudson h = Hudson.getInstance();
            hash.add(h, h.getNumExecutors()*100);
            for (Node n : h.getNodes())
                hash.add(n,n.getNumExecutors()*100);
965 966

            Label lbl = p.task.getAssignedLabel();
967 968
            for (Node n : hash.list(p.task.getFullDisplayName())) {
                Computer c = n.toComputer();
K
kohsuke 已提交
969
                if (c==null || c.isOffline())    continue;
970
                if (lbl!=null && !lbl.contains(n))  continue;
971
                c.startFlyWeightTask(new WorkUnitContext(p).createWorkUnit(p.task));
972 973 974 975 976 977 978 979 980
                return;
            }
            // if the execution get here, it means we couldn't schedule it anywhere.
            // so do the scheduling like other normal jobs.
        }
        
        buildables.put(p.task,p);
    }

981 982 983 984
    public Api getApi() {
        return new Api(this);
    }

K
kohsuke 已提交
985 986 987 988 989 990
    /**
     * Marks {@link Task}s that are not persisted.
     * @since 1.311
     */
    public interface TransientTask extends Task {}

991 992 993 994 995 996 997
    /**
     * Marks {@link Task}s that do not consume {@link Executor}.
     * @see OneOffExecutor
     * @since 1.318
     */
    public interface FlyweightTask extends Task {}

K
kohsuke 已提交
998
    /**
999
     * Marks {@link Task}s that are not affected by the {@linkplain Hudson#isQuietingDown()}  quieting down},
K
kohsuke 已提交
1000 1001 1002 1003 1004 1005
     * because these tasks keep other tasks executing.
     *
     * @since 1.336 
     */
    public interface NonBlockingTask extends Task {}

K
kohsuke 已提交
1006 1007
    /**
     * Task whose execution is controlled by the queue.
1008
     *
1009
     * <p>
K
kohsuke 已提交
1010 1011 1012
     * {@link #equals(Object) Value equality} of {@link Task}s is used
     * to collapse two tasks into one. This is used to avoid infinite
     * queue backlog.
1013 1014 1015
     *
     * <p>
     * Pending {@link Task}s are persisted when Hudson shuts down, so
K
kohsuke 已提交
1016 1017
     * it needs to be persistable via XStream. To create a non-persisted
     * transient Task, extend {@link TransientTask} marker interface.
1018 1019 1020 1021 1022
     *
     * <p>
     * Plugins are encouraged to extend from {@link AbstractQueueTask}
     * instead of implementing this interface directly, to maintain
     * compatibility with future changes to this interface.
K
kohsuke 已提交
1023
     */
1024
    public interface Task extends ModelObject, ExecutionUnit {
1025
        /**
1026 1027 1028
         * If this task needs to be run on a node with a particular label,
         * return that {@link Label}. Otherwise null, indicating
         * it can run on anywhere.
1029
         */
1030
        Label getAssignedLabel();
1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041

        /**
         * If the previous execution of this task run on a certain node
         * and this task prefers to run on the same node, return that.
         * Otherwise null.
         */
        Node getLastBuiltOn();

        /**
         * Returns true if the execution should be blocked
         * for temporary reasons.
1042 1043
         *
         * <p>
1044
         * Short-hand for {@code getCauseOfBlockage()!=null}.
1045 1046 1047 1048
         */
        boolean isBuildBlocked();

        /**
1049 1050
         * @deprecated as of 1.330
         *      Use {@link CauseOfBlockage#getShortDescription()} instead.
1051 1052 1053
         */
        String getWhyBlocked();

1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066
        /**
         * If the execution of this task should be blocked for temporary reasons,
         * this method returns a non-null object explaining why.
         *
         * <p>
         * Otherwise this method returns null, indicating that the build can proceed right away.
         *
         * <p>
         * This can be used to define mutual exclusion that goes beyond
         * {@link #getResourceList()}.
         */
        CauseOfBlockage getCauseOfBlockage();

1067
        /**
K
kohsuke 已提交
1068
         * Unique name of this task.
K
kohsuke 已提交
1069
         *
K
kohsuke 已提交
1070 1071
         * <p>
         * This method is no longer used, left here for compatibility. Just return {@link #getDisplayName()}.
1072 1073 1074
         */
        String getName();

1075 1076 1077 1078 1079
        /**
         * @see hudson.model.Item#getFullDisplayName()
         */
        String getFullDisplayName();

1080 1081 1082 1083
        /**
         * Estimate of how long will it take to execute this task.
         * Measured in milliseconds.
         *
1084
         * @return -1 if it's impossible to estimate.
1085 1086 1087
         */
        long getEstimatedDuration();

K
kohsuke 已提交
1088
        /**
1089
         * Creates {@link Executable}, which performs the actual execution of the task.
K
kohsuke 已提交
1090
         */
1091
        Executable createExecutable() throws IOException;
1092 1093 1094 1095 1096 1097 1098 1099

        /**
         * Checks the permission to see if the current user can abort this executable.
         * Returns normally from this method if it's OK.
         *
         * @throws AccessDeniedException if the permission is not granted.
         */
        void checkAbortPermission();
K
kohsuke 已提交
1100 1101 1102 1103 1104 1105

        /**
         * Works just like {@link #checkAbortPermission()} except it indicates the status by a return value,
         * instead of exception.
         */
        boolean hasAbortPermission();
1106 1107
        
        /**
K
kohsuke 已提交
1108 1109 1110 1111 1112 1113 1114 1115
         * Returns the URL of this task relative to the context root of the application.
         *
         * <p>
         * When the user clicks an item in the queue, this is the page where the user is taken to.
         * Hudson expects the current instance to be bound to the URL returned by this method.
         *
         * @return
         *      URL that ends with '/'.
1116 1117
         */
        String getUrl();
1118 1119 1120
        
        /**
         * True if the task allows concurrent builds
1121 1122
         *
         * @since 1.338
1123 1124
         */
        boolean isConcurrentBuild();
1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141

        Collection<? extends ExecutionUnit> getMemberExecutionUnits();
    }

    public interface ExecutionUnit extends ResourceActivity {
        /**
         * Estimate of how long will it take to execute this task.
         * Measured in milliseconds.
         *
         * @return -1 if it's impossible to estimate.
         */
        long getEstimatedDuration();

        /**
         * Creates {@link Executable}, which performs the actual execution of the task.
         */
        Executable createExecutable() throws IOException;
1142 1143 1144 1145 1146
    }

    public interface Executable extends Runnable {
        /**
         * Task from which this executable was created.
K
kohsuke 已提交
1147
         * Never null.
1148 1149 1150 1151 1152 1153 1154
         */
        Task getParent();

        /**
         * Called by {@link Executor} to perform the task
         */
        void run();
1155 1156 1157 1158

        /**
         * Used to render the HTML. Should be a human readable text of what this executable is.
         */
1159
        @Override String toString();
1160 1161
    }

1162
    /*package*/
1163

K
kohsuke 已提交
1164 1165 1166
    /**
     * Item in a queue.
     */
1167
    @ExportedBean(defaultVisibility = 999)
1168
    public static abstract class Item extends Actionable {
1169 1170 1171 1172
        /**
         * VM-wide unique ID that tracks the {@link Task} as it moves through different stages
         * in the queue (each represented by different subtypes of {@link Item}.
         */
1173 1174 1175
    	public final int id;
    	
		/**
K
kohsuke 已提交
1176 1177
         * Project to be built.
         */
1178
        @Exported
1179
        public final Task task;
1180

1181
        private /*almost final*/ transient FutureImpl future;
1182

1183
        /**
1184 1185 1186
         * Build is blocked because another build is in progress,
         * required {@link Resource}s are not available, or otherwise blocked
         * by {@link Task#isBuildBlocked()}.
1187
         */
K
kohsuke 已提交
1188
        @Exported
1189
        public boolean isBlocked() { return this instanceof BlockedItem; }
1190 1191 1192 1193 1194 1195

        /**
         * Build is waiting the executor to become available.
         * This flag is only used in {@link Queue#getItems()} for
         * 'pseudo' items that are actually not really in the queue.
         */
K
kohsuke 已提交
1196
        @Exported
1197
        public boolean isBuildable() { return this instanceof BuildableItem; }
1198

1199 1200 1201 1202 1203 1204
        /**
         * True if the item is starving for an executor for too long.
         */
        @Exported
        public boolean isStuck() { return false; }

1205 1206 1207 1208 1209 1210 1211
        /**
         * Can be used to wait for the completion (either normal, abnormal, or cancellation) of the {@link Task}.
         * <p>
         * Just like {@link #id}, the same object tracks various stages of the queue.
         */
        public Future<Executable> getFuture() { return future; }

1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
        /**
         * Convenience method that returns a read only view of the {@link Cause}s associated with this item in the queue.
         *
         * @return can be empty but never null
         * @since 1.343
         */
        public final List<Cause> getCauses() {
            CauseAction ca = getAction(CauseAction.class);
            if (ca!=null)
                return Collections.unmodifiableList(ca.getCauses());
            return Collections.emptyList();
        }

1225
        protected Item(Task task, List<Action> actions, int id, FutureImpl future) {
1226 1227
            this.task = task;
            this.id = id;
1228
            this.future = future;
1229 1230 1231 1232
            for (Action action: actions) addAction(action);
        }
        
        protected Item(Item item) {
1233
        	this(item.task, item.getActions(), item.id, item.future);
K
kohsuke 已提交
1234 1235
        }

1236
        /**
1237
         * Gets a human-readable status message describing why it's in the queue.
1238
         */
K
kohsuke 已提交
1239
        @Exported
1240 1241 1242 1243 1244
        public final String getWhy() {
            CauseOfBlockage cob = getCauseOfBlockage();
            return cob!=null ? cob.getShortDescription() : null;
        }

K
kohsuke 已提交
1245 1246 1247
        /**
         * Gets an object that describes why this item is in the queue.
         */
1248 1249
        public abstract CauseOfBlockage getCauseOfBlockage();

1250 1251 1252 1253 1254 1255
        /**
         * Gets a human-readable message about the parameters of this item
         * @return String
         */
        @Exported
        public String getParams() {
K
kohsuke 已提交
1256
        	StringBuilder s = new StringBuilder();
1257 1258 1259 1260
        	for(Action action : getActions()) {
        		if(action instanceof ParametersAction) {
        			ParametersAction pa = (ParametersAction)action;
        			for (ParameterValue p : pa.getParameters()) {
K
kohsuke 已提交
1261
        				s.append('\n').append(p.getShortDescription());
1262 1263 1264
        			}
        		}
        	}
K
kohsuke 已提交
1265
        	return s.toString();
1266 1267
        }
        
1268 1269 1270
        public boolean hasCancelPermission() {
            return task.hasAbortPermission();
        }
1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288
        
        public String getDisplayName() {
			// TODO Auto-generated method stub
			return null;
		}

		public String getSearchUrl() {
			// TODO Auto-generated method stub
			return null;
		}

        /**
         * Called from queue.jelly.
         */
        public void doCancelQueue( StaplerRequest req, StaplerResponse rsp ) throws IOException, ServletException {
        	Hudson.getInstance().getQueue().cancel(this);
            rsp.forwardToPreviousPage(req);
        }
1289 1290 1291 1292 1293 1294 1295 1296 1297

        /**
         * Participates in the cancellation logic to set the {@link #future} accordingly.
         */
        /*package*/ void onCancelled() {
            future.setAsCancelled();
        }

        private Object readResolve() {
1298
            this.future = new FutureImpl(task);
1299 1300
            return this;
        }
K
kohsuke 已提交
1301 1302 1303 1304 1305

        @Override
        public String toString() {
            return getClass().getName()+':'+task.toString();
        }
1306 1307 1308 1309 1310
    }
    
    /**
     * An optional interface for actions on Queue.Item.
     * Lets the action cooperate in queue management.
1311 1312
     * 
     * @since 1.300-ish.
1313 1314 1315 1316 1317 1318 1319
     */
    public interface QueueAction extends Action {
    	/**
    	 * Returns whether the new item should be scheduled. 
    	 * An action should return true if the associated task is 'different enough' to warrant a separate execution.
    	 */
    	public boolean shouldSchedule(List<Action> actions);
1320
    }
1321

1322
    /**
K
kohsuke 已提交
1323 1324 1325 1326 1327 1328
     * Extension point for deciding if particular job should be scheduled or not.
     *
     * <p>
     * This handler is consulted every time someone tries to submit a task to the queue.
     * If any of the registered handlers returns false, the task will not be added
     * to the queue, and the task will never get executed. 
1329 1330 1331 1332 1333 1334 1335 1336 1337
     *
     * <p>
     * This extension point is still a subject to change, as we are seeking more
     * comprehensive Queue pluggability. See HUDSON-2072.
     *
     * @since 1.316
     */
    public static abstract class QueueDecisionHandler implements ExtensionPoint {
    	/**
K
kohsuke 已提交
1338
    	 * Returns whether the new item should be scheduled.
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350
    	 */
    	public abstract boolean shouldSchedule(Task p, List<Action> actions);
    	    	
    	/**
    	 * All registered {@link QueueDecisionHandler}s
    	 * @return
    	 */
    	public static ExtensionList<QueueDecisionHandler> all() {
    		return Hudson.getInstance().getExtensionList(QueueDecisionHandler.class);
    	}
    }
    
1351 1352 1353
    /**
     * {@link Item} in the {@link Queue#waitingList} stage.
     */
1354 1355 1356
    public static final class WaitingItem extends Item implements Comparable<WaitingItem> {
    	private static final AtomicInteger COUNTER = new AtomicInteger(0);
    	
1357 1358 1359 1360 1361
        /**
         * This item can be run after this time.
         */
        @Exported
        public Calendar timestamp;
1362

1363
        WaitingItem(Calendar timestamp, Task project, List<Action> actions) {
1364
            super(project, actions, COUNTER.incrementAndGet(), new FutureImpl(project));
1365 1366
            this.timestamp = timestamp;
        }
1367
        
1368 1369 1370 1371 1372 1373 1374
        public int compareTo(WaitingItem that) {
            int r = this.timestamp.getTime().compareTo(that.timestamp.getTime());
            if (r != 0) return r;

            return this.id - that.id;
        }

1375
        public CauseOfBlockage getCauseOfBlockage() {
1376
            long diff = timestamp.getTimeInMillis() - System.currentTimeMillis();
1377
            if (diff > 0)
1378
                return CauseOfBlockage.fromMessage(Messages._Queue_InQuietPeriod(Util.getTimeSpanString(diff)));
1379
            else
1380
                return CauseOfBlockage.fromMessage(Messages._Queue_Unknown());
1381 1382
        }
    }
K
kohsuke 已提交
1383

1384 1385 1386
    /**
     * Common part between {@link BlockedItem} and {@link BuildableItem}.
     */
1387
    public static abstract class NotWaitingItem extends Item {
1388 1389 1390 1391 1392 1393 1394
        /**
         * When did this job exit the {@link Queue#waitingList} phase?
         */
        @Exported
        public final long buildableStartMilliseconds;

        protected NotWaitingItem(WaitingItem wi) {
1395
            super(wi);
1396
            buildableStartMilliseconds = System.currentTimeMillis();
K
kohsuke 已提交
1397
        }
1398

1399
        protected NotWaitingItem(NotWaitingItem ni) {
1400
            super(ni);
1401
            buildableStartMilliseconds = ni.buildableStartMilliseconds;
1402
        }
1403
    }
1404

1405 1406 1407 1408 1409 1410 1411
    /**
     * {@link Item} in the {@link Queue#blockedProjects} stage.
     */
    public final class BlockedItem extends NotWaitingItem {
        public BlockedItem(WaitingItem wi) {
            super(wi);
        }
1412

1413 1414 1415 1416
        public BlockedItem(NotWaitingItem ni) {
            super(ni);
        }

1417
        public CauseOfBlockage getCauseOfBlockage() {
1418 1419 1420
            ResourceActivity r = getBlockingActivity(task);
            if (r != null) {
                if (r == task) // blocked by itself, meaning another build is in progress
1421 1422
                    return CauseOfBlockage.fromMessage(Messages._Queue_InProgress());
                return CauseOfBlockage.fromMessage(Messages._Queue_BlockedBy(r.getDisplayName()));
1423
            }
1424
            return task.getCauseOfBlockage();
1425
        }
1426
    }
1427

1428 1429 1430
    /**
     * {@link Item} in the {@link Queue#buildables} stage.
     */
1431
    public final static class BuildableItem extends NotWaitingItem {
1432 1433 1434 1435 1436 1437 1438 1439
        public BuildableItem(WaitingItem wi) {
            super(wi);
        }

        public BuildableItem(NotWaitingItem ni) {
            super(ni);
        }

1440
        public CauseOfBlockage getCauseOfBlockage() {
1441
            Hudson hudson = Hudson.getInstance();
K
kohsuke 已提交
1442
            if(hudson.isQuietingDown() && !(task instanceof NonBlockingTask))
1443
                return CauseOfBlockage.fromMessage(Messages._Queue_HudsonIsAboutToShutDown());
K
kohsuke 已提交
1444 1445

            Label label = task.getAssignedLabel();
K
kohsuke 已提交
1446
            if (hudson.getNodes().isEmpty())
1447
                label = null;    // no master/slave. pointless to talk about nodes
1448

1449 1450
            if (label != null) {
                if (label.isOffline()) {
1451
                    Set<Node> nodes = label.getNodes();
1452
                    if (nodes.size() != 1)      return new BecauseLabelIsOffline(label);
1453
                    else                        return new BecauseNodeIsOffline(nodes.iterator().next());
1454 1455 1456
                }
            }

1457
            if(label==null)
1458
                return CauseOfBlockage.fromMessage(Messages._Queue_WaitingForNextAvailableExecutor());
1459 1460

            Set<Node> nodes = label.getNodes();
1461
            if (nodes.size() != 1)      return new BecauseLabelIsBusy(label);
1462
            else                        return new BecauseNodeIsBusy(nodes.iterator().next());
1463
        }
1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474

        @Override
        public boolean isStuck() {
            Label label = task.getAssignedLabel();
            if(label!=null && label.isOffline())
                // no executor online to process this job. definitely stuck.
                return true;

            long d = task.getEstimatedDuration();
            long elapsed = System.currentTimeMillis()-buildableStartMilliseconds;
            if(d>=0) {
1475
                // if we were running elsewhere, we would have done this build ten times.
1476
                return elapsed > Math.max(d,60000L)*10;
1477 1478
            } else {
                // more than a day in the queue
1479
                return TimeUnit2.MILLISECONDS.toHours(elapsed)>24;
1480 1481
            }
        }
K
kohsuke 已提交
1482 1483 1484
    }

    private static final Logger LOGGER = Logger.getLogger(Queue.class.getName());
1485

1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527
    /**
     * This {@link XStream} instance is used to persist {@link Task}s.
     */
    public static final XStream XSTREAM = new XStream2();

    static {
        XSTREAM.registerConverter(new AbstractSingleValueConverter() {

			@Override
			@SuppressWarnings("unchecked")
			public boolean canConvert(Class klazz) {
				return hudson.model.Item.class.isAssignableFrom(klazz);
			}

			@Override
			public Object fromString(String string) {
                Object item = Hudson.getInstance().getItemByFullName(string);
                if(item==null)  throw new NoSuchElementException("No such job exists: "+string);
                return item;
			}

			@Override
			public String toString(Object item) {
				return ((hudson.model.Item) item).getFullName();
			}
        });
        XSTREAM.registerConverter(new AbstractSingleValueConverter() {

			@SuppressWarnings("unchecked")
			@Override
			public boolean canConvert(Class klazz) {
				return Run.class.isAssignableFrom(klazz);
			}

			@Override
			public Object fromString(String string) {
				String[] split = string.split("#");
				String projectName = split[0];
				int buildNumber = Integer.parseInt(split[1]);
				Job<?,?> job = (Job<?,?>) Hudson.getInstance().getItemByFullName(projectName);
                if(job==null)  throw new NoSuchElementException("No such job exists: "+projectName);
				Run<?,?> run = job.getBuildByNumber(buildNumber);
K
kohsuke 已提交
1528
                if(run==null)  throw new NoSuchElementException("No such build: "+string);
1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539
				return run;
			}

			@Override
			public String toString(Object object) {
				Run<?,?> run = (Run<?,?>) object;
				return run.getParent().getFullName() + "#" + run.getNumber();
			}
        });
    }

1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555
    /**
     * Regularly invokes {@link Queue#maintain()} and clean itself up when
     * {@link Queue} gets GC-ed.
     */
    private static class MaintainTask extends SafeTimerTask {
        private final WeakReference<Queue> queue;

        MaintainTask(Queue queue) {
            this.queue = new WeakReference<Queue>(queue);

            long interval = 5 * Timer.ONE_SECOND;
            Trigger.timer.schedule(this, interval, interval);
        }

        protected void doRun() {
            Queue q = queue.get();
1556
            if (q != null)
1557 1558 1559 1560 1561
                q.maintain();
            else
                cancel();
        }
    }
1562 1563
    
    /**
1564
     * {@link ArrayList} of {@link Item} with more convenience methods.
1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609
     */
    private static class ItemList<T extends Item> extends ArrayList<T> {
    	public T get(Task task) {
    		for (T item: this) {
    			if (item.task == task) {
    				return item;
    			}
    		}
    		return null;
    	}
    	
    	public List<T> getAll(Task task) {
    		List<T> result = new ArrayList<T>();
    		for (T item: this) {
    			if (item.task == task) {
    				result.add(item);
    			}
    		}
    		return result;
    	}
    	
    	public boolean containsKey(Task task) {
    		return get(task) != null;
    	}
    	
    	public T remove(Task task) {
    		Iterator<T> it = iterator();
    		while (it.hasNext()) {
    			T t = it.next();
    			if (t.task == task) {
    				it.remove();
    				return t;
    			}
    		}
    		return null;
    	}
    	
    	public void put(Task task, T item) {
    		assert item.task == task;
    		add(item);
    	}
    	
    	public ItemList<T> values() {
    		return this;
    	}
1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633

        /**
         * Works like {@link #remove(Task)} but also marks the {@link Item} as cancelled.
         */
        public T cancel(Task p) {
            T x = remove(p);
            if(x!=null) x.onCancelled();
            return x;
        }

        /**
         * Works like {@link #remove(Object)} but also marks the {@link Item} as cancelled.
         */
        public boolean cancel(Item t) {
            boolean r = remove(t);
            if(r)   t.onCancelled();
            return r;
        }

        public void cancelAll() {
            for (T t : this)
                t.onCancelled();
            clear();
        }
1634
    }
1635 1636 1637 1638 1639

    @CLIResolver
    public static Queue getInstance() {
        return Hudson.getInstance().getQueue();
    }
K
kohsuke 已提交
1640 1641 1642 1643 1644 1645 1646 1647

    /**
     * Restores the queue content during the start up.
     */
    @Initializer(after=JOB_LOADED)
    public static void init(Hudson h) {
        h.getQueue().load();
    }
K
kohsuke 已提交
1648
}