mecomp-daemon 0.7.2

RPC server for the Mecomp, the Metadata Enhanced Collection Orientated Music Player
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use mecomp_storage::{
    db::schemas::{
        RecordId,
        analysis::Analysis,
        song::{Song, SongId},
    },
    errors::StorageResult,
};
use surrealdb::{Connection, Surreal};

use super::get_songs_from_things;

/// Get the 'n' most similar songs to the given list of things
///
/// # Errors
///
/// Returns an error if there is an issue with the database
#[inline]
pub async fn get_similar<C: Connection>(
    db: &Surreal<C>,
    things: Vec<RecordId>,
    n: u32,
    settings: &mecomp_core::config::AnalysisSettings,
) -> StorageResult<Vec<Song>> {
    if things.is_empty() || n == 0 {
        return Ok(vec![]);
    }

    // go through the list, and get songs for each thing (depending on what it is)
    let songs: Vec<SongId> = get_songs_from_things(db, &things)
        .await?
        .into_iter()
        .map(|s| s.id)
        .collect();

    // whether to use feature-based or embedding-based analysis
    let use_embeddings = matches!(settings.kind, mecomp_core::config::AnalysisKind::Embedding);

    let analyses = Analysis::read_for_songs(db, songs).await?;
    let neighbors = Analysis::nearest_neighbors_to_many(db, analyses, n, use_embeddings).await?;
    Analysis::read_songs(db, neighbors.into_iter().map(|a| a.id).collect()).await
}