instance_copy_on_write/

lib.rs

/*-
 * instance-copy-on-write - a synchronization primitive based on copy-on-write.
 * 
 * Copyright (C) 2025 Aleksandr Morozov alex@4neko.org
 * 
 * The instance-copy-on-write crate can be redistributed and/or modified
 * under the terms of either of the following licenses:
 *
 *   1. The MIT License (MIT)
 */

 /// Type of the sync code.
 #[derive(Copy, Clone, Debug, PartialEq, Eq)]
 pub enum ICoWLockTypes
 {
    /// Based on atomic and backoff.
    Atomic,

    /// based on the OS RwLock
    RwLock,
 }

impl fmt::Display for ICoWLockTypes
{
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
    {
        match self
        {
            Self::Atomic => 
                write!(f, "atomic"),
            Self::RwLock => 
                write!(f, "rwlock"),
        }
    }
}

/// Errors which may be returned.
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
pub enum ICoWError
{
    /// An attempt to write to the instance using non exclusive copy-on-write operation
    /// while the exclusive is still going.
    ExclusiveLockPending,

    /// Is issued if "exponential backoff has completed and blocking the thread is advised".
    WouldBlock,

    /// Duplicate write operation prevented.
    AlreadyUpdated,
}

impl fmt::Display for ICoWError
{
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result 
    {
        match self
        {
            Self::ExclusiveLockPending => 
                write!(f, "already exlcusivly locked"),
            Self::WouldBlock => 
                write!(f, "cannot perform try operation due to blocking"),
            Self::AlreadyUpdated => 
                write!(f, "the value has been updated before the current thread"),
        }
    }
}

extern crate crossbeam_utils;
extern crate crossbeam_deque;

/// A lightweight CoW implementation.
#[cfg(all(target_has_atomic = "ptr", feature = "prefer_atomic"))]
pub mod cow;

/// A RwLock based copy-on-write.
#[cfg(any(not(target_has_atomic = "ptr"), not(feature = "prefer_atomic")))]
pub mod cow_mutex;

use std::fmt;

#[cfg(all(target_has_atomic = "ptr", feature = "prefer_atomic"))]
pub use cow::{ICoW, ICoWRead, ICoWCopy, ICoWLock};

#[cfg(any(not(target_has_atomic = "ptr"), not(feature = "prefer_atomic")))]
pub use cow_mutex::{ICoW, ICoWRead, ICoWCopy, ICoWLock};


#[cfg(test)]
mod test
{
    use std::{sync::{mpsc, Arc, LazyLock}, time::{Duration, Instant}};

    use super::*;




    #[test]
    fn test_3()
    {
        #[derive(Debug, Clone)]
        struct Test { s: String };

        let icow = ICoW::new(Test{ s: "test".into() });

        for _ in 0..10
        {
            let s = Instant::now();

            let read0 = icow.read();

            let e = s.elapsed();

            println!("{:?}", e);
        }

    }

    #[test]
    fn test_33()
    {
        #[derive(Debug, Clone)]
        struct Test { s: String };

        let icow = ICoW::new(Test{ s: "test".into() });

        let write_ex = icow.try_clone_exclusivly().unwrap();

        {
        let write_ex_err = icow.try_clone_exclusivly();
        assert_eq!(write_ex_err.is_none(), true);
        }

        let write_ex_err = icow.try_clone();
        assert_eq!(write_ex_err.is_none(), true);

        let read1 = icow.try_read();
        assert_eq!(read1.is_none(), true);

        drop(write_ex);

        drop(icow);

    }

    #[test]
    fn test_4()
    {
        #[derive(Debug, Clone)]
        struct Test { s: u32 }

        let icow = ICoW::new(Test{ s: 1 });

        let read0 = icow.read();
        let read1 = icow.read();

        let mut excl_write = icow.try_clone_exclusivly().unwrap();

        excl_write.s = 5;


        excl_write.commit();

        assert_eq!(read0.s, 1);

        let read3 = icow.read();
        assert_eq!(read3.s, 5);

        let mut writing = icow.try_clone().unwrap();
        writing.s = 4;

        writing.commit().unwrap();

        assert_eq!(read0.s, 1);
        assert_eq!(read3.s, 5);

        let read4 = icow.read();
        assert_eq!(read4.s, 4);
    }

    #[test]
    fn test_5()
    {
        #[derive(Debug, Clone)]
        struct Test { s: u32 }

        let icow = Arc::new(ICoW::new(Test{ s: 1 }));

        let read0 = icow.read();
        let read2 = icow.read();
        
        let c_icow = icow.clone();

        let (se, rc) = mpsc::channel::<()>();
        let handler0 = 
            std::thread::spawn(move || 
                {
                    let mut lock0 = c_icow.try_clone_exclusivly().unwrap();

                    se.send(()).unwrap();

                    lock0.s = 5;

                    std::thread::sleep(Duration::from_micros(2));

                    lock0.commit();
                }
            );

        rc.recv().unwrap();

        let s = Instant::now();

        let read1 = icow.read();
        
        let e = s.elapsed();

        println!("{:?}", e);

        
        assert_eq!(read1.s, 5);
        assert_eq!(read0.s, 1);

        handler0.join().unwrap();

        let weak0 = read0.weak();
        let weak1 = read1.weak();

        drop(read0);
        drop(read1);

        assert_eq!(weak0.upgrade().is_some(), true);
        assert_eq!(weak1.upgrade().is_some(), true);

        drop(read2);
        assert_eq!(weak0.upgrade().is_none(), true);
        assert_eq!(weak1.upgrade().is_some(), true);
    }


