sqlx_build_trust_postgres/
row.rs

1use 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
11/// Implementation of [`Row`] for PostgreSQL.
12pub 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}