cpuallocator: add WithAllocFlags option.

Allow overriding the default allocation flags using the WithAllocFlags option. Signed-off-by: Krisztian Litkey <[email protected]>
containers · Nov 19, 2024 · a78bf6b · a78bf6b
1 parent 8c3015b
commit a78bf6b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pkg/cpuallocator/allocator.go b/pkg/cpuallocator/allocator.go
@@ -94,6 +94,14 @@ func WithPriority(p CPUPriority) Option {
 	}
 }
 
+// WithAllocFlags sets the allocation flags for the allocation.
+func WithAllocFlags(flags AllocFlag) Option {
+	return func(a *allocatorHelper) error {
+		a.flags = flags
+		return nil
+	}
+}
+
 type cpuAllocator struct {
 	logger.Logger
 	sys           sysfs.System  // wrapped sysfs.System instance