mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 15:16:12 -04:00
Timothy Hutchins
ff9438a3ed
This is the start of the implementing for the Linux platform. I aim to make this as similar to the Windows platform as possible, although due to the fundamental difference in the two OS's, there will be things that need to be different. As such, I will be using the Windows platform files as a base, and reimplementing the functions with Linux equivalents. I am sure this will have bugs, which by my best attempt will be fixed when found.
73 lines
2.2 KiB
C++
73 lines
2.2 KiB
C++
// Copyright 2020-2023 Google LLC
|
|
//
|
|
// 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
|
|
//
|
|
// https://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 PLATFORM_IMPL_LINUX_THREAD_POOL_H_
|
|
#define PLATFORM_IMPL_LINUX_THREAD_POOL_H_
|
|
|
|
#include <memory>
|
|
#include <queue>
|
|
#include <utility>
|
|
#include <vector>
|
|
#include <thread>
|
|
|
|
#include "absl/base/thread_annotations.h"
|
|
#include "absl/synchronization/mutex.h"
|
|
#include "internal/platform/runnable.h"
|
|
|
|
namespace nearby {
|
|
namespace linux {
|
|
|
|
class ThreadPool {
|
|
public:
|
|
virtual ~ThreadPool();
|
|
static std::unique_ptr<ThreadPool> Create(int max_pool_size);
|
|
|
|
// Runs a task on thread pool. The result indicates whether the task is put
|
|
// into the thread pool.
|
|
bool Run(Runnable task) ABSL_LOCKS_EXCLUDED(mutex_);
|
|
|
|
// The thread pool is closed immediately if there is no outstanding work,
|
|
// I/O, timer, or wait objects that are bound to the pool; otherwise, the
|
|
// thread pool is released asynchronously after the outstanding objects are
|
|
// freed.
|
|
void ShutDown() ABSL_LOCKS_EXCLUDED(mutex_);
|
|
|
|
private:
|
|
ThreadPool(std::unique_ptr<std::vector<std::thread>> &thread_pool, int max_pool_size);
|
|
|
|
void RunNextTask();
|
|
|
|
// Starts each task and injects a function that removes the task when it is finished
|
|
std::thread tasks_runner_;
|
|
|
|
// Protects the access to tasks of the thread pool.
|
|
mutable absl::Mutex mutex_;
|
|
|
|
// The task queue of the thread pool. Thread pool will pick up task to run
|
|
// when it is idle.
|
|
std::queue<Runnable> tasks_ ABSL_GUARDED_BY(mutex_);
|
|
|
|
// All the threads run in the pool
|
|
std::unique_ptr<std::vector<std::thread>> thread_pool_ ABSL_GUARDED_BY(mutex_);
|
|
|
|
// The maximum thread count in the thread pool
|
|
int max_pool_size_ ABSL_GUARDED_BY(mutex_) = 0;
|
|
|
|
};
|
|
|
|
} // namespace linux
|
|
} // namespace nearby
|
|
|
|
#endif // PLATFORM_IMPL_LINUX_THREAD_POOL_H_
|