mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-18 18:46:17 +00:00
houseme
99 lines
3.7 KiB
Rust
99 lines
3.7 KiB
Rust
// Using Global Accessories
|
|
use rustfs_config::notify;
|
|
use rustfs_notify::arn::TargetID;
|
|
use rustfs_notify::global::notification_system;
|
|
use rustfs_notify::{init_logger, BucketNotificationConfig, Event, EventName, LogLevel, NotificationError, KVS};
|
|
use std::time::Duration;
|
|
use tracing::info;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), NotificationError> {
|
|
init_logger(LogLevel::Debug);
|
|
|
|
// Get global NotificationSystem instance
|
|
let system = notification_system();
|
|
|
|
// --- Initial configuration ---
|
|
let mut config = rustfs_notify::Config::new();
|
|
let current_root = rustfs_utils::dirs::get_project_root().expect("failed to get project root");
|
|
// Webhook target
|
|
let mut webhook_kvs = KVS::new();
|
|
webhook_kvs.set("enable", "on");
|
|
webhook_kvs.set("endpoint", "http://127.0.0.1:3020/webhook");
|
|
// webhook_kvs.set("queue_dir", "./logs/webhook");
|
|
webhook_kvs.set(
|
|
"queue_dir",
|
|
current_root
|
|
.clone()
|
|
.join("/deploy/logs/notify/webhook")
|
|
.to_str()
|
|
.unwrap()
|
|
.to_string(),
|
|
);
|
|
let mut webhook_targets = std::collections::HashMap::new();
|
|
webhook_targets.insert("1".to_string(), webhook_kvs);
|
|
config.insert("notify_webhook".to_string(), webhook_targets);
|
|
|
|
// Load the initial configuration and initialize the system
|
|
*system.config.write().await = config;
|
|
system.init().await?;
|
|
info!("✅ System initialized with Webhook target.");
|
|
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
|
|
// --- Dynamically update system configuration: Add an MQTT Target ---
|
|
info!("\n---> Dynamically adding MQTT target...");
|
|
let mut mqtt_kvs = 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");
|
|
mqtt_kvs.set("username", "test");
|
|
mqtt_kvs.set("password", "123456");
|
|
mqtt_kvs.set("queue_limit", "10000");
|
|
// mqtt_kvs.set("queue_dir", "./logs/mqtt");
|
|
mqtt_kvs.set("queue_dir", current_root.join("/deploy/logs/notify/mqtt").to_str().unwrap().to_string());
|
|
system.set_target_config("notify_mqtt", "1", mqtt_kvs).await?;
|
|
info!("✅ MQTT target added and system reloaded.");
|
|
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
|
|
// --- Loading and managing Bucket configurations ---
|
|
info!("\n---> Loading bucket notification config...");
|
|
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()),
|
|
);
|
|
system.load_bucket_notification_config("my-bucket", &bucket_config).await?;
|
|
info!("✅ Bucket 'my-bucket' config loaded.");
|
|
|
|
// --- Send events ---
|
|
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. Both Webhook and MQTT targets should receive it.");
|
|
|
|
tokio::time::sleep(Duration::from_secs(2)).await;
|
|
|
|
// --- Dynamically remove configuration ---
|
|
info!("\n---> Dynamically removing Webhook target...");
|
|
system.remove_target_config("notify_webhook", "1").await?;
|
|
info!("✅ Webhook target removed and system reloaded.");
|
|
|
|
info!("\n---> Removing bucket notification config...");
|
|
system.remove_bucket_notification_config("my-bucket").await;
|
|
info!("✅ Bucket 'my-bucket' config removed.");
|
|
|
|
info!("\nDemo completed successfully");
|
|
Ok(())
|
|
}
|