    #[test]
    fn test_6()
    {
        #[derive(Debug, Clone)]
        struct Test { s: u32 }

        let icow = Arc::new(ICoW::new(Test{ s: 1 }));

        let read0 = icow.read();
        let read2 = icow.read();
        
        let c_icow = icow.clone();

        let (se, rc) = mpsc::channel::<()>();
        let handler0 = 
            std::thread::spawn(move || 
                {
                    let read2 = c_icow.read();

                    let mut lock0 = c_icow.try_clone_exclusivly().unwrap();

                    se.send(()).unwrap();

                    lock0.s = 5;

                    std::thread::sleep(Duration::from_nanos(50));
                    lock0.commit();

                    let read3 = c_icow.read();

                    assert_eq!(read2.s, 1);
                    assert_eq!(read3.s, 5);
                }
            );

        rc.recv().unwrap();

        for _ in 0..100000000
        {
            let read1 = icow.read();

            if read1.s == 1
            {
                continue;
            }
            else
            {
                break;
            }
        }

        let read1 = icow.read();
        assert_eq!(read1.item.s, 5);

        handler0.join().unwrap();

        return;
    }


    #[test]
    fn test_7()
    {
        #[derive(Debug, Clone)]
        struct TestStruct { s: u32 }

        impl TestStruct
        {
            fn new(s: u32) -> Self
            {
                return Self{ s: s };
            }
        }

        static VAL: LazyLock<ICoW<TestStruct>> = 
            LazyLock::new(|| ICoW::new(TestStruct::new(1))); 
        

        
        
            let borrow1 = VAL.read();
            assert_eq!(borrow1.item.s, 1);

            let (mpsc_send, mpsc_rcv) = mpsc::channel::<u64>();
            let (mpsc_send2, mpsc_rcv2) = mpsc::channel::<u64>();

            let thread1 = 
                std::thread::spawn(move ||
                    {
                        for _ in 0..1000
                        {
                            let _ = mpsc_rcv2.recv();
                            let borrow1 = VAL.read();

                            let mut transaction = VAL.try_clone_exclusivly().unwrap();

                            transaction.s = 5;

                            

                            std::thread::sleep(Duration::from_nanos(1001));
                            transaction.commit();
                            let borrow2 = VAL.read();
                            

                            assert_eq!(borrow1.item.s, 1);

                            assert_eq!(borrow2.item.s, 5);

                            let _ = mpsc_send.send(1);
                        }
                    }
                );
            
            

            for x in 0..1000
            {
                println!("{}", x);
                mpsc_send2.send(1).unwrap();
                let _ = mpsc_rcv.recv();

                let borrow1 = VAL.read();
                assert_eq!(borrow1.s, 5);

                let mut transaction = VAL.try_clone_exclusivly().unwrap();
                transaction.s = 1;
                transaction.commit();

                let borrow1 = VAL.read();
                assert_eq!(borrow1.s, 1);
                
            }
            
            

            thread1.join().unwrap();

            
    }


    #[test]
    fn test_8()
    {
        #[derive(Debug, Clone)]
        struct Test { s: u32 }

        let icow = Arc::new(ICoW::new(Test{ s: 1 }));

        for _ in 0..20
        {
            let read0 = icow.read();
            let read2 = icow.read();
            
            let c_icow = icow.clone();

            //let (se, rc) = mpsc::channel::<()>();
            let handler0 = 
                std::thread::spawn(move || 
                    {
                        let read2 = c_icow.read();

                        let mut lock0 = c_icow.try_clone().unwrap();

                        //se.send(()).unwrap();

                        lock0.s = 5;

                        std::thread::sleep(Duration::from_micros(1));
                        lock0.commit_blocking().unwrap();

                        let read3 = c_icow.read();

                        assert_eq!(read2.s, 1);
                        assert_eq!(read3.s, 5);
                    }
                );

            //rc.recv().unwrap();

            for i in 0..1000000000
            {
                let read1 = icow.read();

                if read1.s == 1
                {
                    continue;
                }
                else
                {
                    println!("{}", i);
                    break;
                }
            }

            let read1 = icow.read();
            assert_eq!(read1.s, 5);

            handler0.join().unwrap();

            let mut lock0 = icow.try_clone().unwrap();
            lock0.s = 1;
            lock0.commit_blocking().unwrap();
        }

        return;
    }
}