Queue.java 39.7 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
import hudson.Util;
K
kohsuke 已提交
28
import hudson.XmlFile;
K
kohsuke 已提交
29
import hudson.model.Node.Mode;
30 31
import hudson.triggers.SafeTimerTask;
import hudson.triggers.Trigger;
K
kohsuke 已提交
32
import hudson.util.OneShotEvent;
33
import hudson.util.TimeUnit2;
34
import hudson.util.XStream2;
K
kohsuke 已提交
35

36 37 38 39 40
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
41
import java.lang.ref.WeakReference;
K
kohsuke 已提交
42
import java.util.ArrayList;
43
import java.util.Arrays;
K
kohsuke 已提交
44 45 46 47 48 49
import java.util.Calendar;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
50
import java.util.NoSuchElementException;
K
kohsuke 已提交
51 52
import java.util.Set;
import java.util.TreeSet;
S
stephenconnolly 已提交
53
import java.util.Map.Entry;
54
import java.util.concurrent.atomic.AtomicInteger;
K
kohsuke 已提交
55 56 57
import java.util.logging.Level;
import java.util.logging.Logger;

K
kohsuke 已提交
58
import javax.management.timer.Timer;
59
import javax.servlet.ServletException;
K
kohsuke 已提交
60 61

import org.acegisecurity.AccessDeniedException;
62 63
import org.kohsuke.stapler.StaplerRequest;
import org.kohsuke.stapler.StaplerResponse;
K
kohsuke 已提交
64 65 66 67 68 69
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 已提交
70 71
/**
 * Build queue.
72 73
 *
 * <p>
74 75
 * 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}
76
 * so that we can keep track of additional data used for deciding what to exeucte when.
77 78
 *
 * <p>
79 80 81 82 83 84 85 86
 * Items in queue goes through several stages, as depicted below:
 * <pre>
 * (enter) --> waitingList --+--> blockedProjects
 *                           |        ^
 *                           |        |
 *                           |        v
 *                           +--> buildables ---> (executed)
 * </pre>
87 88
 *
 * <p>
89 90 91
 * 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 已提交
92 93
 * @author Kohsuke Kawaguchi
 */
94
@ExportedBean
95
public class Queue extends ResourceController implements Saveable {
K
kohsuke 已提交
96
    /**
97
     * Items that are waiting for its quiet period to pass.
98 99
     *
     * <p>
K
kohsuke 已提交
100 101 102
     * This consists of {@link Item}s that cannot be run yet
     * because its time has not yet come.
     */
103
    private final Set<WaitingItem> waitingList = new TreeSet<WaitingItem>();
K
kohsuke 已提交
104 105

    /**
106
     * {@link Task}s that can be built immediately
107 108 109
     * but blocked because another build is in progress,
     * required {@link Resource}s are not available, or otherwise blocked
     * by {@link Task#isBuildBlocked()}.
K
kohsuke 已提交
110
     */
111
    private final ItemList<BlockedItem> blockedProjects = new ItemList<BlockedItem>();
K
kohsuke 已提交
112 113

    /**
114
     * {@link Task}s that can be built immediately
K
kohsuke 已提交
115 116
     * that are waiting for available {@link Executor}.
     */
117
    private final ItemList<BuildableItem> buildables = new ItemList<BuildableItem>();
118

K
kohsuke 已提交
119 120
    /**
     * Data structure created for each idle {@link Executor}.
121
     * This is a job offer from the queue to an executor.
122 123
     *
     * <p>
K
kohsuke 已提交
124 125 126
     * 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 已提交
127
     */
K
kohsuke 已提交
128 129
    public static class JobOffer {
        public final Executor executor;
K
kohsuke 已提交
130 131 132 133 134

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

K
kohsuke 已提交
137 138 139 140
        /**
         * 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 已提交
141
        private BuildableItem item;
K
kohsuke 已提交
142

K
kohsuke 已提交
143
        private JobOffer(Executor executor) {
K
kohsuke 已提交
144 145 146
            this.executor = executor;
        }

147 148 149
        public void set(BuildableItem p) {
            assert this.item == null;
            this.item = p;
K
kohsuke 已提交
150 151 152 153
            event.signal();
        }

        public boolean isAvailable() {
154
            return item == null && !executor.getOwner().isOffline() && executor.getOwner().isAcceptingTasks();
K
kohsuke 已提交
155 156 157 158 159 160 161
        }

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

        public boolean isNotExclusive() {
162
            return getNode().getMode() == Mode.NORMAL;
K
kohsuke 已提交
163 164 165
        }
    }

S
stephenconnolly 已提交
166
    /**
K
kohsuke 已提交
167
     * The executors that are currently waiting for a job to run.
S
stephenconnolly 已提交
168
     */
K
kohsuke 已提交
169
    private final Map<Executor,JobOffer> parked = new HashMap<Executor,JobOffer>();
K
kohsuke 已提交
170

171 172 173 174
    private final LoadBalancer loadBalancer;

