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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
use sqlite3_connector as ffi;
use statement::{State, Statement};
use {Result, Value};
pub struct Cursor {
state: Option<State>,
values: Option<Vec<Value>>,
statement: Statement,
}
impl Cursor {
pub fn bind(&mut self, values: &[Value]) -> Result<()> {
self.state = None;
self.statement.reset()?;
for (i, value) in values.iter().enumerate() {
self.statement.bind(i + 1, value)?;
}
Ok(())
}
#[inline]
pub fn count(&self) -> usize {
self.statement.count()
}
pub fn next(&mut self) -> Result<Option<&[Value]>> {
match self.state {
Some(State::Row) => {}
Some(State::Done) => return Ok(None),
_ => {
self.state = Some(self.statement.next()?);
return self.next();
}
}
self.values = match self.values.take() {
Some(mut values) => {
for (i, value) in values.iter_mut().enumerate() {
*value = self.statement.read(i)?;
}
Some(values)
}
_ => {
let count = self.statement.count();
let mut values = Vec::with_capacity(count);
for i in 0..count {
values.push(self.statement.read(i)?);
}
Some(values)
}
};
self.state = Some(self.statement.next()?);
Ok(Some(self.values.as_ref().unwrap()))
}
#[inline]
pub fn as_raw(&self) -> ffi::Sqlite3StmtHandle {
self.statement.as_raw()
}
}
#[inline]
pub fn new(statement: Statement) -> Cursor {
Cursor {
state: None,
values: None,
statement,
}
}