97 lines
3.2 KiB
Rust
97 lines
3.2 KiB
Rust
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
|
|
use domain::{
|
|
errors::DomainError,
|
|
models::{
|
|
CastMember, CrewMember, ExternalPersonId, IndexableDocument, Person, PersonId,
|
|
},
|
|
ports::{MovieProfileRepository, MovieRepository, PersonCommand, SearchCommand},
|
|
};
|
|
|
|
use crate::commands::EnrichMovieCommand;
|
|
|
|
pub async fn execute(
|
|
movie_repository: &Arc<dyn MovieRepository>,
|
|
profile_repository: &Arc<dyn MovieProfileRepository>,
|
|
person_command: &Arc<dyn PersonCommand>,
|
|
search_command: &Arc<dyn SearchCommand>,
|
|
cmd: EnrichMovieCommand,
|
|
) -> Result<(), DomainError> {
|
|
// 1. Persist the enriched profile (also handles movie_cast, movie_crew, genres, keywords)
|
|
profile_repository.upsert(&cmd.profile).await?;
|
|
|
|
// 2. Upsert persons extracted from cast + crew (no reads — only upsert)
|
|
let persons = extract_persons(&cmd.profile.cast, &cmd.profile.crew);
|
|
if !persons.is_empty() {
|
|
person_command.upsert_batch(&persons).await?;
|
|
}
|
|
|
|
// 3. Fetch the movie for the search index document
|
|
let Some(movie) = movie_repository.get_movie_by_id(&cmd.movie_id).await? else {
|
|
tracing::warn!(movie_id = %cmd.movie_id.value(), "enrich_movie: movie not found after profile upsert");
|
|
return Ok(());
|
|
};
|
|
|
|
// 4. Index the movie in search
|
|
search_command
|
|
.index(IndexableDocument::Movie {
|
|
id: cmd.movie_id.clone(),
|
|
movie: Box::new(movie),
|
|
profile: Some(Box::new(cmd.profile.clone())),
|
|
})
|
|
.await?;
|
|
|
|
// 5. Index each unique person in search (no reads — persons built from in-memory data)
|
|
for person in &persons {
|
|
search_command
|
|
.index(IndexableDocument::Person {
|
|
id: person.id().clone(),
|
|
person: Box::new(person.clone()),
|
|
})
|
|
.await?;
|
|
}
|
|
|
|
tracing::info!(
|
|
movie_id = %cmd.movie_id.value(),
|
|
persons = persons.len(),
|
|
"enrich_movie: profile stored and search index updated"
|
|
);
|
|
Ok(())
|
|
}
|
|
|
|
/// Build unique Person values from cast and crew.
|
|
/// Uses deterministic UUIDv5 so the same tmdb_person_id always maps to the same PersonId.
|
|
/// No DB reads — persons are built entirely from in-memory TMDb data.
|
|
fn extract_persons(cast: &[CastMember], crew: &[CrewMember]) -> Vec<Person> {
|
|
let mut seen: HashMap<u64, Person> = HashMap::new();
|
|
|
|
for member in cast {
|
|
seen.entry(member.tmdb_person_id).or_insert_with(|| {
|
|
let ext = ExternalPersonId::new(format!("tmdb:{}", member.tmdb_person_id));
|
|
Person::new(
|
|
PersonId::from_external(&ext),
|
|
ext,
|
|
member.name.clone(),
|
|
Some("Acting".to_string()),
|
|
member.profile_path.clone(),
|
|
)
|
|
});
|
|
}
|
|
|
|
for member in crew {
|
|
seen.entry(member.tmdb_person_id).or_insert_with(|| {
|
|
let ext = ExternalPersonId::new(format!("tmdb:{}", member.tmdb_person_id));
|
|
Person::new(
|
|
PersonId::from_external(&ext),
|
|
ext,
|
|
member.name.clone(),
|
|
Some(member.department.clone()),
|
|
member.profile_path.clone(),
|
|
)
|
|
});
|
|
}
|
|
|
|
seen.into_values().collect()
|
|
}
|