blob: 354c308bc1f5a3267c8fecfff8b28ef956f4fbaf [file] [log] [blame]
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef TEST_TASK_RUNNER_THREAD_H_
#define TEST_TASK_RUNNER_THREAD_H_
#include <condition_variable>
#include <mutex>
#include <thread>
#include "perfetto/base/task_runner.h"
#include "src/base/test/test_task_runner.h"
namespace perfetto {
// Used to perform initialization work on a background TaskRunnerThread.
class ThreadDelegate {
public:
virtual ~ThreadDelegate();
// Invoked on the target thread before the message loop is started.
virtual void Initialize(base::TaskRunner* task_runner) = 0;
};
// Background thread which spins a task runner until completed or the thread is
// destroyed. If the thread is destroyed before the task runner completes, the
// task runner is quit and the thread is joined.
class TaskRunnerThread {
public:
explicit TaskRunnerThread(const char* name);
~TaskRunnerThread();
// Blocks until the thread has been created and Initialize() has been
// called.
void Start(std::unique_ptr<ThreadDelegate> delegate);
// Blocks until the thread has been stopped and joined.
void Stop();
uint64_t GetThreadCPUTimeNs();
private:
void Run(std::unique_ptr<ThreadDelegate> delegate);
const char* const name_;
std::thread thread_;
std::condition_variable ready_;
// All variables below this point are protected by |mutex_|.
std::mutex mutex_;
base::UnixTaskRunner* runner_ = nullptr;
};
} // namespace perfetto
#endif // TEST_TASK_RUNNER_THREAD_H_