test(domain): add transpilter to application

counit-server/src/application.rs

@@ -2,13 +2,16 @@ use std::sync::Arc;
 use anyhow::{bail, Result};
 use tracing::warn;
 use crate::configuration::Configuration;
+use crate::domain::domain_transpiler::DomainTranspiler;
 use crate::repository::semantic::Semantic;
 
 #[derive(Clone)]
 pub struct Application {
     /// User-provided configuration
     pub config: Arc<Configuration>,
 
+    pub transpiler: Arc<DomainTranspiler>,
+
     /// Semantic search subsystem
     pub(crate) semantic: Option<Semantic>,
 }
@@ -32,8 +35,16 @@ impl Application {
             }
         };
 
+        let mut transpiler: Arc<DomainTranspiler>;
+        if config.domain_language_dir.is_some() {
+            transpiler = Arc::new(DomainTranspiler::new(config.domain_language_dir.as_ref().unwrap()));
+        } else {
+            transpiler = Arc::new(DomainTranspiler::empty());
+        };
+
         Ok(Application {
             config,
+            transpiler,
             semantic,
         })
     }

counit-server/src/domain/domain_loader.rs → counit-server/src/domain/domain_transpiler.rs

@@ -6,13 +6,29 @@ pub struct DomainTranspiler {
 }
 
 impl DomainTranspiler {
-    pub fn new() -> Self {
+
+    pub fn empty() -> Self {
         DomainTranspiler {
             domain_records: Vec::new(),
         }
     }
 
-    pub fn load<P: AsRef<Path>>(&mut self, path: P) {
+    pub fn new<P: AsRef<Path>>(path: P) -> Self {
+        let mut transpiler = DomainTranspiler {
+            domain_records: Vec::new(),
+        };
+
+        // check path is exists
+        if !path.as_ref().exists() {
+            return transpiler;
+        }
+
+        transpiler.load(path);
+
+        return transpiler;
+    }
+
+    fn load<P: AsRef<Path>>(&mut self, path: P) {
         use walkdir::WalkDir;
         for entry in WalkDir::new(path) {
             let entry = entry.unwrap();
@@ -80,8 +96,7 @@ mod tests {
         let model_dir = domain_dir();
 
 
-        let mut loader = DomainTranspiler::new();
-        loader.load(model_dir);
+        let mut loader = DomainTranspiler::new(model_dir);
         assert_eq!(loader.domain_records.len(), 29);
     }
 
@@ -97,8 +112,7 @@ mod tests {
     fn transpile() {
         let model_dir = domain_dir();
 
-        let mut loader = DomainTranspiler::new();
-        loader.load(model_dir);
+        let mut loader = DomainTranspiler::new(model_dir);
         assert_eq!(loader.domain_records.len(), 29);
 
         assert_eq!(loader.transpile("本币"), "本币(Domestic Currency)");

counit-server/src/domain/mod.rs

@@ -1,2 +1,2 @@
 pub mod domain_record;
-pub mod domain_loader;
+pub mod domain_transpiler;

counit-server/src/main.rs

@@ -8,6 +8,7 @@ use tracing::info;
 
 use crate::application::Application;
 use crate::configuration::Configuration;
+use crate::domain::domain_transpiler::DomainTranspiler;
 use crate::server::{agent_api, archguard_api, semantic_api, translate_api};
 
 pub mod server;