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
use statement::{State, Statement, Bindable, Readable};
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;
try!(self.statement.reset());
for (i, value) in values.iter().enumerate() {
try!(self.statement.bind(i + 1, value));
}
Ok(())
}
pub fn next(&mut self) -> Result<Option<&[Value]>> {
match self.state {
Some(State::Row) => {},
Some(State::Done) => return Ok(None),
_ => {
self.state = Some(try!(self.statement.next()));
return self.next();
},
}
let values = match self.values.take() {
Some(mut values) => {
for (i, value) in values.iter_mut().enumerate() {
match value {
&mut Value::Binary(ref mut value) => {
*value = try!(self.statement.read(i));
},
&mut Value::Float(ref mut value) => {
*value = try!(self.statement.read(i));
},
&mut Value::Integer(ref mut value) => {
*value = try!(self.statement.read(i));
},
&mut Value::String(ref mut value) => {
*value = try!(self.statement.read(i));
},
&mut Value::Null => {},
}
}
values
},
_ => {
let count = self.statement.columns();
let mut values = Vec::with_capacity(count);
for i in 0..count {
values.push(try!(self.statement.read(i)));
}
values
},
};
self.state = Some(try!(self.statement.next()));
self.values = Some(values);
Ok(Some(self.values.as_ref().unwrap()))
}
#[inline]
pub fn columns(&self) -> usize {
self.statement.columns()
}
}
#[inline]
pub fn new<'l>(statement: Statement<'l>) -> Cursor<'l> {
Cursor { state: None, values: None, statement: statement }
}