rbdc_mysql/
row.rs

1use crate::meta_data::MysqlMetaData;
2use crate::protocol;
3use crate::result_set::{MySqlColumn, MySqlTypeInfo};
4use crate::types::Decode;
5use crate::value::{MySqlValue, MySqlValueFormat, MySqlValueRef};
6use rbdc::db::{MetaData, Row};
7use rbdc::ext::ustr::UStr;
8use rbdc::Error;
9use rbs::Value;
10use std::collections::HashMap;
11use std::sync::Arc;
12use crate::options::MySqlConnectOptions;
13
14/// Implementation of [`Row`] for MySQL.
15#[derive(Debug)]
16pub struct MySqlRow {
17    pub row: protocol::Row,
18    pub format: MySqlValueFormat,
19    pub columns: Arc<Vec<MySqlColumn>>,
20    pub column_names: Arc<HashMap<UStr, (usize, MySqlTypeInfo)>>,
21    pub option: Arc<MySqlConnectOptions>,
22}
23
24impl MySqlRow {
25    pub fn columns(&self) -> &[MySqlColumn] {
26        &self.columns
27    }
28
29    pub fn try_get(&self, index: usize) -> Result<MySqlValueRef<'_>, Error> {
30        let column: &MySqlColumn = &self.columns[index];
31        let value = self.row.get(index as usize);
32        Ok(MySqlValueRef {
33            format: self.format,
34            type_info: column.type_info.clone(),
35            value,
36        })
37    }
38
39    pub fn try_take(&mut self, index: usize) -> Option<MySqlValue> {
40        let column: &MySqlColumn = &self.columns[index];
41        let value = self.row.take(index)?;
42        Some(MySqlValue {
43            value: Some(value),
44            type_info: column.type_info.clone(),
45            format: self.format,
46            option: self.option.clone(),
47        })
48    }
49}
50
51impl Row for MySqlRow {
52    fn meta_data(&self) -> Box<dyn MetaData> {
53        Box::new(MysqlMetaData {
54            inner: self.column_names.clone(),
55        })
56    }
57
58    fn get(&mut self, i: usize) -> Result<Value, Error> {
59        match self.try_take(i) {
60            None => Ok(Value::Null),
61            Some(v) => Value::decode(v),
62        }
63    }
64}