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

use crate::model::algorithms::RegisterFunction;
use async_graphql::dynamic::Object;
use dynamic_graphql::internal::{OutputTypeName, Register, Registry, ResolveOwned, TypeName};
use itertools::Itertools;
use std::{collections::HashMap, sync::MutexGuard};

pub trait AlgorithmEntryPoint<'a>:
    Register + TypeName + OutputTypeName + ResolveOwned<'a> + Sync
{
    fn predefined_algos() -> HashMap<&'static str, RegisterFunction>;
    fn lock_plugins() -> MutexGuard<'static, HashMap<String, RegisterFunction>>;
    fn register_algos(registry: Registry) -> Registry {
        let mut registry = registry;
        let mut object = Object::new(Self::get_type_name());

        for (name, register_algo) in Self::predefined_algos() {
            (registry, object) = register_algo(name, registry, object);
        }

        let mut plugins = Self::lock_plugins();
        let plugin_names = plugins.keys().cloned().collect_vec();
        for name in plugin_names {
            let register_algo = plugins.remove(&name).unwrap();
            (registry, object) = register_algo(&name, registry, object);
        }

        registry.register_type(object)
    }
}