mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-09 22:59:59 +00:00
houseme
110 lines
3.9 KiB
Rust
110 lines
3.9 KiB
Rust
use notify::arn::TargetID;
|
|
use notify::global::notification_system;
|
|
use notify::{
|
|
init_logger, BucketNotificationConfig, Event, EventName, LogLevel, NotificationError,
|
|
};
|
|
use std::time::Duration;
|
|
use tracing::info;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), NotificationError> {
|
|
init_logger(LogLevel::Debug);
|
|
|
|
let system = notification_system();
|
|
|
|
// --- 初始配置 (Webhook 和 MQTT) ---
|
|
let mut config = notify::Config::new();
|
|
|
|
// Webhook target configuration
|
|
let mut webhook_kvs = notify::KVS::new();
|
|
webhook_kvs.set("enable", "on");
|
|
webhook_kvs.set("endpoint", "http://127.0.0.1:3020/webhook");
|
|
webhook_kvs.set("auth_token", "secret-token");
|
|
// webhook_kvs.set("queue_dir", "/tmp/data/webhook");
|
|
webhook_kvs.set(
|
|
"queue_dir",
|
|
"/Users/qun/Documents/rust/rustfs/notify/logs/webhook",
|
|
);
|
|
webhook_kvs.set("queue_limit", "10000");
|
|
let mut webhook_targets = std::collections::HashMap::new();
|
|
webhook_targets.insert("1".to_string(), webhook_kvs);
|
|
config.insert("notify_webhook".to_string(), webhook_targets);
|
|
|
|
// MQTT target configuration
|
|
let mut mqtt_kvs = notify::KVS::new();
|
|
mqtt_kvs.set("enable", "on");
|
|
mqtt_kvs.set("broker", "mqtt://localhost:1883");
|
|
mqtt_kvs.set("topic", "rustfs/events");
|
|
mqtt_kvs.set("qos", "1"); // AtLeastOnce
|
|
mqtt_kvs.set("username", "test");
|
|
mqtt_kvs.set("password", "123456");
|
|
// webhook_kvs.set("queue_dir", "/tmp/data/mqtt");
|
|
mqtt_kvs.set(
|
|
"queue_dir",
|
|
"/Users/qun/Documents/rust/rustfs/notify/logs/mqtt",
|
|
);
|
|
mqtt_kvs.set("queue_limit", "10000");
|
|
|
|
let mut mqtt_targets = std::collections::HashMap::new();
|
|
mqtt_targets.insert("1".to_string(), mqtt_kvs);
|
|
config.insert("notify_mqtt".to_string(), mqtt_targets);
|
|
|
|
// 加载配置并初始化系统
|
|
*system.config.write().await = config;
|
|
system.init().await?;
|
|
info!("✅ System initialized with Webhook and MQTT targets.");
|
|
|
|
// --- 1. 查询当前活动的 Target ---
|
|
let active_targets = system.get_active_targets().await;
|
|
info!("\n---> Currently active targets: {:?}", active_targets);
|
|
assert_eq!(active_targets.len(), 2);
|
|
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
|
|
// --- 2. 精确删除一个 Target (例如 MQTT) ---
|
|
info!("\n---> Removing MQTT target...");
|
|
let mqtt_target_id = TargetID::new("1".to_string(), "mqtt".to_string());
|
|
system.remove_target(&mqtt_target_id, "notify_mqtt").await?;
|
|
info!("✅ MQTT target removed.");
|
|
|
|
// --- 3. 再次查询活动的 Target ---
|
|
let active_targets_after_removal = system.get_active_targets().await;
|
|
info!(
|
|
"\n---> Active targets after removal: {:?}",
|
|
active_targets_after_removal
|
|
);
|
|
assert_eq!(active_targets_after_removal.len(), 1);
|
|
assert_eq!(active_targets_after_removal[0].id, "1".to_string());
|
|
|
|
// --- 4. 发送事件进行验证 ---
|
|
// 配置一个规则，指向 Webhook 和已删除的 MQTT
|
|
let mut bucket_config = BucketNotificationConfig::new("us-east-1");
|
|
bucket_config.add_rule(
|
|
&[EventName::ObjectCreatedPut],
|
|
"*".to_string(),
|
|
TargetID::new("1".to_string(), "webhook".to_string()),
|
|
);
|
|
bucket_config.add_rule(
|
|
&[EventName::ObjectCreatedPut],
|
|
"*".to_string(),
|
|
TargetID::new("1".to_string(), "mqtt".to_string()), // 这个规则会匹配，但找不到 Target
|
|
);
|
|
system
|
|
.load_bucket_notification_config("my-bucket", &bucket_config)
|
|
.await?;
|
|
|
|
info!("\n---> Sending an event...");
|
|
let event = Event::new_test_event("my-bucket", "document.pdf", EventName::ObjectCreatedPut);
|
|
system
|
|
.send_event("my-bucket", "s3:ObjectCreated:Put", "document.pdf", event)
|
|
.await;
|
|
info!(
|
|
"✅ Event sent. Only the Webhook target should receive it. Check logs for warnings about the missing MQTT target."
|
|
);
|
|
|
|
tokio::time::sleep(Duration::from_secs(2)).await;
|
|
|
|
info!("\nDemo completed successfully");
|
|
Ok(())
|
|
}
|