    public Queue(LoadBalancer loadBalancer) {
        this.loadBalancer =  loadBalancer.sanitize();
175 176 177 178 179
        // 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);
    }

K
kohsuke 已提交
180 181 182 183 184
    /**
     * Loads the queue contents that was {@link #save() saved}.
     */
    public synchronized void load() {
        try {
K
TAB->WS  
kohsuke 已提交
185
            // first try the old format
K
kohsuke 已提交
186
            File queueFile = getQueueFile();
K
kohsuke 已提交
187 188
            if (queueFile.exists()) {
                BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(queueFile)));
K
TAB->WS  
kohsuke 已提交
189 190 191 192
                String line;
                while ((line = in.readLine()) != null) {
                    AbstractProject j = Hudson.getInstance().getItemByFullName(line, AbstractProject.class);
                    if (j != null)
M
mindless 已提交
193
                        j.scheduleBuild();
K
TAB->WS  
kohsuke 已提交
194 195 196 197 198 199 200
                }
                in.close();
                // discard the queue file now that we are done
                queueFile.delete();
            } else {
                queueFile = getXMLQueueFile();
                if (queueFile.exists()) {
201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223
                    List list = (List) new XmlFile(XSTREAM, queueFile).read();
                    if (!list.isEmpty()) {
                    	if (list.get(0) instanceof Queue.Task) {
                    		// backward compatiblity
                    		for (Task task : (List<Task>) list) {
                    			add(task, 0);
                    		}
                    	} else if (list.get(0) instanceof Item) {
                    		int maxId = 0;
                    		for (Item item: (List<Item>) list) {
                    			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);
                    			}
                    		}
                    		WaitingItem.COUNTER.set(maxId);
                    	}
K
TAB->WS  
kohsuke 已提交
224
                    }
225 226 227 228 229 230 231 232 233

                    // 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 已提交
234 235 236
                    queueFile.delete();
                }
            }
237 238
        } catch (IOException e) {
            LOGGER.log(Level.WARNING, "Failed to load the queue file " + getQueueFile(), e);
K
kohsuke 已提交
239 240 241 242 243 244 245
        }
    }

    /**
     * Persists the queue contents to the disk.
     */
    public synchronized void save() {
246 247
        if(BulkChange.contains(this))  return;
        
K
kohsuke 已提交
248
        // write out the tasks on the queue
249
    	ArrayList<Queue.Item> items = new ArrayList<Queue.Item>();
K
kohsuke 已提交
250
    	for (Item item: getItems()) {
251
    	    items.add(item);
K
kohsuke 已提交
252 253
    	}
    	
K
kohsuke 已提交
254
        try {
255
            new XmlFile(XSTREAM, getXMLQueueFile()).write(items);
256 257
        } catch (IOException e) {
            LOGGER.log(Level.WARNING, "Failed to write out the queue file " + getQueueFile(), e);
K
kohsuke 已提交
258 259 260
        }
    }

261 262 263 264 265 266 267 268 269 270
    /**
     * Wipes out all the items currently in the queue, as if all of them are cancelled at once.
     */
    public synchronized void clear() {
        waitingList.clear();
        blockedProjects.clear();
        buildables.clear();
        scheduleMaintenance();
    }

K
kohsuke 已提交
271
    private File getQueueFile() {
K
TAB->WS  
kohsuke 已提交
272 273
        return new File(Hudson.getInstance().getRootDir(), "queue.txt");
    }
K
kohsuke 已提交
274

