mirror of
https://github.com/kata-containers/cgroups-rs.git
synced 2026-08-05 02:13:23 +00:00
Make Cgroup can be stored in struct
- Change type of hier to remove lifetimes - impl Clone, Default, Debug for Cgroup Signed-off-by: Tim Zhang <tim@hyper.sh>
This commit is contained in:
+48
-50
@@ -77,59 +77,57 @@ macro_rules! gen_setter {
|
||||
}
|
||||
|
||||
/// A control group builder instance
|
||||
pub struct CgroupBuilder<'a> {
|
||||
pub struct CgroupBuilder {
|
||||
name: String,
|
||||
hierarchy: Box<&'a dyn Hierarchy>,
|
||||
/// Internal, unsupported field: use the associated builders instead.
|
||||
resources: Resources,
|
||||
}
|
||||
|
||||
impl<'a> CgroupBuilder<'a> {
|
||||
impl CgroupBuilder {
|
||||
/// Start building a control group with the supplied hierarchy and name pair.
|
||||
///
|
||||
/// Note that this does not actually create the control group until `build()` is called.
|
||||
pub fn new(name: &'a str, hierarchy: Box<&'a dyn Hierarchy>) -> CgroupBuilder<'a> {
|
||||
pub fn new(name: &str) -> CgroupBuilder {
|
||||
CgroupBuilder {
|
||||
name: name.to_owned(),
|
||||
hierarchy: hierarchy,
|
||||
resources: Resources::default(),
|
||||
}
|
||||
}
|
||||
|
||||
/// Builds the memory resources of the control group.
|
||||
pub fn memory(self) -> MemoryResourceBuilder<'a> {
|
||||
pub fn memory(self) -> MemoryResourceBuilder {
|
||||
MemoryResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the pid resources of the control group.
|
||||
pub fn pid(self) -> PidResourceBuilder<'a> {
|
||||
pub fn pid(self) -> PidResourceBuilder {
|
||||
PidResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the cpu resources of the control group.
|
||||
pub fn cpu(self) -> CpuResourceBuilder<'a> {
|
||||
pub fn cpu(self) -> CpuResourceBuilder {
|
||||
CpuResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the devices resources of the control group, disallowing or
|
||||
/// allowing access to certain devices in the system.
|
||||
pub fn devices(self) -> DeviceResourceBuilder<'a> {
|
||||
pub fn devices(self) -> DeviceResourceBuilder {
|
||||
DeviceResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the network resources of the control group, setting class id, or
|
||||
/// various priorities on networking interfaces.
|
||||
pub fn network(self) -> NetworkResourceBuilder<'a> {
|
||||
pub fn network(self) -> NetworkResourceBuilder {
|
||||
NetworkResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the hugepage/hugetlb resources available to the control group.
|
||||
pub fn hugepages(self) -> HugepagesResourceBuilder<'a> {
|
||||
pub fn hugepages(self) -> HugepagesResourceBuilder {
|
||||
HugepagesResourceBuilder { cgroup: self }
|
||||
}
|
||||
|
||||
/// Builds the block I/O resources available for the control group.
|
||||
pub fn blkio(self) -> BlkIoResourcesBuilder<'a> {
|
||||
pub fn blkio(self) -> BlkIoResourcesBuilder {
|
||||
BlkIoResourcesBuilder {
|
||||
cgroup: self,
|
||||
throttling_iops: false,
|
||||
@@ -137,19 +135,19 @@ impl<'a> CgroupBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Finalize the control group, consuming the builder and creating the control group.
|
||||
pub fn build(self) -> Cgroup<'a> {
|
||||
let cg = Cgroup::new(*self.hierarchy, self.name);
|
||||
pub fn build(self, hier: Box<dyn Hierarchy>) -> Cgroup {
|
||||
let cg = Cgroup::new(hier, self.name);
|
||||
let _ret = cg.apply(&self.resources);
|
||||
cg
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the memory controller of a control group.
|
||||
pub struct MemoryResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct MemoryResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> MemoryResourceBuilder<'a> {
|
||||
impl MemoryResourceBuilder {
|
||||
gen_setter!(
|
||||
memory,
|
||||
MemController,
|
||||
@@ -182,17 +180,17 @@ impl<'a> MemoryResourceBuilder<'a> {
|
||||
gen_setter!(memory, MemController, set_swappiness, swappiness, u64);
|
||||
|
||||
/// Finish the construction of the memory resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the pid controller of a control group.
|
||||
pub struct PidResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct PidResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> PidResourceBuilder<'a> {
|
||||
impl PidResourceBuilder {
|
||||
gen_setter!(
|
||||
pid,
|
||||
PidController,
|
||||
@@ -202,17 +200,17 @@ impl<'a> PidResourceBuilder<'a> {
|
||||
);
|
||||
|
||||
/// Finish the construction of the pid resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the cpuset & cpu controllers of a control group.
|
||||
pub struct CpuResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct CpuResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> CpuResourceBuilder<'a> {
|
||||
impl CpuResourceBuilder {
|
||||
gen_setter!(cpu, CpuSetController, set_cpus, cpus, String);
|
||||
gen_setter!(cpu, CpuSetController, set_mems, mems, String);
|
||||
gen_setter!(cpu, CpuController, set_shares, shares, u64);
|
||||
@@ -222,17 +220,17 @@ impl<'a> CpuResourceBuilder<'a> {
|
||||
gen_setter!(cpu, CpuController, set_rt_period, realtime_period, u64);
|
||||
|
||||
/// Finish the construction of the cpu resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the devices controller of a control group.
|
||||
pub struct DeviceResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct DeviceResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> DeviceResourceBuilder<'a> {
|
||||
impl DeviceResourceBuilder {
|
||||
/// Restrict (or allow) a device to the tasks inside the control group.
|
||||
pub fn device(
|
||||
mut self,
|
||||
@@ -241,7 +239,7 @@ impl<'a> DeviceResourceBuilder<'a> {
|
||||
devtype: crate::devices::DeviceType,
|
||||
allow: bool,
|
||||
access: Vec<crate::devices::DevicePermissions>,
|
||||
) -> DeviceResourceBuilder<'a> {
|
||||
) -> DeviceResourceBuilder {
|
||||
self.cgroup.resources.devices.devices.push(DeviceResource {
|
||||
major,
|
||||
minor,
|
||||
@@ -253,22 +251,22 @@ impl<'a> DeviceResourceBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Finish the construction of the devices resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the net_cls & net_prio controllers of a control group.
|
||||
pub struct NetworkResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct NetworkResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> NetworkResourceBuilder<'a> {
|
||||
impl NetworkResourceBuilder {
|
||||
gen_setter!(network, NetclsController, set_class, class_id, u64);
|
||||
|
||||
/// Set the priority of the tasks when operating on a networking device defined by `name` to be
|
||||
/// `priority`.
|
||||
pub fn priority(mut self, name: String, priority: u64) -> NetworkResourceBuilder<'a> {
|
||||
pub fn priority(mut self, name: String, priority: u64) -> NetworkResourceBuilder {
|
||||
self.cgroup
|
||||
.resources
|
||||
.network
|
||||
@@ -278,19 +276,19 @@ impl<'a> NetworkResourceBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Finish the construction of the network resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the hugepages controller of a control group.
|
||||
pub struct HugepagesResourceBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct HugepagesResourceBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
}
|
||||
|
||||
impl<'a> HugepagesResourceBuilder<'a> {
|
||||
impl HugepagesResourceBuilder {
|
||||
/// Limit the usage of certain hugepages (determined by `size`) to be at most `limit` bytes.
|
||||
pub fn limit(mut self, size: String, limit: u64) -> HugepagesResourceBuilder<'a> {
|
||||
pub fn limit(mut self, size: String, limit: u64) -> HugepagesResourceBuilder {
|
||||
self.cgroup
|
||||
.resources
|
||||
.hugepages
|
||||
@@ -300,18 +298,18 @@ impl<'a> HugepagesResourceBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Finish the construction of the network resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
/// A builder that configures the blkio controller of a control group.
|
||||
pub struct BlkIoResourcesBuilder<'a> {
|
||||
cgroup: CgroupBuilder<'a>,
|
||||
pub struct BlkIoResourcesBuilder {
|
||||
cgroup: CgroupBuilder,
|
||||
throttling_iops: bool,
|
||||
}
|
||||
|
||||
impl<'a> BlkIoResourcesBuilder<'a> {
|
||||
impl BlkIoResourcesBuilder {
|
||||
gen_setter!(blkio, BlkIoController, set_weight, weight, u16);
|
||||
gen_setter!(blkio, BlkIoController, set_leaf_weight, leaf_weight, u16);
|
||||
|
||||
@@ -322,7 +320,7 @@ impl<'a> BlkIoResourcesBuilder<'a> {
|
||||
minor: u64,
|
||||
weight: Option<u16>,
|
||||
leaf_weight: Option<u16>,
|
||||
) -> BlkIoResourcesBuilder<'a> {
|
||||
) -> BlkIoResourcesBuilder {
|
||||
self.cgroup
|
||||
.resources
|
||||
.blkio
|
||||
@@ -337,19 +335,19 @@ impl<'a> BlkIoResourcesBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Start configuring the I/O operations per second metric.
|
||||
pub fn throttle_iops(mut self) -> BlkIoResourcesBuilder<'a> {
|
||||
pub fn throttle_iops(mut self) -> BlkIoResourcesBuilder {
|
||||
self.throttling_iops = true;
|
||||
self
|
||||
}
|
||||
|
||||
/// Start configuring the bytes per second metric.
|
||||
pub fn throttle_bps(mut self) -> BlkIoResourcesBuilder<'a> {
|
||||
pub fn throttle_bps(mut self) -> BlkIoResourcesBuilder {
|
||||
self.throttling_iops = false;
|
||||
self
|
||||
}
|
||||
|
||||
/// Limit the read rate of the current metric for a certain device.
|
||||
pub fn read(mut self, major: u64, minor: u64, rate: u64) -> BlkIoResourcesBuilder<'a> {
|
||||
pub fn read(mut self, major: u64, minor: u64, rate: u64) -> BlkIoResourcesBuilder {
|
||||
let throttle = BlkIoDeviceThrottleResource { major, minor, rate };
|
||||
if self.throttling_iops {
|
||||
self.cgroup
|
||||
@@ -368,7 +366,7 @@ impl<'a> BlkIoResourcesBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Limit the write rate of the current metric for a certain device.
|
||||
pub fn write(mut self, major: u64, minor: u64, rate: u64) -> BlkIoResourcesBuilder<'a> {
|
||||
pub fn write(mut self, major: u64, minor: u64, rate: u64) -> BlkIoResourcesBuilder {
|
||||
let throttle = BlkIoDeviceThrottleResource { major, minor, rate };
|
||||
if self.throttling_iops {
|
||||
self.cgroup
|
||||
@@ -387,7 +385,7 @@ impl<'a> BlkIoResourcesBuilder<'a> {
|
||||
}
|
||||
|
||||
/// Finish the construction of the blkio resources of a control group.
|
||||
pub fn done(self) -> CgroupBuilder<'a> {
|
||||
pub fn done(self) -> CgroupBuilder {
|
||||
self.cgroup
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user