Function parse_results

src/pglite/parse.rs:11–65

(
    messages: &[BackendMessage],
    default_parsers: &ParserMap,
    options: Option<&QueryOptions>,
    blob: Option<Vec<u8>>,
)

};

pub fn parse_results(
    messages: &[BackendMessage],
    default_parsers: &ParserMap,
    options: Option<&QueryOptions>,
    blob: Option<Vec<u8>>,
) -> Vec<Results> {
    let mut result_sets: Vec<Results> = Vec::new();
    let mut current_fields: Vec<FieldInfo> = Vec::new();
    let mut current_rows: Vec<Value> = Vec::new();
    let mut affected_rows = 0usize;

    let empty_parsers = HashMap::new();
    let (row_mode, parsers_override) = options
        .map(|opts| (opts.row_mode, &opts.parsers))
        .unwrap_or((None, &empty_parsers));

    let parser_lookup = ParserLookup::new(default_parsers, parsers_override);

    for message in messages {
        match message {
            BackendMessage::RowDescription(desc) => {
                current_fields = map_fields(desc);
            }
            BackendMessage::DataRow(row) => {
                if current_fields.is_empty() {
                    continue;
                }
                let row_value = map_row(row, &current_fields, &parser_lookup, row_mode);
                current_rows.push(row_value);
            }
            BackendMessage::CommandComplete(cmd) => {
                affected_rows += retrieve_row_count(cmd);
                result_sets.push(Results {
                    rows: std::mem::take(&mut current_rows),
                    fields: current_fields.clone(),
                    affected_rows: Some(affected_rows),
                    blob: blob.clone(),
                });
                current_fields.clear();
            }
            _ => {}
        }
    }

    if result_sets.is_empty() {
        result_sets.push(Results {
            rows: Vec::new(),
            fields: Vec::new(),
            affected_rows: Some(0),
            blob,
        })
    }

    result_sets
}

pub fn parse_describe_statement_results(messages: &[BackendMessage]) -> Vec<i32> {
    messages