| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "sky/tools/debugger/trace_collector.h" | 5 #include "sky/tools/debugger/trace_collector.h" |
| 6 | 6 |
| 7 namespace sky { | 7 namespace sky { |
| 8 namespace debugger { | 8 namespace debugger { |
| 9 | 9 |
| 10 TraceCollector::TraceCollector(mojo::ScopedDataPipeConsumerHandle source) | 10 TraceCollector::TraceCollector(mojo::ScopedDataPipeConsumerHandle source) |
| 11 : drainer_(this, source.Pass()), is_complete_(false) { | 11 : drainer_(this, source.Pass()), is_complete_(false) { |
| 12 } | 12 } |
| 13 | 13 |
| 14 TraceCollector::~TraceCollector() { | 14 TraceCollector::~TraceCollector() { |
| 15 } | 15 } |
| 16 | 16 |
| 17 void TraceCollector::GetTrace(TraceCallback callback) { | 17 void TraceCollector::GetTrace(TraceCallback callback) { |
| 18 DCHECK(!callback_.is_null()); | 18 DCHECK(callback_.is_null()); |
| 19 DCHECK(!callback.is_null()); |
| 19 if (is_complete_) { | 20 if (is_complete_) { |
| 20 callback.Run(GetTraceAsString()); | 21 callback.Run(GetTraceAsString()); |
| 21 return; | 22 return; |
| 22 } | 23 } |
| 23 callback_ = callback; | 24 callback_ = callback; |
| 24 } | 25 } |
| 25 | 26 |
| 26 void TraceCollector::OnDataAvailable(const void* data, size_t num_bytes) { | 27 void TraceCollector::OnDataAvailable(const void* data, size_t num_bytes) { |
| 27 DCHECK(!is_complete_); | 28 DCHECK(!is_complete_); |
| 28 const char* chars = static_cast<const char*>(data); | 29 const char* chars = static_cast<const char*>(data); |
| 29 trace_.insert(trace_.end(), chars, chars + num_bytes); | 30 trace_.insert(trace_.end(), chars, chars + num_bytes); |
| 30 } | 31 } |
| 31 | 32 |
| 32 void TraceCollector::OnDataComplete() { | 33 void TraceCollector::OnDataComplete() { |
| 33 DCHECK(!is_complete_); | 34 DCHECK(!is_complete_); |
| 34 is_complete_ = true; | 35 is_complete_ = true; |
| 35 if (!callback_.is_null()) | 36 if (!callback_.is_null()) |
| 36 callback_.Run(GetTraceAsString()); | 37 callback_.Run(GetTraceAsString()); |
| 37 } | 38 } |
| 38 | 39 |
| 39 std::string TraceCollector::GetTraceAsString() { | 40 std::string TraceCollector::GetTraceAsString() { |
| 40 return std::string(&trace_.front(), trace_.size()); | 41 return std::string(&trace_.front(), trace_.size()); |
| 41 } | 42 } |
| 42 | 43 |
| 43 } // namespace debugger | 44 } // namespace debugger |
| 44 } // namespace sky | 45 } // namespace sky |
| OLD | NEW |