Change to map

This commit is contained in:
Jett Chen 2024-11-21 19:58:15 +08:00
parent d60b70ffb4
commit b806b28a6e

View file

@ -104,15 +104,12 @@ pub fn list_devices(user_id: i64) -> Result<Vec<String>> {
let conn = rusqlite::Connection::open(PATH)?;
let mut stmt = conn.prepare("SELECT name FROM device WHERE user_id = ?")?;
let devices = stmt.query_map([user_id], |row| {
let name: String = row.get(0)?;
Ok(name)
})?;
let mut result = Vec::new();
for device in devices {
result.push(device?);
}
let result: Vec<String> = stmt
.query_map([user_id], |row| {
let name: String = row.get(0)?;
Ok(name)
})?
.collect::<Result<Vec<_>, _>>()?;
Ok(result)
}