275
    /*package*/ File getXMLQueueFile() {
K
TAB->WS  
kohsuke 已提交
276 277
        return new File(Hudson.getInstance().getRootDir(), "queue.xml");
    }
K
kohsuke 已提交
278 279 280

    /**
     * Schedule a new build for this project.
281
     *
282 283 284
     * @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 已提交
285
     */
286 287
    public boolean add(AbstractProject p) {
        return add(p, p.getQuietPeriod());
288 289 290 291
    }

    /**
     * Schedules a new build with a custom quiet period.
292 293
     *
     * <p>
K
kohsuke 已提交
294 295
     * Left for backward compatibility with &lt;1.114.
     *
296 297
     * @since 1.105
     */
298 299
    public synchronized boolean add(AbstractProject p, int quietPeriod) {
        return add((Task) p, quietPeriod);
K
kohsuke 已提交
300
    }
301
    
K
kohsuke 已提交
302 303 304
    /**
     * Schedules an execution of a task.
     *
305 306 307
     * @param quietPeriod Number of seconds that the task will be placed in queue.
     *                    Useful when the same task is likely scheduled for multiple
     *                    times.
308 309 310
     * @return true if the project 'p' is actually added to the queue.
     *         false if the queue contained it and therefore the add()
     *         was noop, or just changed the due date of the task.
K
kohsuke 已提交
311 312
     * @since 1.114
     */
313 314 315 316 317
    private synchronized boolean add(Task p, int quietPeriod, List<Action> actions) {
    	boolean taskConsumed=false;
    	List<Item> items = getItems(p);
    	Calendar due = new GregorianCalendar();
    	due.add(Calendar.SECOND, quietPeriod);
318

319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
    	List<Item> duplicatesInQueue = new ArrayList<Item>();
    	for(Item item : items) {
    		boolean shouldScheduleItem = false;
    		for (Action action: item.getActions()) {
    			if (action instanceof QueueAction)
    				shouldScheduleItem |= ((QueueAction) action).shouldSchedule(actions);
    		}
    		for (Action action: actions) {
    			if (action instanceof QueueAction) {
    				shouldScheduleItem |= ((QueueAction) action).shouldSchedule(item.getActions());
    			}
    		}
    		if(!shouldScheduleItem) {
    			duplicatesInQueue.add(item);
    		}
    	}
    	if (duplicatesInQueue.size() == 0) {
    		LOGGER.fine(p.getFullDisplayName() + " added to queue");
337

338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371
    		// put the item in the queue
    		waitingList.add(new WaitingItem(due,p,actions));
    		taskConsumed=true;
    	} else {
    		// the requested build is already queued, so will not be added
    		List<WaitingItem> waitingDuplicates = new ArrayList<WaitingItem>();
    		for(Item item : duplicatesInQueue) {
    			for(Action a : actions) {
    				if(a instanceof FoldableAction) {
    					((FoldableAction)a).foldIntoExisting(item.task, item.getActions());
    				}
    			}
    			if ((item instanceof WaitingItem))
    				waitingDuplicates.add((WaitingItem)item);
    		}
    		if(duplicatesInQueue.size() == 0) {
    			// all duplicates in the queue are already in the blocked or 
    			// buildable stage no need to requeue
    			return false;
    		}
    		// TODO: avoid calling scheduleMaintenance() if none of the waiting items 
    		// actually change
    		for(WaitingItem wi : waitingDuplicates) {
    			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
    			}
K
kohsuke 已提交
372

373 374 375 376 377 378 379 380 381
    			// waitingList is sorted, so when we change a timestamp we need to maintain order
    			waitingList.remove(wi);
    			wi.timestamp = due;
    			waitingList.add(wi);
    		}

    	}
    	scheduleMaintenance();   // let an executor know that a new item is in the queue.
    	return taskConsumed;
K
kohsuke 已提交
382
    }
383 384 385 386 387 388 389 390
    
    public synchronized boolean add(Task p, int quietPeriod) {
    	return add(p, quietPeriod, new Action[0]);
    }

    public synchronized boolean add(Task p, int quietPeriod, Action... actions) {
    	return add(p, quietPeriod, Arrays.asList(actions));
    }
K
kohsuke 已提交
391

