Mapping from sqlx down to DataLoader style hashmap.

[hyperliskdev]

struct OrderLoader {
    pool: sqlx::PgPool,
}

#[async_trait::async_trait]
impl Loader<i32> for OrderLoader {
    type Value = Order;
    type Error = Arc<sqlx::Error>;


    async fn load(&self, keys: &[i32]) -> Result<HashMap<i32, Self::Value>, Self::Error> {

        Ok(sqlx::query_as("SELECT * FROM orders WHERE id = ANY($1)")
            .bind(keys)
            .fetch_all(&self.pool)
            .map_ok(|order| (order.id, order))
            .map_err(Arc::new)

        )
    }
}

I want to turn the query_as in the load function into the result as a hashmap. After the fetch, I get a Vec. I cant seem to find a way to handle the error and map the order hashmap using the map_ok and map_err. How can I get the PgRows into the hashmap?