123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965 |
- //! A C representation of Rust's `std::option::Option` used accross the FFI
- //! boundary for Solana program interfaces
- //!
- //! This implementation mostly matches `std::option` except iterators since the iteration
- //! trait requires returning `std::option::Option`
- use std::pin::Pin;
- use std::{
- convert, hint, mem,
- ops::{Deref, DerefMut},
- };
- /// A C representation of Rust's `std::option::Option`
- #[repr(C)]
- #[derive(Copy, PartialEq, PartialOrd, Eq, Ord, Debug, Hash)]
- pub enum COption<T> {
- /// No value
- None,
- /// Some value `T`
- Some(T),
- }
- /////////////////////////////////////////////////////////////////////////////
- // Type implementation
- /////////////////////////////////////////////////////////////////////////////
- impl<T> COption<T> {
- /////////////////////////////////////////////////////////////////////////
- // Querying the contained values
- /////////////////////////////////////////////////////////////////////////
- /// Returns `true` if the option is a [`COption::Some`] value.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x: COption<u32> = COption::Some(2);
- /// assert_eq!(x.is_some(), true);
- ///
- /// let x: COption<u32> = COption::None;
- /// assert_eq!(x.is_some(), false);
- /// ```
- ///
- /// [`COption::Some`]: #variant.COption::Some
- #[must_use = "if you intended to assert that this has a value, consider `.unwrap()` instead"]
- #[inline]
- pub fn is_some(&self) -> bool {
- match *self {
- COption::Some(_) => true,
- COption::None => false,
- }
- }
- /// Returns `true` if the option is a [`COption::None`] value.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x: COption<u32> = COption::Some(2);
- /// assert_eq!(x.is_none(), false);
- ///
- /// let x: COption<u32> = COption::None;
- /// assert_eq!(x.is_none(), true);
- /// ```
- ///
- /// [`COption::None`]: #variant.COption::None
- #[must_use = "if you intended to assert that this doesn't have a value, consider \
- `.and_then(|| panic!(\"`COption` had a value when expected `COption::None`\"))` instead"]
- #[inline]
- pub fn is_none(&self) -> bool {
- !self.is_some()
- }
- /// Returns `true` if the option is a [`COption::Some`] value containing the given value.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// #![feature(option_result_contains)]
- ///
- /// let x: COption<u32> = COption::Some(2);
- /// assert_eq!(x.contains(&2), true);
- ///
- /// let x: COption<u32> = COption::Some(3);
- /// assert_eq!(x.contains(&2), false);
- ///
- /// let x: COption<u32> = COption::None;
- /// assert_eq!(x.contains(&2), false);
- /// ```
- #[must_use]
- #[inline]
- pub fn contains<U>(&self, x: &U) -> bool
- where
- U: PartialEq<T>,
- {
- match self {
- COption::Some(y) => x == y,
- COption::None => false,
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Adapter for working with references
- /////////////////////////////////////////////////////////////////////////
- /// Converts from `&COption<T>` to `COption<&T>`.
- ///
- /// # Examples
- ///
- /// Converts an `COption<`[`String`]`>` into an `COption<`[`usize`]`>`, preserving the original.
- /// The [`map`] method takes the `self` argument by value, consuming the original,
- /// so this technique uses `as_ref` to first take an `COption` to a reference
- /// to the value inside the original.
- ///
- /// [`map`]: enum.COption.html#method.map
- /// [`String`]: ../../std/string/struct.String.html
- /// [`usize`]: ../../std/primitive.usize.html
- ///
- /// ```ignore
- /// let text: COption<String> = COption::Some("Hello, world!".to_string());
- /// // First, cast `COption<String>` to `COption<&String>` with `as_ref`,
- /// // then consume *that* with `map`, leaving `text` on the stack.
- /// let text_length: COption<usize> = text.as_ref().map(|s| s.len());
- /// println!("still can print text: {:?}", text);
- /// ```
- #[inline]
- pub fn as_ref(&self) -> COption<&T> {
- match *self {
- COption::Some(ref x) => COption::Some(x),
- COption::None => COption::None,
- }
- }
- /// Converts from `&mut COption<T>` to `COption<&mut T>`.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = COption::Some(2);
- /// match x.as_mut() {
- /// COption::Some(v) => *v = 42,
- /// COption::None => {},
- /// }
- /// assert_eq!(x, COption::Some(42));
- /// ```
- #[inline]
- pub fn as_mut(&mut self) -> COption<&mut T> {
- match *self {
- COption::Some(ref mut x) => COption::Some(x),
- COption::None => COption::None,
- }
- }
- /// Converts from [`Pin`]`<&COption<T>>` to `COption<`[`Pin`]`<&T>>`.
- ///
- /// [`Pin`]: ../pin/struct.Pin.html
- #[inline]
- #[allow(clippy::wrong_self_convention)]
- pub fn as_pin_ref(self: Pin<&Self>) -> COption<Pin<&T>> {
- unsafe { Pin::get_ref(self).as_ref().map(|x| Pin::new_unchecked(x)) }
- }
- /// Converts from [`Pin`]`<&mut COption<T>>` to `COption<`[`Pin`]`<&mut T>>`.
- ///
- /// [`Pin`]: ../pin/struct.Pin.html
- #[inline]
- #[allow(clippy::wrong_self_convention)]
- pub fn as_pin_mut(self: Pin<&mut Self>) -> COption<Pin<&mut T>> {
- unsafe {
- Pin::get_unchecked_mut(self)
- .as_mut()
- .map(|x| Pin::new_unchecked(x))
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Getting to contained values
- /////////////////////////////////////////////////////////////////////////
- /// Unwraps an option, yielding the content of a [`COption::Some`].
- ///
- /// # Panics
- ///
- /// Panics if the value is a [`COption::None`] with a custom panic message provided by
- /// `msg`.
- ///
- /// [`COption::Some`]: #variant.COption::Some
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some("value");
- /// assert_eq!(x.expect("the world is ending"), "value");
- /// ```
- ///
- /// ```ignore{.should_panic}
- /// let x: COption<&str> = COption::None;
- /// x.expect("the world is ending"); // panics with `the world is ending`
- /// ```
- #[inline]
- pub fn expect(self, msg: &str) -> T {
- match self {
- COption::Some(val) => val,
- COption::None => expect_failed(msg),
- }
- }
- /// Moves the value `v` out of the `COption<T>` if it is [`COption::Some(v)`].
- ///
- /// In general, because this function may panic, its use is discouraged.
- /// Instead, prefer to use pattern matching and handle the [`COption::None`]
- /// case explicitly.
- ///
- /// # Panics
- ///
- /// Panics if the self value equals [`COption::None`].
- ///
- /// [`COption::Some(v)`]: #variant.COption::Some
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some("air");
- /// assert_eq!(x.unwrap(), "air");
- /// ```
- ///
- /// ```ignore{.should_panic}
- /// let x: COption<&str> = COption::None;
- /// assert_eq!(x.unwrap(), "air"); // fails
- /// ```
- #[inline]
- pub fn unwrap(self) -> T {
- match self {
- COption::Some(val) => val,
- COption::None => panic!("called `COption::unwrap()` on a `COption::None` value"),
- }
- }
- /// Returns the contained value or a default.
- ///
- /// Arguments passed to `unwrap_or` are eagerly evaluated; if you are passing
- /// the result of a function call, it is recommended to use [`unwrap_or_else`],
- /// which is lazily evaluated.
- ///
- /// [`unwrap_or_else`]: #method.unwrap_or_else
- ///
- /// # Examples
- ///
- /// ```ignore
- /// assert_eq!(COption::Some("car").unwrap_or("bike"), "car");
- /// assert_eq!(COption::None.unwrap_or("bike"), "bike");
- /// ```
- #[inline]
- pub fn unwrap_or(self, def: T) -> T {
- match self {
- COption::Some(x) => x,
- COption::None => def,
- }
- }
- /// Returns the contained value or computes it from a closure.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let k = 10;
- /// assert_eq!(COption::Some(4).unwrap_or_else(|| 2 * k), 4);
- /// assert_eq!(COption::None.unwrap_or_else(|| 2 * k), 20);
- /// ```
- #[inline]
- pub fn unwrap_or_else<F: FnOnce() -> T>(self, f: F) -> T {
- match self {
- COption::Some(x) => x,
- COption::None => f(),
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Transforming contained values
- /////////////////////////////////////////////////////////////////////////
- /// Maps an `COption<T>` to `COption<U>` by applying a function to a contained value.
- ///
- /// # Examples
- ///
- /// Converts an `COption<`[`String`]`>` into an `COption<`[`usize`]`>`, consuming the original:
- ///
- /// [`String`]: ../../std/string/struct.String.html
- /// [`usize`]: ../../std/primitive.usize.html
- ///
- /// ```ignore
- /// let maybe_some_string = COption::Some(String::from("Hello, World!"));
- /// // `COption::map` takes self *by value*, consuming `maybe_some_string`
- /// let maybe_some_len = maybe_some_string.map(|s| s.len());
- ///
- /// assert_eq!(maybe_some_len, COption::Some(13));
- /// ```
- #[inline]
- pub fn map<U, F: FnOnce(T) -> U>(self, f: F) -> COption<U> {
- match self {
- COption::Some(x) => COption::Some(f(x)),
- COption::None => COption::None,
- }
- }
- /// Applies a function to the contained value (if any),
- /// or returns the provided default (if not).
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some("foo");
- /// assert_eq!(x.map_or(42, |v| v.len()), 3);
- ///
- /// let x: COption<&str> = COption::None;
- /// assert_eq!(x.map_or(42, |v| v.len()), 42);
- /// ```
- #[inline]
- pub fn map_or<U, F: FnOnce(T) -> U>(self, default: U, f: F) -> U {
- match self {
- COption::Some(t) => f(t),
- COption::None => default,
- }
- }
- /// Applies a function to the contained value (if any),
- /// or computes a default (if not).
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let k = 21;
- ///
- /// let x = COption::Some("foo");
- /// assert_eq!(x.map_or_else(|| 2 * k, |v| v.len()), 3);
- ///
- /// let x: COption<&str> = COption::None;
- /// assert_eq!(x.map_or_else(|| 2 * k, |v| v.len()), 42);
- /// ```
- #[inline]
- pub fn map_or_else<U, D: FnOnce() -> U, F: FnOnce(T) -> U>(self, default: D, f: F) -> U {
- match self {
- COption::Some(t) => f(t),
- COption::None => default(),
- }
- }
- /// Transforms the `COption<T>` into a [`Result<T, E>`], mapping [`COption::Some(v)`] to
- /// [`Ok(v)`] and [`COption::None`] to [`Err(err)`].
- ///
- /// Arguments passed to `ok_or` are eagerly evaluated; if you are passing the
- /// result of a function call, it is recommended to use [`ok_or_else`], which is
- /// lazily evaluated.
- ///
- /// [`Result<T, E>`]: ../../std/result/enum.Result.html
- /// [`Ok(v)`]: ../../std/result/enum.Result.html#variant.Ok
- /// [`Err(err)`]: ../../std/result/enum.Result.html#variant.Err
- /// [`COption::None`]: #variant.COption::None
- /// [`COption::Some(v)`]: #variant.COption::Some
- /// [`ok_or_else`]: #method.ok_or_else
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some("foo");
- /// assert_eq!(x.ok_or(0), Ok("foo"));
- ///
- /// let x: COption<&str> = COption::None;
- /// assert_eq!(x.ok_or(0), Err(0));
- /// ```
- #[inline]
- pub fn ok_or<E>(self, err: E) -> Result<T, E> {
- match self {
- COption::Some(v) => Ok(v),
- COption::None => Err(err),
- }
- }
- /// Transforms the `COption<T>` into a [`Result<T, E>`], mapping [`COption::Some(v)`] to
- /// [`Ok(v)`] and [`COption::None`] to [`Err(err())`].
- ///
- /// [`Result<T, E>`]: ../../std/result/enum.Result.html
- /// [`Ok(v)`]: ../../std/result/enum.Result.html#variant.Ok
- /// [`Err(err())`]: ../../std/result/enum.Result.html#variant.Err
- /// [`COption::None`]: #variant.COption::None
- /// [`COption::Some(v)`]: #variant.COption::Some
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some("foo");
- /// assert_eq!(x.ok_or_else(|| 0), Ok("foo"));
- ///
- /// let x: COption<&str> = COption::None;
- /// assert_eq!(x.ok_or_else(|| 0), Err(0));
- /// ```
- #[inline]
- pub fn ok_or_else<E, F: FnOnce() -> E>(self, err: F) -> Result<T, E> {
- match self {
- COption::Some(v) => Ok(v),
- COption::None => Err(err()),
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Boolean operations on the values, eager and lazy
- /////////////////////////////////////////////////////////////////////////
- /// Returns [`COption::None`] if the option is [`COption::None`], otherwise returns `optb`.
- ///
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some(2);
- /// let y: COption<&str> = COption::None;
- /// assert_eq!(x.and(y), COption::None);
- ///
- /// let x: COption<u32> = COption::None;
- /// let y = COption::Some("foo");
- /// assert_eq!(x.and(y), COption::None);
- ///
- /// let x = COption::Some(2);
- /// let y = COption::Some("foo");
- /// assert_eq!(x.and(y), COption::Some("foo"));
- ///
- /// let x: COption<u32> = COption::None;
- /// let y: COption<&str> = COption::None;
- /// assert_eq!(x.and(y), COption::None);
- /// ```
- #[inline]
- pub fn and<U>(self, optb: COption<U>) -> COption<U> {
- match self {
- COption::Some(_) => optb,
- COption::None => COption::None,
- }
- }
- /// Returns [`COption::None`] if the option is [`COption::None`], otherwise calls `f` with the
- /// wrapped value and returns the result.
- ///
- /// COption::Some languages call this operation flatmap.
- ///
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// fn sq(x: u32) -> COption<u32> { COption::Some(x * x) }
- /// fn nope(_: u32) -> COption<u32> { COption::None }
- ///
- /// assert_eq!(COption::Some(2).and_then(sq).and_then(sq), COption::Some(16));
- /// assert_eq!(COption::Some(2).and_then(sq).and_then(nope), COption::None);
- /// assert_eq!(COption::Some(2).and_then(nope).and_then(sq), COption::None);
- /// assert_eq!(COption::None.and_then(sq).and_then(sq), COption::None);
- /// ```
- #[inline]
- pub fn and_then<U, F: FnOnce(T) -> COption<U>>(self, f: F) -> COption<U> {
- match self {
- COption::Some(x) => f(x),
- COption::None => COption::None,
- }
- }
- /// Returns [`COption::None`] if the option is [`COption::None`], otherwise calls `predicate`
- /// with the wrapped value and returns:
- ///
- /// - [`COption::Some(t)`] if `predicate` returns `true` (where `t` is the wrapped
- /// value), and
- /// - [`COption::None`] if `predicate` returns `false`.
- ///
- /// This function works similar to [`Iterator::filter()`]. You can imagine
- /// the `COption<T>` being an iterator over one or zero elements. `filter()`
- /// lets you decide which elements to keep.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// fn is_even(n: &i32) -> bool {
- /// n % 2 == 0
- /// }
- ///
- /// assert_eq!(COption::None.filter(is_even), COption::None);
- /// assert_eq!(COption::Some(3).filter(is_even), COption::None);
- /// assert_eq!(COption::Some(4).filter(is_even), COption::Some(4));
- /// ```
- ///
- /// [`COption::None`]: #variant.COption::None
- /// [`COption::Some(t)`]: #variant.COption::Some
- /// [`Iterator::filter()`]: ../../std/iter/trait.Iterator.html#method.filter
- #[inline]
- pub fn filter<P: FnOnce(&T) -> bool>(self, predicate: P) -> Self {
- if let COption::Some(x) = self {
- if predicate(&x) {
- return COption::Some(x);
- }
- }
- COption::None
- }
- /// Returns the option if it contains a value, otherwise returns `optb`.
- ///
- /// Arguments passed to `or` are eagerly evaluated; if you are passing the
- /// result of a function call, it is recommended to use [`or_else`], which is
- /// lazily evaluated.
- ///
- /// [`or_else`]: #method.or_else
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some(2);
- /// let y = COption::None;
- /// assert_eq!(x.or(y), COption::Some(2));
- ///
- /// let x = COption::None;
- /// let y = COption::Some(100);
- /// assert_eq!(x.or(y), COption::Some(100));
- ///
- /// let x = COption::Some(2);
- /// let y = COption::Some(100);
- /// assert_eq!(x.or(y), COption::Some(2));
- ///
- /// let x: COption<u32> = COption::None;
- /// let y = COption::None;
- /// assert_eq!(x.or(y), COption::None);
- /// ```
- #[inline]
- pub fn or(self, optb: COption<T>) -> COption<T> {
- match self {
- COption::Some(_) => self,
- COption::None => optb,
- }
- }
- /// Returns the option if it contains a value, otherwise calls `f` and
- /// returns the result.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// fn nobody() -> COption<&'static str> { COption::None }
- /// fn vikings() -> COption<&'static str> { COption::Some("vikings") }
- ///
- /// assert_eq!(COption::Some("barbarians").or_else(vikings), COption::Some("barbarians"));
- /// assert_eq!(COption::None.or_else(vikings), COption::Some("vikings"));
- /// assert_eq!(COption::None.or_else(nobody), COption::None);
- /// ```
- #[inline]
- pub fn or_else<F: FnOnce() -> COption<T>>(self, f: F) -> COption<T> {
- match self {
- COption::Some(_) => self,
- COption::None => f(),
- }
- }
- /// Returns [`COption::Some`] if exactly one of `self`, `optb` is [`COption::Some`], otherwise returns [`COption::None`].
- ///
- /// [`COption::Some`]: #variant.COption::Some
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = COption::Some(2);
- /// let y: COption<u32> = COption::None;
- /// assert_eq!(x.xor(y), COption::Some(2));
- ///
- /// let x: COption<u32> = COption::None;
- /// let y = COption::Some(2);
- /// assert_eq!(x.xor(y), COption::Some(2));
- ///
- /// let x = COption::Some(2);
- /// let y = COption::Some(2);
- /// assert_eq!(x.xor(y), COption::None);
- ///
- /// let x: COption<u32> = COption::None;
- /// let y: COption<u32> = COption::None;
- /// assert_eq!(x.xor(y), COption::None);
- /// ```
- #[inline]
- pub fn xor(self, optb: COption<T>) -> COption<T> {
- match (self, optb) {
- (COption::Some(a), COption::None) => COption::Some(a),
- (COption::None, COption::Some(b)) => COption::Some(b),
- _ => COption::None,
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Entry-like operations to insert if COption::None and return a reference
- /////////////////////////////////////////////////////////////////////////
- /// Inserts `v` into the option if it is [`COption::None`], then
- /// returns a mutable reference to the contained value.
- ///
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = COption::None;
- ///
- /// {
- /// let y: &mut u32 = x.get_or_insert(5);
- /// assert_eq!(y, &5);
- ///
- /// *y = 7;
- /// }
- ///
- /// assert_eq!(x, COption::Some(7));
- /// ```
- #[inline]
- pub fn get_or_insert(&mut self, v: T) -> &mut T {
- self.get_or_insert_with(|| v)
- }
- /// Inserts a value computed from `f` into the option if it is [`COption::None`], then
- /// returns a mutable reference to the contained value.
- ///
- /// [`COption::None`]: #variant.COption::None
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = COption::None;
- ///
- /// {
- /// let y: &mut u32 = x.get_or_insert_with(|| 5);
- /// assert_eq!(y, &5);
- ///
- /// *y = 7;
- /// }
- ///
- /// assert_eq!(x, COption::Some(7));
- /// ```
- #[inline]
- pub fn get_or_insert_with<F: FnOnce() -> T>(&mut self, f: F) -> &mut T {
- if let COption::None = *self {
- *self = COption::Some(f())
- }
- match *self {
- COption::Some(ref mut v) => v,
- COption::None => unsafe { hint::unreachable_unchecked() },
- }
- }
- /////////////////////////////////////////////////////////////////////////
- // Misc
- /////////////////////////////////////////////////////////////////////////
- /// Replaces the actual value in the option by the value given in parameter,
- /// returning the old value if present,
- /// leaving a [`COption::Some`] in its place without deinitializing either one.
- ///
- /// [`COption::Some`]: #variant.COption::Some
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = COption::Some(2);
- /// let old = x.replace(5);
- /// assert_eq!(x, COption::Some(5));
- /// assert_eq!(old, COption::Some(2));
- ///
- /// let mut x = COption::None;
- /// let old = x.replace(3);
- /// assert_eq!(x, COption::Some(3));
- /// assert_eq!(old, COption::None);
- /// ```
- #[inline]
- pub fn replace(&mut self, value: T) -> COption<T> {
- mem::replace(self, COption::Some(value))
- }
- }
- impl<T: Copy> COption<&T> {
- /// Maps an `COption<&T>` to an `COption<T>` by copying the contents of the
- /// option.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = 12;
- /// let opt_x = COption::Some(&x);
- /// assert_eq!(opt_x, COption::Some(&12));
- /// let copied = opt_x.copied();
- /// assert_eq!(copied, COption::Some(12));
- /// ```
- pub fn copied(self) -> COption<T> {
- self.map(|&t| t)
- }
- }
- impl<T: Copy> COption<&mut T> {
- /// Maps an `COption<&mut T>` to an `COption<T>` by copying the contents of the
- /// option.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = 12;
- /// let opt_x = COption::Some(&mut x);
- /// assert_eq!(opt_x, COption::Some(&mut 12));
- /// let copied = opt_x.copied();
- /// assert_eq!(copied, COption::Some(12));
- /// ```
- pub fn copied(self) -> COption<T> {
- self.map(|&mut t| t)
- }
- }
- impl<T: Clone> COption<&T> {
- /// Maps an `COption<&T>` to an `COption<T>` by cloning the contents of the
- /// option.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let x = 12;
- /// let opt_x = COption::Some(&x);
- /// assert_eq!(opt_x, COption::Some(&12));
- /// let cloned = opt_x.cloned();
- /// assert_eq!(cloned, COption::Some(12));
- /// ```
- pub fn cloned(self) -> COption<T> {
- self.map(|t| t.clone())
- }
- }
- impl<T: Clone> COption<&mut T> {
- /// Maps an `COption<&mut T>` to an `COption<T>` by cloning the contents of the
- /// option.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let mut x = 12;
- /// let opt_x = COption::Some(&mut x);
- /// assert_eq!(opt_x, COption::Some(&mut 12));
- /// let cloned = opt_x.cloned();
- /// assert_eq!(cloned, COption::Some(12));
- /// ```
- pub fn cloned(self) -> COption<T> {
- self.map(|t| t.clone())
- }
- }
- impl<T: Default> COption<T> {
- /// Returns the contained value or a default
- ///
- /// Consumes the `self` argument then, if [`COption::Some`], returns the contained
- /// value, otherwise if [`COption::None`], returns the [default value] for that
- /// type.
- ///
- /// # Examples
- ///
- /// Converts a string to an integer, turning poorly-formed strings
- /// into 0 (the default value for integers). [`parse`] converts
- /// a string to any other type that implements [`FromStr`], returning
- /// [`COption::None`] on error.
- ///
- /// ```ignore
- /// let good_year_from_input = "1909";
- /// let bad_year_from_input = "190blarg";
- /// let good_year = good_year_from_input.parse().ok().unwrap_or_default();
- /// let bad_year = bad_year_from_input.parse().ok().unwrap_or_default();
- ///
- /// assert_eq!(1909, good_year);
- /// assert_eq!(0, bad_year);
- /// ```
- ///
- /// [`COption::Some`]: #variant.COption::Some
- /// [`COption::None`]: #variant.COption::None
- /// [default value]: ../default/trait.Default.html#tymethod.default
- /// [`parse`]: ../../std/primitive.str.html#method.parse
- /// [`FromStr`]: ../../std/str/trait.FromStr.html
- #[inline]
- pub fn unwrap_or_default(self) -> T {
- match self {
- COption::Some(x) => x,
- COption::None => Default::default(),
- }
- }
- }
- impl<T: Deref> COption<T> {
- /// Converts from `COption<T>` (or `&COption<T>`) to `COption<&T::Target>`.
- ///
- /// Leaves the original COption in-place, creating a new one with a reference
- /// to the original one, additionally coercing the contents via [`Deref`].
- ///
- /// [`Deref`]: ../../std/ops/trait.Deref.html
- ///
- /// # Examples
- ///
- /// ```ignore
- /// #![feature(inner_deref)]
- ///
- /// let x: COption<String> = COption::Some("hey".to_owned());
- /// assert_eq!(x.as_deref(), COption::Some("hey"));
- ///
- /// let x: COption<String> = COption::None;
- /// assert_eq!(x.as_deref(), COption::None);
- /// ```
- pub fn as_deref(&self) -> COption<&T::Target> {
- self.as_ref().map(|t| t.deref())
- }
- }
- impl<T: DerefMut> COption<T> {
- /// Converts from `COption<T>` (or `&mut COption<T>`) to `COption<&mut T::Target>`.
- ///
- /// Leaves the original `COption` in-place, creating a new one containing a mutable reference to
- /// the inner type's `Deref::Target` type.
- ///
- /// # Examples
- ///
- /// ```ignore
- /// #![feature(inner_deref)]
- ///
- /// let mut x: COption<String> = COption::Some("hey".to_owned());
- /// assert_eq!(x.as_deref_mut().map(|x| {
- /// x.make_ascii_uppercase();
- /// x
- /// }), COption::Some("HEY".to_owned().as_mut_str()));
- /// ```
- pub fn as_deref_mut(&mut self) -> COption<&mut T::Target> {
- self.as_mut().map(|t| t.deref_mut())
- }
- }
- impl<T, E> COption<Result<T, E>> {
- /// Transposes an `COption` of a [`Result`] into a [`Result`] of an `COption`.
- ///
- /// [`COption::None`] will be mapped to [`Ok`]`(`[`COption::None`]`)`.
- /// [`COption::Some`]`(`[`Ok`]`(_))` and [`COption::Some`]`(`[`Err`]`(_))` will be mapped to
- /// [`Ok`]`(`[`COption::Some`]`(_))` and [`Err`]`(_)`.
- ///
- /// [`COption::None`]: #variant.COption::None
- /// [`Ok`]: ../../std/result/enum.Result.html#variant.Ok
- /// [`COption::Some`]: #variant.COption::Some
- /// [`Err`]: ../../std/result/enum.Result.html#variant.Err
- ///
- /// # Examples
- ///
- /// ```ignore
- /// #[derive(Debug, Eq, PartialEq)]
- /// struct COption::SomeErr;
- ///
- /// let x: Result<COption<i32>, COption::SomeErr> = Ok(COption::Some(5));
- /// let y: COption<Result<i32, COption::SomeErr>> = COption::Some(Ok(5));
- /// assert_eq!(x, y.transpose());
- /// ```
- #[inline]
- pub fn transpose(self) -> Result<COption<T>, E> {
- match self {
- COption::Some(Ok(x)) => Ok(COption::Some(x)),
- COption::Some(Err(e)) => Err(e),
- COption::None => Ok(COption::None),
- }
- }
- }
- // This is a separate function to reduce the code size of .expect() itself.
- #[inline(never)]
- #[cold]
- fn expect_failed(msg: &str) -> ! {
- panic!("{}", msg)
- }
- // // This is a separate function to reduce the code size of .expect_none() itself.
- // #[inline(never)]
- // #[cold]
- // fn expect_none_failed(msg: &str, value: &dyn fmt::Debug) -> ! {
- // panic!("{}: {:?}", msg, value)
- // }
- /////////////////////////////////////////////////////////////////////////////
- // Trait implementations
- /////////////////////////////////////////////////////////////////////////////
- impl<T: Clone> Clone for COption<T> {
- #[inline]
- fn clone(&self) -> Self {
- match self {
- COption::Some(x) => COption::Some(x.clone()),
- COption::None => COption::None,
- }
- }
- #[inline]
- fn clone_from(&mut self, source: &Self) {
- match (self, source) {
- (COption::Some(to), COption::Some(from)) => to.clone_from(from),
- (to, from) => *to = from.clone(),
- }
- }
- }
- impl<T> Default for COption<T> {
- /// Returns [`COption::None`][COption::COption::None].
- ///
- /// # Examples
- ///
- /// ```ignore
- /// let opt: COption<u32> = COption::default();
- /// assert!(opt.is_none());
- /// ```
- #[inline]
- fn default() -> COption<T> {
- COption::None
- }
- }
- impl<T> From<T> for COption<T> {
- fn from(val: T) -> COption<T> {
- COption::Some(val)
- }
- }
- impl<'a, T> From<&'a COption<T>> for COption<&'a T> {
- fn from(o: &'a COption<T>) -> COption<&'a T> {
- o.as_ref()
- }
- }
- impl<'a, T> From<&'a mut COption<T>> for COption<&'a mut T> {
- fn from(o: &'a mut COption<T>) -> COption<&'a mut T> {
- o.as_mut()
- }
- }
- impl<T> COption<COption<T>> {
- /// Converts from `COption<COption<T>>` to `COption<T>`
- ///
- /// # Examples
- /// Basic usage:
- /// ```ignore
- /// #![feature(option_flattening)]
- /// let x: COption<COption<u32>> = COption::Some(COption::Some(6));
- /// assert_eq!(COption::Some(6), x.flatten());
- ///
- /// let x: COption<COption<u32>> = COption::Some(COption::None);
- /// assert_eq!(COption::None, x.flatten());
- ///
- /// let x: COption<COption<u32>> = COption::None;
- /// assert_eq!(COption::None, x.flatten());
- /// ```
- /// Flattening once only removes one level of nesting:
- /// ```ignore
- /// #![feature(option_flattening)]
- /// let x: COption<COption<COption<u32>>> = COption::Some(COption::Some(COption::Some(6)));
- /// assert_eq!(COption::Some(COption::Some(6)), x.flatten());
- /// assert_eq!(COption::Some(6), x.flatten().flatten());
- /// ```
- #[inline]
- pub fn flatten(self) -> COption<T> {
- self.and_then(convert::identity)
- }
- }
|