Introdução a pipelines de agregação em Rust

1
{
2
   '_id': ObjectId('573a1392f29313caabcdb497'),
3
   'awards': {'nominations': 7,
4
               'text': 'Won 1 Oscar. Another 2 wins & 7 nominations.',
5
               'wins': 3},
6
   'cast': ['Janet Gaynor', 'Fredric March', 'Adolphe Menjou', 'May Robson'],
7
   'countries': ['USA'],
8
   'directors': ['William A. Wellman', 'Jack Conway'],
9
   'fullplot': 'Esther Blodgett is just another starry-eyed farm kid trying to '
10
               'break into the movies. Waitressing at a Hollywood party, she '
11
               'catches the eye of alcoholic star Norman Maine, is given a test, '
12
               'and is caught up in the Hollywood glamor machine (ruthlessly '
13
               'satirized). She and her idol Norman marry; but his career '
14
               'abruptly dwindles to nothing',
15
   'genres': ['Drama'],
16
   'imdb': {'id': 29606, 'rating': 7.7, 'votes': 5005},
17
   'languages': ['English'],
18
   'lastupdated': '2015-09-01 00:55:54.333000000',
19
   'plot': 'A young woman comes to Hollywood with dreams of stardom, but '
20
            'achieves them only with the help of an alcoholic leading man whose '
21
            'best days are behind him.',
22
   'poster': 'https://m.media-amazon.com/images/M/MV5BMmE5ODI0NzMtYjc5Yy00MzMzLTk5OTQtN2Q3MzgwOTllMTY3XkEyXkFqcGdeQXVyNjc0MzMzNjA@._V1_SY1000_SX677_AL_.jpg',
23
   'rated': 'NOT RATED',
24
   'released': datetime.datetime(1937, 4, 27, 0, 0),
25
   'runtime': 111,
26
   'title': 'A Star Is Born',
27
   'tomatoes': {'critic': {'meter': 100, 'numReviews': 11, 'rating': 7.4},
28
               'dvd': datetime.datetime(2004, 11, 16, 0, 0),
29
               'fresh': 11,
30
               'lastUpdated': datetime.datetime(2015, 8, 26, 18, 58, 34),
31
               'production': 'Image Entertainment Inc.',
32
               'rotten': 0,
33
               'viewer': {'meter': 79, 'numReviews': 2526, 'rating': 3.6},
34
               'website': 'http://www.vcientertainment.com/Film-Categories?product_id=73'},
35
   'type': 'movie',
36
   'writers': ['Dorothy Parker (screen play)',
37
               'Alan Campbell (screen play)',
38
               'Robert Carson (screen play)',
39
               'William A. Wellman (from a story by)',
40
               'Robert Carson (from a story by)'],
41
   'year': 1937}

1
// Load the MongoDB connection string from an environment variable:
2
let client_uri =
3
   env::var("MONGODB_URI").expect("You must set the MONGODB_URI environment var!");
4
5
// An extra line of code to work around a DNS issue on Windows:
6
let options =
7
   ClientOptions::parse_with_resolver_config(&client_uri, ResolverConfig::cloudflare())
8
      .await?;
9
let client = mongodb::Client::with_options(options)?;
10
11
// Get the 'movies' collection from the 'sample_mflix' database:
12
let movies = client.database("sample_mflix").collection("movies");

1
// Usually implemented outside your main function:
2
#[derive(Deserialize)]
3
struct MovieSummary {
4
   title: String,
5
   cast: Vec<String>,
6
   year: i32,
7
}
8
9
impl fmt::Display for MovieSummary {
10
   fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
11
      write!(
12
         f,
13
         "{}, {}, {}",
14
         self.title,
15
         self.cast.get(0).unwrap_or(&"- no cast -".to_owned()),
16
         self.year
17
      )
18
   }
19
}
20
21
// Inside main():
22
let pipeline = vec![
23
   doc! {
24
      // filter on movie title:
25
      "$match": {
26
         "title": "A Star Is Born"
27
      }
28
   },
29
   doc! {
30
      // sort by year, ascending:
31
      "$sort": {
32
         "year": 1
33
      }
34
   },
35
];
36
37
// Look up "A Star is Born" in ascending year order:
38
let mut results = movies.aggregate(pipeline, None).await?;
39
// Loop through the results, convert each of them to a MovieSummary, and then print out.
40
while let Some(result) = results.next().await {
41
   // Use serde to deserialize into the MovieSummary struct:
42
   let doc: MovieSummary = bson::from_document(result?)?;
43
   println!("* {}", doc);
44
}

1
// Sort by year, descending:
2
let stage_sort_year_descending = doc! {
3
   "$sort": {
4
      "year": -1
5
   }
6
};
7
8
// Limit to 1 document:
9
let stage_limit_1 = doc! { "$limit": 1 };
10
11
let pipeline = vec![stage_match_title, stage_sort_year_descending, stage_limit_1];

1
{
2
   '_id': ObjectId('5a9427648b0beebeb69579d3'),
3
   'movie_id': ObjectId('573a1390f29313caabcd4217'),
4
   'date': datetime.datetime(1983, 4, 27, 20, 39, 15),
5
   'email': 'cameron_duran@fakegmail.com',
6
   'name': 'Cameron Duran',
7
   'text': 'Quasi dicta culpa asperiores quaerat perferendis neque. Est animi '
8
           'pariatur impedit itaque exercitationem.'}

1
* Midnight, Claudette Colbert, 1939
2
  - Sansa Stark <sansa_stark@fakegmail.com>: Error ex culpa dignissimos assumenda voluptates vel. Qui inventore 
3
  - Theon Greyjoy <theon_greyjoy@fakegmail.com>: Animi dolor minima culpa sequi voluptate. Possimus necessitatibu
4
  - Donna Smith <donna_smith@fakegmail.com>: Et esse nulla ducimus tempore aliquid. Suscipit iste dignissimos v

1
// Define a struct to hold grouped data by year:
2
#[derive(Debug, Deserialize)]
3
struct YearSummary {
4
   _id: i32,
5
   #[serde(default)]
6
   movie_count: i64,
7
   #[serde(default)]
8
   movie_titles: Vec<String>,
9
}
10
11
// Some movies have "year" values ending with 'è'.
12
// This stage will filter them out:
13
let stage_filter_valid_years = doc! {
14
   "$match": {
15
      "year": {
16
         "$type": "number",
17
      }
18
   }
19
};
20
21
/*
22
* Group movies by year, producing 'year-summary' documents that look like:
23
* {
24
*     '_id': 1917,
25
* }
26
*/
27
let stage_group_year = doc! {
28
   "$group": {
29
      "_id": "$year",
30
   }
31
};
32
33
let pipeline = vec![stage_filter_valid_years, stage_group_year];
34
35
// Loop through the 'year-summary' documents:
36
let mut results = movies.aggregate(pipeline, None).await?;
37
// Loop through the yearly summaries and print their debug representation:
38
while let Some(result) = results.next().await {
39
   let doc: YearSummary = bson::from_document(result?)?;
40
   println!("* {:?}", doc);
41
}

1
* YearSummary { _id: 1986, movie_count: 206, movie_titles: ["Defense of the Realm", "F/X", "Mala Noche", "Witch from Nepal", ... ]}