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, &param)?;
    /// 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<'p> Raii<'p, 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),
        }
    }
}