98 lines
2.1 KiB
Rust
98 lines
2.1 KiB
Rust
use std::fmt::Display;
|
|
use std::num::{NonZeroU16, NonZeroU64, ParseIntError};
|
|
use std::str::FromStr;
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
/// Wrapper type for a port number.
|
|
#[derive(Serialize, Deserialize, Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub struct Port(NonZeroU16);
|
|
|
|
impl Port {
|
|
pub const fn get(self) -> u16 {
|
|
self.0.get()
|
|
}
|
|
|
|
pub fn new(amt: u16) -> Option<Self> {
|
|
NonZeroU16::new(amt).map(Self)
|
|
}
|
|
}
|
|
|
|
impl Default for Port {
|
|
fn default() -> Self {
|
|
Self(unsafe { NonZeroU16::new_unchecked(443) })
|
|
}
|
|
}
|
|
|
|
impl FromStr for Port {
|
|
type Err = <NonZeroU16 as FromStr>::Err;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
NonZeroU16::from_str(s).map(Self)
|
|
}
|
|
}
|
|
|
|
impl Display for Port {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
self.0.fmt(f)
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Serialize, Deserialize, Default, Debug, Hash, Eq, PartialEq)]
|
|
pub struct Mebibytes(usize);
|
|
|
|
impl Mebibytes {
|
|
#[cfg(test)]
|
|
pub fn new(size: usize) -> Self {
|
|
Mebibytes(size)
|
|
}
|
|
}
|
|
|
|
impl FromStr for Mebibytes {
|
|
type Err = ParseIntError;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
s.parse::<usize>().map(Self)
|
|
}
|
|
}
|
|
|
|
pub struct Bytes(pub usize);
|
|
|
|
impl Bytes {
|
|
pub const fn get(&self) -> usize {
|
|
self.0
|
|
}
|
|
}
|
|
|
|
impl From<Mebibytes> for Bytes {
|
|
fn from(mib: Mebibytes) -> Self {
|
|
Self(mib.0 << 20)
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Deserialize, Debug, Hash, Eq, PartialEq)]
|
|
pub struct KilobitsPerSecond(NonZeroU64);
|
|
|
|
impl KilobitsPerSecond {
|
|
#[cfg(test)]
|
|
pub fn new(size: u64) -> Option<Self> {
|
|
NonZeroU64::new(size).map(Self)
|
|
}
|
|
}
|
|
|
|
impl FromStr for KilobitsPerSecond {
|
|
type Err = ParseIntError;
|
|
|
|
fn from_str(s: &str) -> Result<Self, Self::Err> {
|
|
s.parse::<NonZeroU64>().map(Self)
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Serialize, Debug, Hash, Eq, PartialEq)]
|
|
pub struct BytesPerSecond(NonZeroU64);
|
|
|
|
impl From<KilobitsPerSecond> for BytesPerSecond {
|
|
fn from(kbps: KilobitsPerSecond) -> Self {
|
|
Self(unsafe { NonZeroU64::new_unchecked(kbps.0.get() * 125) })
|
|
}
|
|
}
|