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
46

use crate::error::SqlResult;
use crate::handle::SqlHandle;
use crate::poll::*;
use futures::Async;
use odbc_sys::*;

fn end_transaction_impl<T: SqlHandle>(handle: &mut T, commit: bool) -> SqlPoll {
    let value = if commit { SQL_COMMIT } else { SQL_ROLLBACK };

    let ret = unsafe { SQLEndTran(T::TYPE, handle.handle(), value) };

    match ret {
        SQL_SUCCESS | SQL_SUCCESS_WITH_INFO => Ok(Async::Ready(())),
        SQL_STILL_EXECUTING => Ok(Async::NotReady),
        SQL_ERROR => Err(handle.get_detailed_error(ret)),
        _ => panic!("Unexpected SQLEndTran return code: {:?}", ret),
    }
}

pub trait SqlEndTransaction: SqlHandle {
    fn commit(&mut self) -> SqlResult {
        end_transaction_impl(self, true).unwrap_async()
    }

    fn rollback(&mut self) -> SqlResult {
        end_transaction_impl(self, false).unwrap_async()
    }

    fn end_transaction(&mut self, commit: bool) -> SqlResult {
        end_transaction_impl(self, commit).unwrap_async()
    }

    fn commit_async(self) -> SqlFuture<Self> {
        self.end_transaction_async(true)
    }

    fn rollback_async(self) -> SqlFuture<Self> {
        self.end_transaction_async(false)
    }

    fn end_transaction_async(self, commit: bool) -> SqlFuture<Self> {
        let f = SqlValueFuture::new(self, move |handle| end_transaction_impl(handle, commit));

        flatten_value_future(f)
    }
}