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#[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}