5 releases
0.1.4 | Sep 25, 2023 |
---|---|
0.1.3 | Sep 22, 2023 |
0.1.2 | Sep 21, 2023 |
0.1.1 | Sep 21, 2023 |
0.1.0 | Sep 2, 2023 |
#885 in Asynchronous
148 downloads per month
Used in rnacos
21KB
399 lines
BeanFactory
一个依赖注入容器。
对actix的actor,支持注入依赖的对象到actor对象中。
样例
use std::{collections::HashMap, sync::Arc};
use actix::prelude::*;
use bean_factory::bean;
use bean_factory::BeanDefinition;
use bean_factory::BeanFactory;
use bean_factory::Inject;
#[bean(actor)]
#[derive(Default)]
pub struct ConfigService {
pub(crate) config_map: HashMap<Arc<String>, Arc<String>>,
}
impl Actor for ConfigService {
type Context = Context<Self>;
}
#[derive(Debug, Message)]
#[rtype(result = "anyhow::Result<ConfigResult>")]
pub enum ConfigCmd {
Set(Arc<String>, Arc<String>),
Query(Arc<String>),
}
pub enum ConfigResult {
None,
Value(Arc<String>),
}
impl Handler<ConfigCmd> for ConfigService {
type Result = anyhow::Result<ConfigResult>;
fn handle(&mut self, msg: ConfigCmd, _ctx: &mut Self::Context) -> Self::Result {
match msg {
ConfigCmd::Set(key, val) => {
self.config_map.insert(key, val);
Ok(ConfigResult::None)
}
ConfigCmd::Query(key) => {
if let Some(v) = self.config_map.get(&key) {
Ok(ConfigResult::Value(v.clone()))
} else {
Ok(ConfigResult::None)
}
}
}
}
}
#[bean(inject)]
#[derive(Default)]
pub struct ConfigApi {
pub(crate) config_service: Option<Addr<ConfigService>>,
}
impl Inject for ConfigApi {
type Context = Context<Self>;
fn inject(
&mut self,
factory_data: bean_factory::FactoryData,
_factory: bean_factory::BeanFactory,
_ctx: &mut Self::Context,
) {
self.config_service = factory_data.get_actor();
if self.config_service.is_some() {
println!("ConfigApi:inject success");
} else {
println!("ConfigApi:inject none");
}
println!("ConfigApi:inject complete");
}
}
impl Actor for ConfigApi {
type Context = Context<Self>;
}
impl Handler<ConfigCmd> for ConfigApi {
type Result = ResponseActFuture<Self, anyhow::Result<ConfigResult>>;
fn handle(&mut self, msg: ConfigCmd, _ctx: &mut Self::Context) -> Self::Result {
let config_service = self.config_service.clone();
let fut = async {
if let Some(addr) = config_service {
println!("inject success. use inject config_service handle msg");
addr.send(msg).await?
} else {
Err(anyhow::anyhow!("inject failed. config_service is none"))
}
}
.into_actor(self); //.map(|r,_act,_ctx|{r});
Box::pin(fut)
}
}
#[actix::main]
async fn main() -> anyhow::Result<()> {
std::env::set_var("RUST_LOG","info");
env_logger::builder().init();
let factory = BeanFactory::new();
factory.register(BeanDefinition::actor_with_inject_from_default::<ConfigApi>());
factory.register(BeanDefinition::actor_from_default::<ConfigService>());
let _factory_data = factory.init().await;
let api_addr: Addr<ConfigApi> = factory.get_actor().await.unwrap();
let key = Arc::new("key".to_owned());
api_addr.do_send(ConfigCmd::Set(
key.clone(),
Arc::new("test value".to_owned()),
));
match api_addr.send(ConfigCmd::Query(key.clone())).await?? {
ConfigResult::None => println!("not found value"),
ConfigResult::Value(val) => println!("query value:{}", &val),
};
Ok(())
}
Dependencies
~6–14MB
~177K SLoC