polodb 5.1.4

The server of PoloDB, compatible with MongoDB
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
41
42
43
44
45

use std::sync::{Arc, Mutex};
use polodb_core::Transaction;

#[derive(Clone)]
pub(crate) struct SessionContext {
    inner: Arc<SessionContextInner>,
}

impl SessionContext {

    pub(crate) fn new(txn: Transaction) -> SessionContext {
        SessionContext {
            inner: Arc::new(SessionContextInner {
                txn: Mutex::new(Some(txn)),
            }),
        }
    }

    pub(crate) fn get_transaction(&self) -> Option<Transaction> {
        let txn = self.inner.txn.lock().unwrap();
        txn.clone()
    }

    pub(crate) fn clear_transaction(&self) {
        let mut txn = self.inner.txn.lock().unwrap();
        *txn = None;
    }

}

impl Default for SessionContext {

    fn default() -> Self {
        SessionContext {
            inner: Arc::new(SessionContextInner {
                txn: Mutex::new(None),
            }),
        }
    }

}

struct SessionContextInner {
    txn: Mutex<Option<Transaction>>,
}