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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
use ffi;
use statement::{State, Statement};
use {Result, Value};
pub struct Cursor<'l> {
state: Option<State>,
values: Option<Vec<Value>>,
statement: Statement<'l>,
}
impl<'l> Cursor<'l> {
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(())
}
pub fn bind_by_name<S, V>(&mut self, values: V) -> Result<()>
where
S: AsRef<str>,
V: IntoIterator<Item = (S, Value)>,
{
self.state = None;
self.statement.reset()?;
for (name, value) in values {
if let Some(i) = self.statement.parameter_index(name.as_ref())? {
self.statement.bind(i, &value)?;
}
}
Ok(())
}
#[inline]
pub fn column_count(&self) -> usize {
self.statement.column_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.column_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) -> *mut ffi::sqlite3_stmt {
self.statement.as_raw()
}
#[deprecated(since = "0.26.0", note = "Please use `column_count` instead.")]
pub fn count(&self) -> usize {
self.column_count()
}
}
#[inline]
pub fn new<'l>(statement: Statement<'l>) -> Cursor<'l> {
Cursor {
state: None,
values: None,
statement: statement,
}
}