From f115c95deb1a49a181f3c896fcfdc0a7e0684f10 Mon Sep 17 00:00:00 2001
From: Feruzjon Muyassarov <feruzjon.muyassarov@intel.com>
Date: Wed, 21 Aug 2024 15:56:56 +0300
Subject: [PATCH] utils: add functions for setting cpu frequency scaling
 governor

Signed-off-by: Feruzjon Muyassarov <feruzjon.muyassarov@intel.com>
---
 pkg/utils/sysfs.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/pkg/utils/sysfs.go b/pkg/utils/sysfs.go
index 0dfb70c..6e72338 100644
--- a/pkg/utils/sysfs.go
+++ b/pkg/utils/sysfs.go
@@ -34,6 +34,10 @@ func setCPUFreqValue(cpu ID, setting string, value int) error {
 	return writeFileInt(cpuFreqPath(cpu, setting), value)
 }
 
+func SetCPUScalingGovernor(cpu ID, governor string) error {
+	return writeFileStr(cpuFreqPath(cpu, "scaling_governor"), governor)
+}
+
 // GetCPUFreqValue returns information of the currently used CPU frequency
 func GetCPUFreqValue(cpu ID, setting string) (int, error) {
 	raw, err := os.ReadFile(cpuFreqPath(cpu, setting))
@@ -63,6 +67,17 @@ func SetCPUScalingMaxFreq(cpu ID, freq int) error {
 	return setCPUFreqValue(cpu, "scaling_max_freq", freq)
 }
 
+// SetScalingGovernorForCPUs sets the scaling_governor of a given set of CPUs
+func SetScalingGovernorForCPUs(cpus []ID, governor string) error {
+	for _, cpu := range cpus {
+		if err := SetCPUScalingGovernor(cpu, governor); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
 // SetCPUsScalingMinFreq sets the scaling_min_freq value of a given set of CPUs
 func SetCPUsScalingMinFreq(cpus []ID, freq int) error {
 	for _, cpu := range cpus {
@@ -129,6 +144,10 @@ func writeFileInt(path string, value int) error {
 	return os.WriteFile(path, []byte(strconv.Itoa(value)), 0644)
 }
 
+func writeFileStr(path string, value string) error {
+	return os.WriteFile(path, []byte(value), 0644)
+}
+
 func readFileInt(path string) (int, error) {
 	data, err := os.ReadFile(path)
 	if err != nil {