K
kohsuke 已提交
392
    /**
393
     * Cancels the item in the queue. If the item is scheduled more than once, cancels the first occurrence.
K
kohsuke 已提交
394
     *
395 396
     * @return true if the project was indeed in the queue and was removed.
     *         false if this was no-op.
K
kohsuke 已提交
397
     */
K
kohsuke 已提交
398
    public synchronized boolean cancel(Task p) {
K
kohsuke 已提交
399
        LOGGER.fine("Cancelling " + p.getFullDisplayName());
K
kohsuke 已提交
400 401
        for (Iterator<WaitingItem> itr = waitingList.iterator(); itr.hasNext();) {
            Item item = itr.next();
K
kohsuke 已提交
402
            if (item.task.equals(p)) {
K
kohsuke 已提交
403
                itr.remove();
K
kohsuke 已提交
404
                return true;
K
kohsuke 已提交
405 406
            }
        }
K
kohsuke 已提交
407
        // use bitwise-OR to make sure that both branches get evaluated all the time
408
        return blockedProjects.remove(p)!=null | buildables.remove(p)!=null;
K
kohsuke 已提交
409
    }
410 411 412 413 414 415
    
    public synchronized boolean cancel(Item item) {
        LOGGER.fine("Cancelling " + item.task.getFullDisplayName() + " item#" + item.id);
        // use bitwise-OR to make sure that both branches get evaluated all the time
        return (item instanceof WaitingItem && waitingList.remove(item)) | blockedProjects.remove(item) | buildables.remove(item);
    }
K
kohsuke 已提交
416 417

    public synchronized boolean isEmpty() {
418
        return waitingList.isEmpty() && blockedProjects.isEmpty() && buildables.isEmpty();
K
kohsuke 已提交
419 420
    }

421
    private synchronized WaitingItem peek() {
422
        return waitingList.iterator().next();
K
kohsuke 已提交
423 424 425 426 427
    }

    /**
     * Gets a snapshot of items in the queue.
     */
428
    @Exported(inline=true)
K
kohsuke 已提交
429
    public synchronized Item[] getItems() {
430 431 432
        Item[] r = new Item[waitingList.size() + blockedProjects.size() + buildables.size()];
        waitingList.toArray(r);
        int idx = waitingList.size();
433 434 435 436
        for (BlockedItem p : blockedProjects.values())
            r[idx++] = p;
        for (BuildableItem p : buildables.values())
            r[idx++] = p;
K
kohsuke 已提交
437 438
        return r;
    }
439 440 441 442 443 444 445
    
    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;
    	return null;
    }
K
kohsuke 已提交
446

K
kohsuke 已提交
447 448 449
    /**
     * Gets all the {@link BuildableItem}s that are waiting for an executor in the given {@link Computer}.
     */
450 451 452 453
    public synchronized List<BuildableItem> getBuildableItems(Computer c) {
        List<BuildableItem> result = new ArrayList<BuildableItem>();
        for (BuildableItem p : buildables.values()) {
            Label l = p.task.getAssignedLabel();
454 455 456 457 458
            if (l != null) {
                // if a project has assigned label, it can be only built on it
                if (!l.contains(c.getNode()))
                    continue;
            }
459
            result.add(p);
460
        }
461
        return result;
462 463
    }

K
kohsuke 已提交
464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481
    /**
     * Gets the snapshot of {@link #buildables}.
     */
    public synchronized List<BuildableItem> getBuildableItems() {
        return new ArrayList<BuildableItem>(buildables.values());
    }

    /**
     * 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++;
        return r;
    }

K
kohsuke 已提交
482 483 484 485 486
    /**
     * Gets the information about the queue item for the given project.
     *
     * @return null if the project is not in the queue.
     */
487 488
    public synchronized Item getItem(Task t) {
        BlockedItem bp = blockedProjects.get(t);
489
        if (bp!=null)
490 491
            return bp;
        BuildableItem bi = buildables.get(t);
492
        if(bi!=null)
493 494
            return bi;

495
        for (Item item : waitingList) {
496
            if (item.task == t)
K
kohsuke 已提交
497 498 499 500 501
                return item;
        }
        return null;
    }

