blob: b28d94addb61818babd9f804aabe0496d567605e [file] [log] [blame]
#include <functional>
#include <string>
#include <unordered_map>
#include <pdx/service.h>
#include "cpu_set.h"
#include "task.h"
namespace android {
namespace dvr {
// PerformanceService manages compute partitions usings cpusets. Different
// cpusets are assigned specific purposes and performance characteristics;
// clients may request for threads to be moved into these cpusets to help
// achieve system performance goals.
class PerformanceService : public pdx::ServiceBase<PerformanceService> {
pdx::Status<void> HandleMessage(pdx::Message& message) override;
bool IsInitialized() const override;
std::string DumpState(size_t max_length) override;
friend BASE;
pdx::Status<void> OnSetSchedulerPolicy(pdx::Message& message, pid_t task_id,
const std::string& scheduler_class);
pdx::Status<void> OnSetCpuPartition(pdx::Message& message, pid_t task_id,
const std::string& partition);
pdx::Status<void> OnSetSchedulerClass(pdx::Message& message, pid_t task_id,
const std::string& scheduler_class);
pdx::Status<std::string> OnGetCpuPartition(pdx::Message& message,
pid_t task_id);
CpuSetManager cpuset_;
int sched_fifo_min_priority_;
int sched_fifo_max_priority_;
// Scheduler class config type.
struct SchedulerClassConfig {
unsigned long timer_slack;
int scheduler_policy;
int priority;
std::function<bool(const pdx::Message& message, const Task& task)>
// Check the permisison of the given task to use this scheduler class. If a
// permission check function is not set then operations are only allowed on
// tasks in the sender's process.
bool IsAllowed(const pdx::Message& sender, const Task& task) const {
if (permission_check)
return permission_check(sender, task);
else if (!task || task.thread_group_id() != sender.GetProcessId())
return false;
return true;
std::unordered_map<std::string, SchedulerClassConfig> scheduler_classes_;
std::function<bool(const pdx::Message& message, const Task& task)>
PerformanceService(const PerformanceService&) = delete;
void operator=(const PerformanceService&) = delete;
} // namespace dvr
} // namespace android