sqlx_build_trust_postgres/
row.rs1use crate::column::ColumnIndex;
2use crate::error::Error;
3use crate::message::DataRow;
4use crate::statement::PgStatementMetadata;
5use crate::value::PgValueFormat;
6use crate::{PgColumn, PgValueRef, Postgres};
7use std::sync::Arc;
8
9pub(crate) use sqlx_core::row::Row;
10
11pub struct PgRow {
13 pub(crate) data: DataRow,
14 pub(crate) format: PgValueFormat,
15 pub(crate) metadata: Arc<PgStatementMetadata>,
16}
17
18impl Row for PgRow {
19 type Database = Postgres;
20
21 fn columns(&self) -> &[PgColumn] {
22 &self.metadata.columns
23 }
24
25 fn try_get_raw<I>(&self, index: I) -> Result<PgValueRef<'_>, Error>
26 where
27 I: ColumnIndex<Self>,
28 {
29 let index = index.index(self)?;
30 let column = &self.metadata.columns[index];
31 let value = self.data.get(index);
32
33 Ok(PgValueRef {
34 format: self.format,
35 row: Some(&self.data.storage),
36 type_info: column.type_info.clone(),
37 value,
38 })
39 }
40}
41
42impl ColumnIndex<PgRow> for &'_ str {
43 fn index(&self, row: &PgRow) -> Result<usize, Error> {
44 row.metadata
45 .column_names
46 .get(*self)
47 .ok_or_else(|| Error::ColumnNotFound((*self).into()))
48 .map(|v| *v)
49 }
50}