502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517
    /**
     * 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));
        for (Item item : waitingList) {
            if (item.task == t)
                result.add(item);
        }
        return result;
    }

518 519
    /**
     * Left for backward compatibility.
520
     *
521 522
     * @see #getItem(Task)
    public synchronized Item getItem(AbstractProject p) {
523
        return getItem((Task) p);
524
    }
K
kohsuke 已提交
525
     */
526

K
kohsuke 已提交
527
    /**
K
kohsuke 已提交
528
     * Returns true if this queue contains the said project.
K
kohsuke 已提交
529
     */
530 531
    public synchronized boolean contains(Task t) {
        if (blockedProjects.containsKey(t) || buildables.containsKey(t))
K
kohsuke 已提交
532
            return true;
533
        for (Item item : waitingList) {
534
            if (item.task == t)
K
kohsuke 已提交
535 536 537 538 539 540 541
                return true;
        }
        return false;
    }

    /**
     * Called by the executor to fetch something to build next.
542
     * <p>
K
kohsuke 已提交
543 544
     * This method blocks until a next project becomes buildable.
     */
545
    public Queue.Item pop() throws InterruptedException {
K
kohsuke 已提交
546
        final Executor exec = Executor.currentExecutor();
547

K
kohsuke 已提交
548
        try {
549
            while (true) {
K
kohsuke 已提交
550 551 552
                final JobOffer offer = new JobOffer(exec);
                long sleep = -1;

553
                synchronized (this) {
K
kohsuke 已提交
554 555
                    // consider myself parked
                    assert !parked.containsKey(exec);
556
                    parked.put(exec, offer);
K
kohsuke 已提交
557

K
kohsuke 已提交
558
                    // reuse executor thread to do a queue maintenance.
K
kohsuke 已提交
559 560 561 562 563
                    // at the end of this we get all the buildable jobs
                    // in the buildables field.
                    maintain();

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

                        // one last check to make sure this build is not blocked.
569
                        if (isBuildBlocked(p.task)) {
570
                            itr.remove();
571
                            blockedProjects.put(p.task,new BlockedItem(p));
572 573
                            continue;
                        }
574

575
                        JobOffer runner = loadBalancer.choose(p.task, getAvailableJobOffers());
576
                        if (runner == null)
K
kohsuke 已提交
577 578 579 580 581 582 583 584 585 586 587 588 589 590 591
                            // 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;

                        // found a matching executor. use it.
                        runner.set(p);
                        itr.remove();
                    }

                    // 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.

592
                    if (!waitingList.isEmpty()) {
K
kohsuke 已提交
593
                        // wait until the first item in the queue is due
594 595
                        sleep = peek().timestamp.getTimeInMillis() - new GregorianCalendar().getTimeInMillis();
                        if (sleep < 100) sleep = 100;    // avoid wait(0)
K
kohsuke 已提交
596 597 598 599 600
                    }
                }

                // this needs to be done outside synchronized block,
                // so that executors can maintain a queue while others are sleeping
601
                if (sleep == -1)
K
kohsuke 已提交
602 603 604 605
                    offer.event.block();
                else
                    offer.event.block(sleep);

606
                synchronized (this) {
607
                    // retract the offer object
608
                    assert parked.get(exec) == offer;
609 610
                    parked.remove(exec);

K
kohsuke 已提交
611
                    // am I woken up because I have a project to build?
612 613
                    if (offer.item != null) {
                        LOGGER.fine("Pop returning " + offer.item + " for " + exec.getName());
K
kohsuke 已提交
614
                        // if so, just build it
615
                        return offer.item;
K
kohsuke 已提交
616 617 618 619 620
                    }
                    // otherwise run a queue maintenance
                }
            }
        } finally {
621
            synchronized (this) {
K
kohsuke 已提交
622
                // remove myself from the parked list
623
                JobOffer offer = parked.remove(exec);
624
                if (offer != null && offer.item != null) {
625 626 627 628 629
                    // 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.
630
                    if (!contains(offer.item.task))
631
                        buildables.put(offer.item.task,offer.item);
K
kohsuke 已提交
632
                }
633 634 635 636 637 638

                // 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 已提交
639 640 641 642 643
            }
        }
    }

    /**
644
     * Creates a sublist of {@link #parked} that only consists of available offers.
K
kohsuke 已提交
645
     */
