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