blob: 790ab3c8016dde6ea53f0af9b6a6c8da843fd88c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/test/base/tracing.h"
#include "base/debug/trace_event.h"
#include "base/memory/singleton.h"
#include "base/message_loop.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/trace_controller.h"
#include "content/public/browser/trace_subscriber.h"
namespace {
class InProcessTraceController : public content::TraceSubscriber {
public:
static InProcessTraceController* GetInstance() {
return Singleton<InProcessTraceController>::get();
}
InProcessTraceController() {}
virtual ~InProcessTraceController() {}
bool BeginTracing(const std::string& categories) {
return content::TraceController::GetInstance()->BeginTracing(
this, categories);
}
bool EndTracing(std::string* json_trace_output) {
using namespace base::debug;
TraceResultBuffer::SimpleOutput output;
trace_buffer_.SetOutputCallback(output.GetCallback());
trace_buffer_.Start();
if (!content::TraceController::GetInstance()->EndTracingAsync(this))
return false;
// Wait for OnEndTracingComplete() to quit the message loop.
// OnTraceDataCollected may be called multiple times while blocking here.
ui_test_utils::RunMessageLoop();
trace_buffer_.Finish();
trace_buffer_.SetOutputCallback(TraceResultBuffer::OutputCallback());
*json_trace_output = output.json_output;
return true;
}
private:
friend struct DefaultSingletonTraits<InProcessTraceController>;
// TraceSubscriber
virtual void OnEndTracingComplete() OVERRIDE {
MessageLoopForUI::current()->Quit();
}
// TraceSubscriber
virtual void OnTraceDataCollected(
const scoped_refptr<base::RefCountedString>& trace_fragment) OVERRIDE {
trace_buffer_.AddFragment(trace_fragment->data());
}
// For collecting trace data asynchronously.
base::debug::TraceResultBuffer trace_buffer_;
DISALLOW_COPY_AND_ASSIGN(InProcessTraceController);
};
} // namespace
namespace tracing {
bool BeginTracing(const std::string& categories) {
return InProcessTraceController::GetInstance()->BeginTracing(categories);
}
bool EndTracing(std::string* json_trace_output) {
return InProcessTraceController::GetInstance()->EndTracing(json_trace_output);
}
} // namespace tracing
|