646 647 648 649 650 651
    private List<JobOffer> getAvailableJobOffers() {
        List<JobOffer> availables = new ArrayList<JobOffer>(parked.size());
        for (JobOffer j : parked.values())
            if(j.isAvailable())
                availables.add(j);
        return availables;
K
kohsuke 已提交
652 653 654 655
    }

    /**
     * Checks the queue and runs anything that can be run.
656 657
     *
     * <p>
K
kohsuke 已提交
658
     * When conditions are changed, this method should be invoked.
659
     * <p>
K
kohsuke 已提交
660 661 662 663 664 665 666
     * 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()) {
667
            if (av.getValue().item == null) {
K
kohsuke 已提交
668 669 670 671 672 673
                av.getValue().event.signal();
                return;
            }
        }
    }

674 675 676
    /**
     * Checks if the given task is blocked.
     */
677
    private boolean isBuildBlocked(Task t) {
678 679 680
        return t.isBuildBlocked() || !canRun(t.getResourceList());
    }

K
kohsuke 已提交
681 682

    /**
K
kohsuke 已提交
683
     * Queue maintenance.
684
     * <p>
685
     * Move projects between {@link #waitingList}, {@link #blockedProjects}, and {@link #buildables}
K
kohsuke 已提交
686 687
     * appropriately.
     */
688
    public synchronized void maintain() {
689 690
        if (LOGGER.isLoggable(Level.FINE))
            LOGGER.fine("Queue maintenance started " + this);
691

692
        Iterator<BlockedItem> itr = blockedProjects.values().iterator();
693
        while (itr.hasNext()) {
694 695
            BlockedItem p = itr.next();
            if (!isBuildBlocked(p.task)) {
K
kohsuke 已提交
696
                // ready to be executed
K
kohsuke 已提交
697
                LOGGER.fine(p.task.getFullDisplayName() + " no longer blocked");
K
kohsuke 已提交
698
                itr.remove();
699
                buildables.put(p.task,new BuildableItem(p));
K
kohsuke 已提交
700 701 702
            }
        }

703
        while (!waitingList.isEmpty()) {
704
            WaitingItem top = peek();
K
kohsuke 已提交
705

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

709
            Task p = top.task;
710
            if (!isBuildBlocked(p)) {
K
kohsuke 已提交
711
                // ready to be executed immediately
712
                waitingList.remove(top);
K
kohsuke 已提交
713
                LOGGER.fine(p.getFullDisplayName() + " ready to build");
714
                buildables.put(p,new BuildableItem(top));
K
kohsuke 已提交
715
            } else {
716
                // this can't be built now because another build is in progress
K
kohsuke 已提交
717
                // set this project aside.
718
                waitingList.remove(top);
K
kohsuke 已提交
719
                LOGGER.fine(p.getFullDisplayName() + " is blocked");
720
                blockedProjects.put(p,new BlockedItem(top));
K
kohsuke 已提交
721 722 723 724
            }
        }
    }

725 726 727 728
    public Api getApi() {
        return new Api(this);
    }

K
kohsuke 已提交
729 730
    /**
     * Task whose execution is controlled by the queue.
731
     *
732
     * <p>
K
kohsuke 已提交
733 734 735
     * {@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.
736 737 738 739
     *
     * <p>
     * Pending {@link Task}s are persisted when Hudson shuts down, so
     * it needs to be persistable.
K
kohsuke 已提交
740
     */
