sea-orm-sync 2.0.0-rc.38

🐚 The sync version of SeaORM
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
43
44
45
46
47
48

use crate::{EntitySchemaInfo, Schema, SchemaBuilder};
use tracing::debug;

#[derive(derive_more::Debug)]
/// The data structure submitted by your Entity to the Entity Registry.
pub struct EntityRegistry {
    /// Please use `module_path!()`.
    pub module_path: &'static str,
    /// Function that returns schema info for the Entity.
    #[debug(skip)]
    pub schema_info: fn(&Schema) -> EntitySchemaInfo,
}

inventory::collect!(EntityRegistry);

/// Macro to register an Entity
pub use inventory::submit as register_entity;

impl EntityRegistry {
    /// Builds a schema from all the registered entities, filtering by prefix.
    pub fn build_schema(schema: Schema, prefix: &str) -> SchemaBuilder {
        let mut schema = SchemaBuilder::new(schema);
        let mut string;
        let mut prefix = prefix.trim_end_matches("*");
        if !prefix.contains("::") {
            string = format!("{prefix}::");
            prefix = &string;
        }
        if let Some((left, right)) = prefix.split_once("::") {
            if left.contains("-") {
                // convert crate name to module path
                let left = left.replace('-', "_");
                string = format!("{left}::{right}");
                prefix = &string;
            }
        }
        debug!("Registering entities with prefix `{prefix}`");
        for entity in inventory::iter::<crate::EntityRegistry>() {
            if entity.module_path.starts_with(prefix) {
                schema.register_entity((entity.schema_info)(schema.helper()));
                debug!("Registered {}", entity.module_path);
            } else {
                debug!("Skipped {}", entity.module_path);
            }
        }
        schema
    }
}