OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "tools/gn/scheduler.h" |
| 6 |
| 7 #include "base/bind.h" |
| 8 #include "tools/gn/ninja_target_writer.h" |
| 9 #include "tools/gn/standard_out.h" |
| 10 |
| 11 Scheduler* g_scheduler = NULL; |
| 12 |
| 13 Scheduler::Scheduler() |
| 14 : pool_(new base::SequencedWorkerPool(32, "worker_")), |
| 15 input_file_manager_(new InputFileManager), |
| 16 verbose_logging_(false), |
| 17 work_count_(0), |
| 18 is_failed_(false) { |
| 19 g_scheduler = this; |
| 20 } |
| 21 |
| 22 Scheduler::~Scheduler() { |
| 23 g_scheduler = NULL; |
| 24 } |
| 25 |
| 26 bool Scheduler::Run() { |
| 27 runner_.Run(); |
| 28 pool_->Shutdown(); |
| 29 return !is_failed(); |
| 30 } |
| 31 |
| 32 void Scheduler::Log(const std::string& verb, const std::string& msg) { |
| 33 if (base::MessageLoop::current() == &main_loop_) { |
| 34 LogOnMainThread(verb, msg); |
| 35 } else { |
| 36 // The run loop always joins on the sub threads, so the lifetime of this |
| 37 // object outlives the invocations of this function, hence "unretained". |
| 38 main_loop_.PostTask(FROM_HERE, |
| 39 base::Bind(&Scheduler::LogOnMainThread, |
| 40 base::Unretained(this), verb, msg)); |
| 41 } |
| 42 } |
| 43 |
| 44 void Scheduler::FailWithError(const Err& err) { |
| 45 DCHECK(err.has_error()); |
| 46 { |
| 47 base::AutoLock lock(lock_); |
| 48 |
| 49 if (is_failed_) |
| 50 return; // Ignore errors once we see one. |
| 51 is_failed_ = true; |
| 52 } |
| 53 |
| 54 if (base::MessageLoop::current() == &main_loop_) { |
| 55 FailWithErrorOnMainThread(err); |
| 56 } else { |
| 57 // The run loop always joins on the sub threads, so the lifetime of this |
| 58 // object outlives the invocations of this function, hence "unretained". |
| 59 main_loop_.PostTask(FROM_HERE, |
| 60 base::Bind(&Scheduler::FailWithErrorOnMainThread, |
| 61 base::Unretained(this), err)); |
| 62 } |
| 63 } |
| 64 |
| 65 void Scheduler::ScheduleWork(const base::Closure& work) { |
| 66 IncrementWorkCount(); |
| 67 pool_->PostWorkerTaskWithShutdownBehavior( |
| 68 FROM_HERE, base::Bind(&Scheduler::DoWork, |
| 69 base::Unretained(this), work), |
| 70 base::SequencedWorkerPool::BLOCK_SHUTDOWN); |
| 71 } |
| 72 |
| 73 void Scheduler::ScheduleTargetFileWrite(const Target* target) { |
| 74 pool_->PostWorkerTaskWithShutdownBehavior( |
| 75 FROM_HERE, base::Bind(&Scheduler::DoTargetFileWrite, |
| 76 base::Unretained(this), target), |
| 77 base::SequencedWorkerPool::BLOCK_SHUTDOWN); |
| 78 } |
| 79 |
| 80 void Scheduler::AddGenDependency(const SourceFile& source_file) { |
| 81 base::AutoLock lock(lock_); |
| 82 gen_dependencies_.push_back(source_file); |
| 83 } |
| 84 |
| 85 std::vector<SourceFile> Scheduler::GetGenDependencies() const { |
| 86 base::AutoLock lock(lock_); |
| 87 return gen_dependencies_; |
| 88 } |
| 89 |
| 90 void Scheduler::IncrementWorkCount() { |
| 91 base::AtomicRefCountInc(&work_count_); |
| 92 } |
| 93 |
| 94 void Scheduler::DecrementWorkCount() { |
| 95 if (!base::AtomicRefCountDec(&work_count_)) { |
| 96 if (base::MessageLoop::current() == &main_loop_) { |
| 97 OnComplete(); |
| 98 } else { |
| 99 main_loop_.PostTask(FROM_HERE, |
| 100 base::Bind(&Scheduler::OnComplete, |
| 101 base::Unretained(this))); |
| 102 } |
| 103 } |
| 104 } |
| 105 |
| 106 void Scheduler::LogOnMainThread(const std::string& verb, |
| 107 const std::string& msg) { |
| 108 OutputString(verb, DECORATION_YELLOW); |
| 109 OutputString(" " + msg + "\n"); |
| 110 } |
| 111 |
| 112 void Scheduler::FailWithErrorOnMainThread(const Err& err) { |
| 113 err.PrintToStdout(); |
| 114 runner_.Quit(); |
| 115 } |
| 116 |
| 117 void Scheduler::DoTargetFileWrite(const Target* target) { |
| 118 NinjaTargetWriter::RunAndWriteFile(target); |
| 119 } |
| 120 |
| 121 void Scheduler::DoWork(const base::Closure& closure) { |
| 122 closure.Run(); |
| 123 DecrementWorkCount(); |
| 124 } |
| 125 |
| 126 void Scheduler::OnComplete() { |
| 127 // Should be called on the main thread. |
| 128 DCHECK(base::MessageLoop::current() == main_loop()); |
| 129 runner_.Quit(); |
| 130 } |
OLD | NEW |