741
    public interface Task extends ModelObject, ResourceActivity {
742
        /**
743 744 745
         * 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.
746
         */
747
        Label getAssignedLabel();
748 749 750 751 752 753 754 755 756 757 758

        /**
         * 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.
759 760
         *
         * <p>
K
kohsuke 已提交
761 762
         * This can be used to define mutual exclusion that goes beyond
         * {@link #getResourceList()}.
763 764 765 766 767 768 769 770 771 772 773
         */
        boolean isBuildBlocked();

        /**
         * When {@link #isBuildBlocked()} is true, this method returns
         * human readable description of why the build is blocked.
         * Used for HTML rendering.
         */
        String getWhyBlocked();

        /**
K
kohsuke 已提交
774
         * Unique name of this task.
K
kohsuke 已提交
775
         *
K
kohsuke 已提交
776 777
         * <p>
         * This method is no longer used, left here for compatibility. Just return {@link #getDisplayName()}.
778 779 780
         */
        String getName();

781 782 783 784 785
        /**
         * @see hudson.model.Item#getFullDisplayName()
         */
        String getFullDisplayName();

786 787 788 789
        /**
         * Estimate of how long will it take to execute this task.
         * Measured in milliseconds.
         *
790
         * @return -1 if it's impossible to estimate.
791 792 793
         */
        long getEstimatedDuration();

K
kohsuke 已提交
794
        /**
795
         * Creates {@link Executable}, which performs the actual execution of the task.
K
kohsuke 已提交
796
         */
797
        Executable createExecutable() throws IOException;
798 799 800 801 802 803 804 805

        /**
         * 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 已提交
806 807 808 809 810 811

        /**
         * Works just like {@link #checkAbortPermission()} except it indicates the status by a return value,
         * instead of exception.
         */
        boolean hasAbortPermission();
812 813
        
        /**
K
kohsuke 已提交
814 815 816 817 818 819 820 821
         * 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 '/'.
822 823 824
         */
        String getUrl();
        
825 826 827 828 829
    }

    public interface Executable extends Runnable {
        /**
         * Task from which this executable was created.
K
kohsuke 已提交
830
         * Never null.
831 832 833 834 835 836 837
         */
        Task getParent();

        /**
         * Called by {@link Executor} to perform the task
         */
        void run();
838 839
    }

K
kohsuke 已提交
840 841 842
    /**
     * Item in a queue.
     */
843
    @ExportedBean(defaultVisibility = 999)
844 845 846 847 848
    public static abstract class Item extends Actionable {
    	
    	public final int id;
    	
		/**
K
kohsuke 已提交
849 850
         * Project to be built.
         */
851
        @Exported
852
        public final Task task;
853
        
854
        /**
855 856 857
         * Build is blocked because another build is in progress,
         * required {@link Resource}s are not available, or otherwise blocked
         * by {@link Task#isBuildBlocked()}.
858
         */
K
kohsuke 已提交
859
        @Exported
860
        public boolean isBlocked() { return this instanceof BlockedItem; }
861 862 863 864 865 866

        /**
         * 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 已提交
867
        @Exported
868
        public boolean isBuildable() { return this instanceof BuildableItem; }
869

870 871 872 873 874 875
        /**
         * True if the item is starving for an executor for too long.
         */
        @Exported
        public boolean isStuck() { return false; }

876 877 878 879 880 881 882 883
        protected Item(Task task, List<Action> actions, int id) {
            this.task = task;
            this.id = id;
            for (Action action: actions) addAction(action);
        }
        
        protected Item(Item item) {
        	this(item.task, item.getActions(), item.id);
K
kohsuke 已提交
884 885
        }

886
        /**
887
         * Gets a human-readable status message describing why it's in the queue.
888
         */
K
kohsuke 已提交
889
        @Exported
890
        public abstract String getWhy();
K
kohsuke 已提交
891

892 893 894
        public boolean hasCancelPermission() {
            return task.hasAbortPermission();
        }
895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924
        
        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);
        }
    }
    
    /**
     * An optional interface for actions on Queue.Item.
     * Lets the action cooperate in queue management.
     */
    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);
925
    }
926

927 928 929
    /**
     * {@link Item} in the {@link Queue#waitingList} stage.
     */
930 931 932
    public static final class WaitingItem extends Item implements Comparable<WaitingItem> {
    	private static final AtomicInteger COUNTER = new AtomicInteger(0);
    	
933 934 935 936 937
        /**
         * This item can be run after this time.
         */
        @Exported
        public Calendar timestamp;
938

939 940
        WaitingItem(Calendar timestamp, Task project, List<Action> actions) {
            super(project, actions, COUNTER.incrementAndGet());
941 942
            this.timestamp = timestamp;
        }
943
        
944 945 946 947 948 949 950 951 952
        public int compareTo(WaitingItem that) {
            int r = this.timestamp.getTime().compareTo(that.timestamp.getTime());
            if (r != 0) return r;

            return this.id - that.id;
        }

        @Override
        public String getWhy() {
953
            long diff = timestamp.getTimeInMillis() - System.currentTimeMillis();
954
            if (diff > 0)
K
i18n  
kohsuke 已提交
955
                return Messages.Queue_InQuietPeriod(Util.getTimeSpanString(diff));
956 957 958 959
            else
                return Messages.Queue_Unknown();
        }
    }
