-
Notifications
You must be signed in to change notification settings - Fork 29
/
input.rs
126 lines (116 loc) · 4.33 KB
/
input.rs
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
115
116
117
118
119
120
121
122
123
124
125
126
use super::types::OdbcType;
use odbc_safe::AutocommitMode;
use statement::types::EncodedValue;
use {ffi, Handle, Raii, Result, Return, Statement};
impl<'a, 'b, S, R, AC: AutocommitMode> Statement<'a, 'b, S, R, AC> {
/// Binds a parameter to a parameter marker in an SQL statement.
///
/// # Result
/// This method will destroy the statement and create a new one which may not outlive the bound
/// parameter. This is to ensure that the statement will not dereference an invalid pointer
/// during execution.
///
/// # Arguments
/// * `parameter_index` - Index of the marker to bind to the parameter. Starting at `1`
/// * `value` - Reference to bind to the marker
///
/// # Example
/// ```
/// # use odbc::*;
/// # fn do_odbc_stuff() -> std::result::Result<(), Box<std::error::Error>> {
/// let env = create_environment_v3().map_err(|e| e.unwrap())?;
/// let conn = env.connect("TestDataSource", "", "")?;
/// let stmt = Statement::with_parent(&conn)?;
/// let param = 1968;
/// let stmt = stmt.bind_parameter(1, ¶m)?;
/// let sql_text = "SELECT TITLE FROM MOVIES WHERE YEAR = ?";
/// if let Data(mut stmt) = stmt.exec_direct(sql_text)? {
/// // ...
/// }
/// # Ok(())
/// # }
/// ```
pub fn bind_parameter<'c, T>(
mut self,
parameter_index: u16,
value: &'c T,
) -> Result<Statement<'a, 'c, S, R, AC>>
where
T: OdbcType<'c>,
T: ?Sized,
'b: 'c,
{
let ind = if value.value_ptr() == 0 as *const Self as ffi::SQLPOINTER {
ffi::SQL_NULL_DATA
} else {
value.column_size() as ffi::SQLLEN
};
let ind_ptr = self.param_ind_buffers.alloc(parameter_index as usize, ind);
//the result of value_ptr is changed per calling.
//binding and saving must have the same value.
let enc_value = value.encoded_value();
self.raii
.bind_input_parameter(parameter_index, value, ind_ptr, &enc_value)
.into_result(&self)?;
// save encoded value to avoid memory reuse.
if enc_value.has_value() {
self.encoded_values.push(enc_value);
}
Ok(self)
}
/// Releasing all parameter buffers set by `bind_parameter`. This method consumes the statement
/// and returns a new one those lifetime is no longer limited by the buffers bound.
pub fn reset_parameters(mut self) -> Result<Statement<'a, 'a, S, R, AC>> {
self.param_ind_buffers.clear();
self.encoded_values.clear();
self.raii.reset_parameters().into_result(&mut self)?;
Ok(Statement::with_raii(self.raii))
}
}
impl Raii<ffi::Stmt> {
fn bind_input_parameter<'c, T>(
&mut self,
parameter_index: u16,
value: &'c T,
str_len_or_ind_ptr: *mut ffi::SQLLEN,
enc_value: &EncodedValue,
) -> Return<()>
where
T: OdbcType<'c>,
T: ?Sized,
{
//if encoded value exists, use it.
let (column_size, value_ptr) = if enc_value.has_value() {
(enc_value.column_size(), enc_value.value_ptr())
} else {
(value.column_size(), value.value_ptr())
};
match unsafe {
ffi::SQLBindParameter(
self.handle(),
parameter_index,
ffi::SQL_PARAM_INPUT,
T::c_data_type(),
T::sql_data_type(),
column_size,
value.decimal_digits(),
value_ptr,
0, // buffer length
str_len_or_ind_ptr, // Note that this ptr has to be valid until statement is executed
)
} {
ffi::SQL_SUCCESS => Return::Success(()),
ffi::SQL_SUCCESS_WITH_INFO => Return::SuccessWithInfo(()),
ffi::SQL_ERROR => Return::Error,
r => panic!("Unexpected return from SQLBindParameter: {:?}", r),
}
}
fn reset_parameters(&mut self) -> Return<()> {
match unsafe { ffi::SQLFreeStmt(self.handle(), ffi::SQL_RESET_PARAMS) } {
ffi::SQL_SUCCESS => Return::Success(()),
ffi::SQL_SUCCESS_WITH_INFO => Return::SuccessWithInfo(()),
ffi::SQL_ERROR => Return::Error,
r => panic!("SQLFreeStmt returned unexpected result: {:?}", r),
}
}
}