blob: 91b2cdf5b1968edc20cf2e38fd69c764c5a0a9bf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
use std::ops::{Deref, DerefMut};
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
pub struct Steal<T>(pub Option<T>);
impl<T> Steal<T> {
pub fn new(value: T) -> Steal<T> {
Steal(Some(value))
}
pub fn steal(&mut self) -> T {
self.0
.take()
.expect("Attempted to steal already stoken value")
}
}
impl<T> From<T> for Steal<T> {
fn from(value: T) -> Self {
Steal::new(value)
}
}
impl<T> Deref for Steal<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
self.0
.as_ref()
.expect("Attempted to dereference stolen value")
}
}
impl<T> DerefMut for Steal<T> {
fn deref_mut(&mut self) -> &mut Self::Target {
self.0
.as_mut()
.expect("Attempted to dereference stolen value")
}
}
|