K
kohsuke 已提交
960

961 962 963
    /**
     * Common part between {@link BlockedItem} and {@link BuildableItem}.
     */
964
    public static abstract class NotWaitingItem extends Item {
965 966 967 968 969 970 971
        /**
         * When did this job exit the {@link Queue#waitingList} phase?
         */
        @Exported
        public final long buildableStartMilliseconds;

        protected NotWaitingItem(WaitingItem wi) {
972
            super(wi);
973
            buildableStartMilliseconds = System.currentTimeMillis();
K
kohsuke 已提交
974
        }
975

976
        protected NotWaitingItem(NotWaitingItem ni) {
977
            super(ni);
978
            buildableStartMilliseconds = ni.buildableStartMilliseconds;
979
        }
980
    }
981

982 983 984 985 986 987 988
    /**
     * {@link Item} in the {@link Queue#blockedProjects} stage.
     */
    public final class BlockedItem extends NotWaitingItem {
        public BlockedItem(WaitingItem wi) {
            super(wi);
        }
989

990 991 992 993 994 995 996 997 998 999 1000 1001 1002
        public BlockedItem(NotWaitingItem ni) {
            super(ni);
        }

        @Override
        public String getWhy() {
            ResourceActivity r = getBlockingActivity(task);
            if (r != null) {
                if (r == task) // blocked by itself, meaning another build is in progress
                    return Messages.Queue_InProgress();
                return Messages.Queue_BlockedBy(r.getDisplayName());
            }
            return task.getWhyBlocked();
1003
        }
1004
    }
1005

1006 1007 1008
    /**
     * {@link Item} in the {@link Queue#buildables} stage.
     */
1009
    public final static class BuildableItem extends NotWaitingItem {
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
        public BuildableItem(WaitingItem wi) {
            super(wi);
        }

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

        @Override
        public String getWhy() {
            Hudson hudson = Hudson.getInstance();
K
kohsuke 已提交
1021 1022 1023 1024
            if(hudson.isQuietingDown())
                return Messages.Queue_HudsonIsAboutToShutDown();

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

            String name = null;
1029 1030 1031 1032
            if (label != null) {
                name = label.getName();
                if (label.isOffline()) {
                    if (label.getNodes().size() > 1)
K
i18n  
kohsuke 已提交
1033
                        return Messages.Queue_AllNodesOffline(name);
1034
                    else
K
i18n  
kohsuke 已提交
1035
                        return Messages.Queue_NodeOffline(name);
1036 1037 1038
                }
            }

K
i18n  
kohsuke 已提交
1039 1040 1041 1042
            if(name==null)
                return Messages.Queue_WaitingForNextAvailableExecutor();
            else
                return Messages.Queue_WaitingForNextAvailableExecutorOn(name);
1043
        }
1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054

        @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) {
1055
                // if we were running elsewhere, we would have done this build ten times.
1056
                return elapsed > Math.max(d,60000L)*10;
1057 1058
            } else {
                // more than a day in the queue
1059
                return TimeUnit2.MILLISECONDS.toHours(elapsed)>24;
1060 1061
            }
        }
K
kohsuke 已提交
1062 1063 1064 1065 1066
    }

    /**
     * Unique number generator
     */
1067
    private int iota = 0;
K
kohsuke 已提交
1068 1069

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

1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112
    /**
     * 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 已提交
1113
                if(run==null)  throw new NoSuchElementException("No such build: "+string);
1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124
				return run;
			}

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

1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140
    /**
     * 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();
1141
            if (q != null)
1142 1143 1144 1145 1146
                q.maintain();
            else
                cancel();
        }
    }
1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196
    
    /**
     * A MultiMap -  LinkedMap crossover as a drop-in replacement for the previously used LinkedHashMap
     * And no, I don't care about performance ;)
     */
    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;
    	}
    }
K
kohsuke 已提交
1197
}