Как правильно обработать rusqlite mappedrows?

Question

[Somewhere Intech]

Сделал по примеру, но ругается на типы данных, почему?

#[derive(Debug)]
pub struct Candle{
	pub timestamp: i32,
	pub open: f64,
	pub close:f64,
	pub min: f64,
	pub max: f64,
	pub ticks: Vec<CandleTick>
}
fn read()->Result<Vec<Candle>>{
	let conn = Connection::open("candles.db")?;
	let mut stmt = conn.prepare("SELECT * from candles limit 10")?;
    let res :Vec<Candle>= stmt.query_map(&[], |row| {
        Candle {
            timestamp: row.get(1).unwrap(),
            open: row.get(2).unwrap(),
            close: row.get(3).unwrap(),
            min: row.get(4).unwrap(),
            max: row.get(5).unwrap(),
            ticks: Vec::new()
        }
    })?;
    Ok(res)
}

unwrap добавил тоже из за ругани что должен быть Result
Вобщем ответ компилятора:

error[E0308]: mismatched types
  --> src/lib.rs:37:9
   |
37 | /         Candle {
38 | |             timestamp: row.get(1).unwrap(),
39 | |             open: row.get(2).unwrap(),
40 | |             close: row.get(3).unwrap(),
...  |
43 | |             ticks: Vec::new()
44 | |         }
   | |_________^ expected enum `std::result::Result`, found struct `Candle`
   |
   = note: expected type `std::result::Result<_, rusqlite::error::Error>`
              found type `Candle`

error[E0308]: try expression alternatives have incompatible types
  --> src/lib.rs:36:27
   |
36 |       let res :Vec<Candle>= stmt.query_map(&[], |row| {
   |  ___________________________^
37 | |         Candle {
38 | |             timestamp: row.get(1).unwrap(),
39 | |             open: row.get(2).unwrap(),
...  |
44 | |         }
45 | |     })?;
   | |_______^ expected struct `std::vec::Vec`, found struct `rusqlite::row::MappedRows`
   |
   = note: expected type `std::vec::Vec<Candle>`
              found type `rusqlite::row::MappedRows<'_, [closure@src/lib.rs:36:47: 45:6]>`

Тип для res обозначил "наобум" конечно..

Answer 1

[Tyranron]

Туториал устарел. Лучше всегда читать официальную доку. Она в Rust очень удобная и в сообществе принято её хорошо сдабривать примерами.

Первая ошибка говорит о том, что Вы возвращаете Candle {...}, в то время как ожидается Result (это как раз устаревшая часть туториала). Просто сделайте Ok(Candle {...}) (пример с документации как раз подобное и показывает).

Вторая ошибка говорит о том, что Вы пытаетесь возвращаемый тип rusqlite::row::MappedRows подставить туда, где явно ожидаете Vec. Rust статически и строго типизированный язык, так не получится. По сути, Вам там просто нужно добавить .collect() после ?. Так как MappedRowsреализует Iterator, то .collect() как раз пробежится по всем значениям итератора и соберёт их в Vec, который Вам и нужен.

Comments

[Somewhere Intech]

Спасибо, правда collect() выдал ошибку: не реализован trait итератора:

error[E0277]: a collection of type `std::vec::Vec<Candle>` cannot be built from an iterator over elements of type `std::result::Result<Candle, rusqlite::error::Error>`
  --> src/lib.rs:47:9
   |
47 |     })?.collect();
   |         ^^^^^^^ a collection of type `std::vec::Vec<Candle>` cannot be built from `std::iter::Iterator<Item=std::result::Result<Candle, rusqlite::error::Error>>`
   |
   = help: the trait `std::iter::FromIterator<std::result::Result<Candle, rusqlite::error::Error>>` is not implemented for `std::vec::Vec<Candle>`

сделал проще:

fn read()->Result<Vec<Candle>>{
	let conn = Connection::open("candles.db")?;
	let mut stmt = conn.prepare("SELECT * from candles")?;
    let rows = stmt.query_map(params![], |row| {
        Ok(Candle {
            timestamp: row.get(1)?,
            open: row.get(2)?,
            close: row.get(3)?,
            min: row.get(4)?,
            max: row.get(5)?,
            ticks: Vec::new()
        })
    })?;
    let mut result:Vec<Candle> = Vec::new();
    for row in rows{
    	result.push(row.unwrap());
    }
    Ok(result)
}

Можно улучшить конструкцию?

[Somewhere Intech]

Сорян, не в трэйте дело, промапил и все ок:
?.map(|x|{x.unwrap()}).collect();
Правда вышло тяжелее по времени на ~15%

[Tyranron]

Johnny Lowhunter, ага, не заметил что там итератор по резалтам вместо айтемов.

Ваш последний вариант лучше всего, но его ещё можно улучшить убрав unnecessary function wrapping:
?.map(Result::unwrap).collect();

По поводу скорости не парьтесь, --release вообще всё меняет. А если Вам действительно нужна скорость, то подобные микро-оптимизации имеют смысл если профайлер показывает что они действительно нужны.

[Tyranron]

Johnny Lowhunter, кстати, крайне рекомендую взять за привычку прогонять cargo clippy, особенно с pedantic уровнём. Он учит многому хорошему, особенно в части комбинаторов. Например, из моей практики недавно: вместо .map(f1).unwrap_or_else(f2) удобнее использовать сразу .map_or_else(f2, f1).