This commit is contained in:
houseme
2025-05-27 11:53:47 +08:00
parent 50e52206cc
commit a9c6bb8f0d
10 changed files with 327 additions and 293 deletions
+112 -12
View File
@@ -1,7 +1,8 @@
use crate::Event; use crate::config::{default_queue_limit, STORE_PREFIX};
use crate::KafkaConfig; use crate::KafkaConfig;
use crate::{ChannelAdapter, ChannelAdapterType}; use crate::{ChannelAdapter, ChannelAdapterType};
use crate::{Error, QueueStore}; use crate::{Error, QueueStore};
use crate::{Event, DEFAULT_RETRY_INTERVAL};
use async_trait::async_trait; use async_trait::async_trait;
use rdkafka::error::KafkaError; use rdkafka::error::KafkaError;
use rdkafka::producer::{FutureProducer, FutureRecord}; use rdkafka::producer::{FutureProducer, FutureRecord};
@@ -11,6 +12,7 @@ use std::path::PathBuf;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use tokio::time::sleep; use tokio::time::sleep;
use ChannelAdapterType::Kafka;
/// Kafka adapter for sending events to a Kafka topic. /// Kafka adapter for sending events to a Kafka topic.
pub struct KafkaAdapter { pub struct KafkaAdapter {
@@ -26,12 +28,24 @@ impl KafkaAdapter {
let producer = rdkafka::config::ClientConfig::new() let producer = rdkafka::config::ClientConfig::new()
.set("bootstrap.servers", &config.brokers) .set("bootstrap.servers", &config.brokers)
.set("message.timeout.ms", config.timeout.to_string()) .set("message.timeout.ms", config.timeout.to_string())
.create()?; .create()
.map_err(|e| Error::msg(format!("Failed to create a Kafka producer: {}", e)))?;
// create a queue store if enabled // create a queue store if enabled
let store = if !config.queue_dir.is_empty() { let store = if !config.common.queue_dir.is_empty() {
let store_path = PathBuf::from(&config.queue_dir); let store_path = PathBuf::from(&config.common.queue_dir).join(format!(
let store = QueueStore::new(store_path, config.queue_limit, Some(".kafka".to_string())); "{}-{}-{}",
STORE_PREFIX,
Kafka.as_str(),
config.common.identifier
));
let queue_limit = if config.queue_limit > 0 {
config.queue_limit
} else {
default_queue_limit()
};
let store = QueueStore::new(store_path, config.queue_limit, Some(".event".to_string()));
if let Err(e) = store.open() { if let Err(e) = store.open() {
tracing::error!("Unable to open queue storage: {}", e); tracing::error!("Unable to open queue storage: {}", e);
None None
@@ -44,19 +58,58 @@ impl KafkaAdapter {
Ok(Self { config, producer, store }) Ok(Self { config, producer, store })
} }
/// Handle backlog events in storage
pub async fn process_backlog(&self) -> Result<(), Error> {
if let Some(store) = &self.store {
let keys = store.list();
for key in keys {
match store.get_multiple(&key) {
Ok(events) => {
for event in events {
// Use the retry interval to send events
if let Err(e) = self.send_with_retry(&event).await {
tracing::error!("Processing of backlog events failed: {}", e);
// If it still fails, we remain in the queue
break;
}
}
// The event is deleted after it has been successfully processed
if let Err(e) = store.del(&key) {
tracing::error!("Failed to delete a handled event: {}", e);
}
}
Err(e) => {
tracing::error!("Fetch events from the queue failed: {}", e);
// If the event cannot be read, it may be corrupted, delete it
if let Err(del_err) = store.del(&key) {
tracing::error!("Failed to delete a corrupted event: {}", del_err);
}
}
}
}
}
Ok(())
}
/// Sends an event to the Kafka topic with retry logic. /// Sends an event to the Kafka topic with retry logic.
async fn send_with_retry(&self, event: &Event) -> Result<(), Error> { async fn send_with_retry(&self, event: &Event) -> Result<(), Error> {
let event_id = event.id.to_string(); let retry_interval = match self.config.retry_interval {
let payload = serde_json::to_string(&event)?; Some(t) => Duration::from_secs(t),
None => Duration::from_secs(DEFAULT_RETRY_INTERVAL), // Default to 3 seconds if not set
};
for attempt in 0..self.max_retries { for attempt in 0..self.max_retries {
let record = FutureRecord::to(&self.topic).key(&event_id).payload(&payload); match self.send_request(event).await {
match self.producer.send(record, Timeout::Never).await {
Ok(_) => return Ok(()), Ok(_) => return Ok(()),
Err((KafkaError::MessageProduction(RDKafkaErrorCode::QueueFull), _)) => { Err((KafkaError::MessageProduction(RDKafkaErrorCode::QueueFull), _)) => {
tracing::warn!("Kafka attempt {} failed: Queue full. Retrying...", attempt + 1); tracing::warn!("Kafka attempt {} failed: Queue full. Retrying...", attempt + 1);
sleep(Duration::from_secs(2u64.pow(attempt))).await; // sleep(Duration::from_secs(2u64.pow(attempt))).await;
sleep(retry_interval).await;
} }
Err((e, _)) => { Err((e, _)) => {
tracing::error!("Kafka send error: {}", e); tracing::error!("Kafka send error: {}", e);
@@ -67,6 +120,38 @@ impl KafkaAdapter {
Err(Error::Custom("Exceeded maximum retry attempts for Kafka message".to_string())) Err(Error::Custom("Exceeded maximum retry attempts for Kafka message".to_string()))
} }
/// Send a single Kafka message
async fn send_request(&self, event: &Event) -> Result<(), Error> {
// Serialize events
let payload = serde_json::to_string(event).map_err(|e| Error::Custom(format!("Serialization event failed: {}", e)))?;
// Create a Kafka record
let record = FutureRecord::to(&self.config.topic).payload(&payload).key(&event.id); // Use the event ID as the key
// Send to Kafka
let delivery_status = self
.producer
.send(record, Duration::from_millis(self.config.timeout))
.await
.map_err(|(e, _)| Error::Custom(format!("Failed to send to Kafka: {}", e)))?;
// Check delivery status
if let Some((err, _)) = delivery_status {
return Err(Error::Kafka(err));
}
Ok(())
}
/// Save the event to the queue
async fn save_to_queue(&self, event: &Event) -> Result<(), Error> {
if let Some(store) = &self.store {
store
.put(event.clone())
.map_err(|e| Error::Custom(format!("Saving events to queue failed: {}", e)))?;
}
Ok(())
}
} }
#[async_trait] #[async_trait]
@@ -76,6 +161,21 @@ impl ChannelAdapter for KafkaAdapter {
} }
async fn send(&self, event: &Event) -> Result<(), Error> { async fn send(&self, event: &Event) -> Result<(), Error> {
self.send_with_retry(event).await // Try to deal with the backlog of events first
let _ = self.process_backlog().await;
// An attempt was made to send the current event
match self.send_with_retry(event).await {
Ok(_) => Ok(()),
Err(e) => {
// If the send fails and the queue is enabled, save to the queue
if let Some(_) = &self.store {
tracing::warn!("Failed to send events to Kafka and saved to a queue: {}", e);
self.save_to_queue(event).await?;
return Ok(());
}
Err(e)
}
}
} }
} }
+70 -11
View File
@@ -1,20 +1,26 @@
use crate::config::STORE_PREFIX;
use crate::store::queue::Store; use crate::store::queue::Store;
use crate::WebhookConfig; use crate::WebhookConfig;
use crate::{ChannelAdapter, ChannelAdapterType}; use crate::{ChannelAdapter, ChannelAdapterType};
use crate::{Error, QueueStore}; use crate::{Error, QueueStore};
use crate::{Event, DEFAULT_RETRY_INTERVAL}; use crate::{Event, DEFAULT_RETRY_INTERVAL};
use async_trait::async_trait; use async_trait::async_trait;
use reqwest::header::{HeaderMap, HeaderName, HeaderValue};
use reqwest::{self, Client, Identity, RequestBuilder}; use reqwest::{self, Client, Identity, RequestBuilder};
use std::fs; use std::fs;
use std::path::PathBuf; use std::path::PathBuf;
use std::sync::Arc; use std::sync::Arc;
use std::time::Duration; use std::time::Duration;
use tokio::time::sleep; use tokio::time::sleep;
use ChannelAdapterType::Webhook;
/// Webhook adapter for sending events to a webhook endpoint. /// Webhook adapter for sending events to a webhook endpoint.
pub struct WebhookAdapter { pub struct WebhookAdapter {
/// Configuration information
config: WebhookConfig, config: WebhookConfig,
/// Event storage queues
store: Option<Arc<QueueStore<Event>>>, store: Option<Arc<QueueStore<Event>>>,
/// HTTP client
client: Client, client: Client,
} }
@@ -65,12 +71,25 @@ impl WebhookAdapter {
} else { } else {
builder.build() builder.build()
} }
.expect("Failed to create HTTP client"); .unwrap_or_else(|e| {
tracing::error!("Failed to create HTTP client: {}", e);
reqwest::Client::new()
});
// create a queue store if enabled // create a queue store if enabled
let store = if !config.common.queue_dir.len() > 0 { let store = if !config.common.queue_dir.len() > 0 {
let store_path = PathBuf::from(&config.common.queue_dir); let store_path = PathBuf::from(&config.common.queue_dir).join(format!(
let store = QueueStore::new(store_path, config.common.queue_limit, Some(".webhook".to_string())); "{}-{}-{}",
STORE_PREFIX,
Webhook.as_str(),
config.common.identifier
));
let queue_limit = if config.common.queue_limit > 0 {
config.common.queue_limit
} else {
crate::config::default_queue_limit()
};
let store = QueueStore::new(store_path, queue_limit, Some(".event".to_string()));
if let Err(e) = store.open() { if let Err(e) = store.open() {
tracing::error!("Unable to open queue storage: {}", e); tracing::error!("Unable to open queue storage: {}", e);
None None
@@ -94,16 +113,22 @@ impl WebhookAdapter {
for event in events { for event in events {
if let Err(e) = self.send_with_retry(&event).await { if let Err(e) = self.send_with_retry(&event).await {
tracing::error!("Processing of backlog events failed: {}", e); tracing::error!("Processing of backlog events failed: {}", e);
continue; // If it still fails, we remain in the queue
break;
} }
} }
// Deleted after successful processing // Deleted after successful processing
let _ = store.del(&key); if let Err(e) = store.del(&key) {
tracing::error!("Failed to delete a handled event: {}", e);
}
} }
Err(e) => { Err(e) => {
tracing::error!("Failed to read events from storage: {}", e); tracing::error!("Failed to read events from storage: {}", e);
// delete the broken entries // delete the broken entries
let _ = store.del(&key); // If the event cannot be read, it may be corrupted, delete it
if let Err(del_err) = store.del(&key) {
tracing::error!("Failed to delete a corrupted event: {}", del_err);
}
} }
} }
} }
@@ -146,11 +171,20 @@ impl WebhookAdapter {
/// Send a single HTTP request /// Send a single HTTP request
async fn send_request(&self, event: &Event) -> Result<(), Error> { async fn send_request(&self, event: &Event) -> Result<(), Error> {
// Send a request // Send a request
let response = self.build_request(event).send().await?; let response = self
.build_request(event)
.send()
.await
.map_err(|e| Error::Custom(format!("Sending a webhook request failed:{}", e)))?;
// Check the response status // Check the response status
if !response.status().is_success() { if !response.status().is_success() {
return Err(Error::Custom(format!("Webhook request failed, status code:{}", response.status()))); let status = response.status();
let body = response
.text()
.await
.unwrap_or_else(|_| "Unable to read response body".to_string());
return Err(Error::Custom(format!("Webhook request failed, status code:{},response:{}", status, body)));
} }
Ok(()) Ok(())
@@ -172,18 +206,32 @@ impl WebhookAdapter {
} }
} }
if let Some(headers) = &self.config.custom_headers { if let Some(headers) = &self.config.custom_headers {
let mut header_map = HeaderMap::new();
for (key, value) in headers { for (key, value) in headers {
request = request.header(key, value); if let (Ok(name), Ok(val)) = (HeaderName::from_bytes(key.as_bytes()), HeaderValue::from_str(value)) {
header_map.insert(name, val);
}
} }
request = request.headers(header_map);
} }
request request
} }
/// Save the event to the queue
async fn save_to_queue(&self, event: &Event) -> Result<(), Error> {
if let Some(store) = &self.store {
store
.put(event.clone())
.map_err(|e| Error::Custom(format!("Saving events to queue failed: {}", e)))?;
}
Ok(())
}
} }
#[async_trait] #[async_trait]
impl ChannelAdapter for WebhookAdapter { impl ChannelAdapter for WebhookAdapter {
fn name(&self) -> String { fn name(&self) -> String {
ChannelAdapterType::Webhook.to_string() Webhook.to_string()
} }
async fn send(&self, event: &Event) -> Result<(), Error> { async fn send(&self, event: &Event) -> Result<(), Error> {
@@ -191,6 +239,17 @@ impl ChannelAdapter for WebhookAdapter {
let _ = self.process_backlog().await; let _ = self.process_backlog().await;
// Send the current event // Send the current event
self.send_with_retry(event).await match self.send_with_retry(event).await {
Ok(_) => Ok(()),
Err(e) => {
// If the send fails and the queue is enabled, save to the queue
if let Some(_) = &self.store {
tracing::warn!("Failed to send the event and saved to the queue: {}", e);
self.save_to_queue(event).await?;
return Ok(());
}
Err(e)
}
}
} }
} }
+3 -3
View File
@@ -9,7 +9,7 @@ use tracing::info;
const DEFAULT_CONFIG_FILE: &str = "event"; const DEFAULT_CONFIG_FILE: &str = "event";
/// The prefix for the configuration file /// The prefix for the configuration file
pub(crate) const STORE_PREFIX: &str = "rustfs"; pub const STORE_PREFIX: &str = "rustfs";
/// The default retry interval for the webhook adapter /// The default retry interval for the webhook adapter
pub const DEFAULT_RETRY_INTERVAL: u64 = 3; pub const DEFAULT_RETRY_INTERVAL: u64 = 3;
@@ -371,7 +371,7 @@ fn default_queue_dir() -> String {
} }
/// Provides the recommended default channel capacity for high concurrency systems /// Provides the recommended default channel capacity for high concurrency systems
fn default_queue_limit() -> u64 { pub(crate) fn default_queue_limit() -> u64 {
env::var("EVENT_CHANNEL_CAPACITY") env::var("EVENT_CHANNEL_CAPACITY")
.unwrap_or_else(|_| "10000".to_string()) .unwrap_or_else(|_| "10000".to_string())
.parse() .parse()
@@ -404,7 +404,7 @@ impl EventNotifierConfig {
// The existing implementation remains the same, but returns EventNotifierConfig // The existing implementation remains the same, but returns EventNotifierConfig
// ... // ...
EventNotifierConfig::default() Self::default()
} }
/// Deserialization configuration /// Deserialization configuration
-5
View File
@@ -33,9 +33,4 @@ pub use event::{Bucket, Event, EventBuilder, Identity, Log, Metadata, Name, Obje
pub use global::{initialize, is_initialized, is_ready, send_event, shutdown}; pub use global::{initialize, is_initialized, is_ready, send_event, shutdown};
pub use notifier::NotifierSystem; pub use notifier::NotifierSystem;
pub use store::event::EventStore; pub use store::event::EventStore;
pub use store::get_event_notifier_config;
pub use store::queue::QueueStore; pub use store::queue::QueueStore;
pub use store::EventSys;
pub use store::GLOBAL_EventSys;
+59 -57
View File
@@ -1,5 +1,4 @@
use crate::store::{CONFIG_FILE, EVENT}; use crate::{adapter, ChannelAdapter, EventNotifierConfig};
use crate::{adapter, ChannelAdapter, EventNotifierConfig, WebhookAdapter};
use common::error::{Error, Result}; use common::error::{Error, Result};
use ecstore::config::com::{read_config, save_config, CONFIG_PREFIX}; use ecstore::config::com::{read_config, save_config, CONFIG_PREFIX};
use ecstore::disk::RUSTFS_META_BUCKET; use ecstore::disk::RUSTFS_META_BUCKET;
@@ -12,6 +11,12 @@ use std::sync::Arc;
use tokio::sync::Mutex; use tokio::sync::Mutex;
use tracing::instrument; use tracing::instrument;
/// * config file
const CONFIG_FILE: &str = "event.json";
/// event sys config
const EVENT: &str = "event";
/// Global storage API access point /// Global storage API access point
pub static GLOBAL_STORE_API: Lazy<Mutex<Option<Arc<ECStore>>>> = Lazy::new(|| Mutex::new(None)); pub static GLOBAL_STORE_API: Lazy<Mutex<Option<Arc<ECStore>>>> = Lazy::new(|| Mutex::new(None));
@@ -44,8 +49,11 @@ impl EventManager {
pub async fn init(&self) -> Result<EventNotifierConfig> { pub async fn init(&self) -> Result<EventNotifierConfig> {
tracing::info!("Event system configuration initialization begins"); tracing::info!("Event system configuration initialization begins");
let cfg = match read_event_config(self.api.clone()).await { let cfg = match read_config_without_migrate(self.api.clone()).await {
Ok(cfg) => cfg, Ok(cfg) => {
tracing::info!("The event system configuration was successfully read");
cfg
}
Err(err) => { Err(err) => {
tracing::error!("Failed to initialize the event system configuration:{:?}", err); tracing::error!("Failed to initialize the event system configuration:{:?}", err);
return Err(err); return Err(err);
@@ -53,6 +61,7 @@ impl EventManager {
}; };
*GLOBAL_EVENT_CONFIG.lock().await = Some(cfg.clone()); *GLOBAL_EVENT_CONFIG.lock().await = Some(cfg.clone());
tracing::info!("The initialization of the event system configuration is complete"); tracing::info!("The initialization of the event system configuration is complete");
Ok(cfg) Ok(cfg)
@@ -152,68 +161,29 @@ impl EventManager {
None => return Err(Error::msg("The global configuration is not initialized")), None => return Err(Error::msg("The global configuration is not initialized")),
}; };
let mut adapters: Vec<Arc<dyn ChannelAdapter>> = Vec::new(); let adapter_configs = config.to_adapter_configs();
match adapter::create_adapters(&adapter_configs) {
// Create a webhook adapter Ok(adapters) => Ok(adapters),
for (_, webhook_config) in &config.webhook { Err(err) => {
if webhook_config.common.enable { tracing::error!("Failed to create adapters: {:?}", err);
#[cfg(feature = "webhook")] Err(Error::from(err))
{
adapters.push(Arc::new(WebhookAdapter::new(webhook_config.clone())));
}
#[cfg(not(feature = "webhook"))]
{
return Err(Error::msg("The webhook feature is not enabled"));
}
} }
} }
// Create a Kafka adapter
for (_, kafka_config) in &config.kafka {
if kafka_config.common.enable {
#[cfg(all(feature = "kafka", target_os = "linux"))]
{
match KafkaAdapter::new(kafka_config.clone()) {
Ok(adapter) => adapters.push(Arc::new(adapter)),
Err(e) => tracing::error!("Failed to create a Kafka adapter:{}", e),
}
}
#[cfg(any(not(feature = "kafka"), not(target_os = "linux")))]
{
return Err(Error::msg("Kafka functionality is not enabled or is not a Linux environment"));
}
}
}
// Create an MQTT adapter
for (_, mqtt_config) in &config.mqtt {
if mqtt_config.common.enable {
#[cfg(feature = "mqtt")]
{
// Implement MQTT adapter creation logic
// ...
}
#[cfg(not(feature = "mqtt"))]
{
return Err(Error::msg("MQTT The feature is not enabled"));
}
}
}
Ok(adapters)
} }
} }
/// Get the Global Storage API
pub async fn get_global_store_api() -> Option<Arc<ECStore>> {
GLOBAL_STORE_API.lock().await.clone()
}
/// Get the Global Storage API /// Get the Global Storage API
pub async fn get_global_event_config() -> Option<EventNotifierConfig> { pub async fn get_global_event_config() -> Option<EventNotifierConfig> {
GLOBAL_EVENT_CONFIG.lock().await.clone() GLOBAL_EVENT_CONFIG.lock().await.clone()
} }
/// Read event configuration /// Read event configuration
async fn read_event_config(api: Arc<ECStore>) -> Result<EventNotifierConfig> { async fn read_event_config<S: StorageAPI>(api: Arc<S>) -> Result<EventNotifierConfig> {
let config_file = get_event_config_file(); let config_file = get_event_config_file();
let data = read_config(api, &config_file).await?; let data = read_config(api, &config_file).await?;
@@ -221,7 +191,7 @@ async fn read_event_config(api: Arc<ECStore>) -> Result<EventNotifierConfig> {
} }
/// Save the event configuration /// Save the event configuration
async fn save_event_config(api: Arc<ECStore>, config: &EventNotifierConfig) -> Result<()> { async fn save_event_config<S: StorageAPI>(api: Arc<S>, config: &EventNotifierConfig) -> Result<()> {
let config_file = get_event_config_file(); let config_file = get_event_config_file();
let data = config.marshal()?; let data = config.marshal()?;
@@ -231,6 +201,38 @@ async fn save_event_config(api: Arc<ECStore>, config: &EventNotifierConfig) -> R
/// Get the event profile path /// Get the event profile path
fn get_event_config_file() -> String { fn get_event_config_file() -> String {
// "event/config.json".to_string()
format!("{}{}{}{}{}", CONFIG_PREFIX, SLASH_SEPARATOR, EVENT, SLASH_SEPARATOR, CONFIG_FILE) format!("{}{}{}{}{}", CONFIG_PREFIX, SLASH_SEPARATOR, EVENT, SLASH_SEPARATOR, CONFIG_FILE)
} }
/// Read the configuration file and create a default configuration if it doesn't exist
pub async fn read_config_without_migrate<S: StorageAPI>(api: Arc<S>) -> Result<EventNotifierConfig> {
let config_file = get_event_config_file();
let data = match read_config(api.clone(), &config_file).await {
Ok(data) => {
if data.is_empty() {
return new_and_save_event_config(api).await;
}
data
}
Err(err) if ecstore::config::error::is_err_config_not_found(&err) => {
tracing::warn!("If the configuration file does not exist, start initializing the default configuration");
return new_and_save_event_config(api).await;
}
Err(err) => {
tracing::error!("Read configuration file error: {:?}", err);
return Err(err);
}
};
// Parse configuration
let cfg = EventNotifierConfig::unmarshal(&data)?;
Ok(cfg)
}
/// Create and save a new configuration
async fn new_and_save_event_config<S: StorageAPI>(api: Arc<S>) -> Result<EventNotifierConfig> {
let cfg = EventNotifierConfig::default();
save_event_config(api, &cfg).await?;
Ok(cfg)
}
-118
View File
@@ -1,121 +1,3 @@
pub(crate) mod event; pub(crate) mod event;
mod manager; mod manager;
pub(crate) mod queue; pub(crate) mod queue;
use crate::NotifierConfig;
use common::error::Result;
use ecstore::config::com::{read_config, save_config, CONFIG_PREFIX};
use ecstore::config::error::is_err_config_not_found;
use ecstore::store::ECStore;
use ecstore::utils::path::SLASH_SEPARATOR;
use ecstore::StorageAPI;
use lazy_static::lazy_static;
use std::sync::{Arc, OnceLock};
use tracing::{error, instrument, warn};
lazy_static! {
pub static ref GLOBAL_EventSys: EventSys = EventSys::new();
pub static ref GLOBAL_EventSysConfig: OnceLock<NotifierConfig> = OnceLock::new();
}
/// * config file
const CONFIG_FILE: &str = "event.json";
/// event sys config
const EVENT: &str = "event";
#[derive(Debug)]
pub struct EventSys {}
impl Default for EventSys {
fn default() -> Self {
Self::new()
}
}
impl EventSys {
pub fn new() -> Self {
Self {}
}
#[instrument(skip_all)]
pub async fn init(&self, api: Arc<ECStore>) -> Result<()> {
tracing::info!("event sys config init start");
let cfg = read_config_without_migrate(api.clone().clone()).await?;
let _ = GLOBAL_EventSysConfig.set(cfg);
tracing::info!("event sys config init done");
Ok(())
}
}
/// get event sys config file
///
/// # Returns
/// NotifierConfig
pub fn get_event_notifier_config() -> &'static NotifierConfig {
GLOBAL_EventSysConfig.get_or_init(NotifierConfig::default)
}
fn get_event_sys_file() -> String {
format!("{}{}{}{}{}", CONFIG_PREFIX, SLASH_SEPARATOR, EVENT, SLASH_SEPARATOR, CONFIG_FILE)
}
/// new server config
///
/// # Returns
/// NotifierConfig
fn new_server_config() -> NotifierConfig {
NotifierConfig::new()
}
async fn new_and_save_server_config<S: StorageAPI>(api: Arc<S>) -> Result<NotifierConfig> {
let cfg = new_server_config();
save_server_config(api, &cfg).await?;
Ok(cfg)
}
/// save server config
///
/// # Parameters
/// - `api`: StorageAPI
/// - `cfg`: configuration to save
///
/// # Returns
/// Result
async fn save_server_config<S: StorageAPI>(api: Arc<S>, cfg: &NotifierConfig) -> Result<()> {
let data = cfg.marshal()?;
let config_file = get_event_sys_file();
save_config(api, &config_file, data).await
}
/// read config without migrate
///
/// # Parameters
/// - `api`: StorageAPI
///
/// # Returns
/// Configuration information
pub async fn read_config_without_migrate<S: StorageAPI>(api: Arc<S>) -> Result<NotifierConfig> {
let config_file = get_event_sys_file();
let data = match read_config(api.clone(), config_file.as_str()).await {
Ok(data) => {
if data.is_empty() {
return new_and_save_server_config(api).await;
}
data
}
Err(err) if is_err_config_not_found(&err) => {
warn!("config not found, start to init");
return new_and_save_server_config(api).await;
}
Err(err) => {
error!("read config error: {:?}", err);
return Err(err);
}
};
// TODO: decrypt if needed
let cfg = NotifierConfig::unmarshal(data.as_slice())?;
Ok(cfg.merge())
}
+13 -7
View File
@@ -192,7 +192,7 @@ where
/// Write the object to a file /// Write the object to a file
fn write_object(&self, key: &Key, item: &T) -> Result<()> { fn write_object(&self, key: &Key, item: &T) -> Result<()> {
// 序列化对象 // Serialize the object
let data = serde_json::to_vec(item)?; let data = serde_json::to_vec(item)?;
self.write_bytes(key, &data) self.write_bytes(key, &data)
} }
@@ -213,7 +213,7 @@ where
let path = self.directory.join(key.to_string()); let path = self.directory.join(key.to_string());
let file_data = if key.compress { let file_data = if key.compress {
// 使用 snap 压缩数据 // Use snap to compress data
let mut encoder = Encoder::new(); let mut encoder = Encoder::new();
encoder encoder
.compress_vec(data) .compress_vec(data)
@@ -271,7 +271,10 @@ where
// Read existing files // Read existing files
let files = self.list_files()?; let files = self.list_files()?;
let mut entries = self.entries.write().map_err(|_| Error::msg("获取写锁失败"))?; let mut entries = self
.entries
.write()
.map_err(|_| Error::msg("Failed to obtain a write lock"))?;
for file in files { for file in files {
if let Ok(meta) = file.metadata() { if let Ok(meta) = file.metadata() {
@@ -293,7 +296,7 @@ where
fn put(&self, item: T) -> Result<Key> { fn put(&self, item: T) -> Result<Key> {
{ {
let entries = self.entries.read().map_err(|_| Error::msg("获取读锁失败"))?; let entries = self.entries.read().map_err(|_| Error::msg("Failed to obtain a read lock"))?;
if entries.len() as u64 >= self.entry_limit { if entries.len() as u64 >= self.entry_limit {
return Err(Error::msg("The storage limit has been reached")); return Err(Error::msg("The storage limit has been reached"));
} }
@@ -314,7 +317,7 @@ where
} }
{ {
let entries = self.entries.read().map_err(|_| Error::msg("获取读锁失败"))?; let entries = self.entries.read().map_err(|_| Error::msg("Failed to obtain a read lock"))?;
if entries.len() as u64 >= self.entry_limit { if entries.len() as u64 >= self.entry_limit {
return Err(Error::msg("The storage limit has been reached")); return Err(Error::msg("The storage limit has been reached"));
} }
@@ -397,7 +400,7 @@ where
fn put_raw(&self, data: &[u8]) -> Result<Key> { fn put_raw(&self, data: &[u8]) -> Result<Key> {
{ {
let entries = self.entries.read().map_err(|_| Error::msg("获取读锁失败"))?; let entries = self.entries.read().map_err(|_| Error::msg("Failed to obtain a read lock"))?;
if entries.len() as u64 >= self.entry_limit { if entries.len() as u64 >= self.entry_limit {
return Err(Error::msg("the storage limit has been reached")); return Err(Error::msg("the storage limit has been reached"));
} }
@@ -441,7 +444,10 @@ where
} }
// Remove the entry from the map // Remove the entry from the map
let mut entries = self.entries.write().map_err(|_| Error::msg("获取写锁失败"))?; let mut entries = self
.entries
.write()
.map_err(|_| Error::msg("Failed to obtain a write lock"))?;
entries.remove(&key.to_string()); entries.remove(&key.to_string());
Ok(()) Ok(())
+54 -62
View File
@@ -2520,7 +2520,7 @@ mod test {
fi3.mod_time = Some(time3); fi3.mod_time = Some(time3);
fm.add_version(fi3).unwrap(); fm.add_version(fi3).unwrap();
// Sort first to ensure latest is at the front // Sort first to ensure latest is at the front
fm.sort_by_mod_time(); fm.sort_by_mod_time();
// Should return the first version's mod time (lastest_mod_time returns first version's time) // Should return the first version's mod time (lastest_mod_time returns first version's time)
@@ -2690,7 +2690,7 @@ mod test {
assert!(result.is_err()); assert!(result.is_err());
} }
#[test] #[test]
fn test_is_latest_delete_marker() { fn test_is_latest_delete_marker() {
// Test the is_latest_delete_marker function with simple data // Test the is_latest_delete_marker function with simple data
// Since the function is complex and requires specific XL format, // Since the function is complex and requires specific XL format,
@@ -2798,9 +2798,7 @@ async fn test_file_info_from_raw() {
let encoded = fm.marshal_msg().unwrap(); let encoded = fm.marshal_msg().unwrap();
let raw_info = RawFileInfo { let raw_info = RawFileInfo { buf: encoded };
buf: encoded,
};
let result = file_info_from_raw(raw_info, "test-bucket", "test-object", false).await; let result = file_info_from_raw(raw_info, "test-bucket", "test-object", false).await;
assert!(result.is_ok()); assert!(result.is_ok());
@@ -2833,26 +2831,26 @@ fn test_file_meta_load_function() {
assert!(result.is_err()); assert!(result.is_err());
} }
#[test] #[test]
fn test_file_meta_read_bytes_header() { fn test_file_meta_read_bytes_header() {
// Test read_bytes_header function - it expects the first 5 bytes to be msgpack bin length // Test read_bytes_header function - it expects the first 5 bytes to be msgpack bin length
// Create a buffer with proper msgpack bin format for a 9-byte binary // Create a buffer with proper msgpack bin format for a 9-byte binary
let mut buf = vec![0xc4, 0x09]; // msgpack bin8 format for 9 bytes let mut buf = vec![0xc4, 0x09]; // msgpack bin8 format for 9 bytes
buf.extend_from_slice(b"test data"); // 9 bytes of data buf.extend_from_slice(b"test data"); // 9 bytes of data
buf.extend_from_slice(b"extra"); // additional data buf.extend_from_slice(b"extra"); // additional data
let result = FileMeta::read_bytes_header(&buf); let result = FileMeta::read_bytes_header(&buf);
assert!(result.is_ok()); assert!(result.is_ok());
let (length, remaining) = result.unwrap(); let (length, remaining) = result.unwrap();
assert_eq!(length, 9); // "test data" length assert_eq!(length, 9); // "test data" length
// remaining should be everything after the 5-byte header (but we only have 2-byte header) // remaining should be everything after the 5-byte header (but we only have 2-byte header)
assert_eq!(remaining.len(), buf.len() - 5); assert_eq!(remaining.len(), buf.len() - 5);
// Test with buffer too small // Test with buffer too small
let small_buf = vec![0u8; 2]; let small_buf = vec![0u8; 2];
let result = FileMeta::read_bytes_header(&small_buf); let result = FileMeta::read_bytes_header(&small_buf);
assert!(result.is_err()); assert!(result.is_err());
} }
#[test] #[test]
fn test_file_meta_get_set_idx() { fn test_file_meta_get_set_idx() {
@@ -3080,11 +3078,11 @@ fn test_file_meta_version_header_ordering() {
// Test partial_cmp // Test partial_cmp
assert!(header1.partial_cmp(&header2).is_some()); assert!(header1.partial_cmp(&header2).is_some());
// Test cmp - header2 should be greater (newer) // Test cmp - header2 should be greater (newer)
use std::cmp::Ordering; use std::cmp::Ordering;
assert_eq!(header1.cmp(&header2), Ordering::Less); // header1 has earlier time assert_eq!(header1.cmp(&header2), Ordering::Less); // header1 has earlier time
assert_eq!(header2.cmp(&header1), Ordering::Greater); // header2 has later time assert_eq!(header2.cmp(&header1), Ordering::Greater); // header2 has later time
assert_eq!(header1.cmp(&header1), Ordering::Equal); assert_eq!(header1.cmp(&header1), Ordering::Equal);
} }
#[test] #[test]
@@ -3110,10 +3108,7 @@ fn test_merge_file_meta_versions_edge_cases() {
version2.header.version_id = Some(Uuid::new_v4()); version2.header.version_id = Some(Uuid::new_v4());
version2.header.mod_time = Some(OffsetDateTime::from_unix_timestamp(2000).unwrap()); version2.header.mod_time = Some(OffsetDateTime::from_unix_timestamp(2000).unwrap());
let versions = vec![ let versions = vec![vec![version1.clone()], vec![version2.clone()]];
vec![version1.clone()],
vec![version2.clone()],
];
let _merged_strict = merge_file_meta_versions(1, true, 10, &versions); let _merged_strict = merge_file_meta_versions(1, true, 10, &versions);
let merged_non_strict = merge_file_meta_versions(1, false, 10, &versions); let merged_non_strict = merge_file_meta_versions(1, false, 10, &versions);
@@ -3191,9 +3186,7 @@ async fn test_get_file_info_edge_cases() {
#[tokio::test] #[tokio::test]
async fn test_file_info_from_raw_edge_cases() { async fn test_file_info_from_raw_edge_cases() {
// Test with empty buffer // Test with empty buffer
let empty_raw = RawFileInfo { let empty_raw = RawFileInfo { buf: vec![] };
buf: vec![],
};
let result = file_info_from_raw(empty_raw, "bucket", "object", false).await; let result = file_info_from_raw(empty_raw, "bucket", "object", false).await;
assert!(result.is_err()); assert!(result.is_err());
@@ -3227,12 +3220,12 @@ fn test_meta_object_edge_cases() {
obj.data_dir = None; obj.data_dir = None;
assert!(obj.use_data_dir()); assert!(obj.use_data_dir());
// Test use_inlinedata (always returns false in current implementation) // Test use_inlinedata (always returns false in current implementation)
obj.size = 128 * 1024; // 128KB threshold obj.size = 128 * 1024; // 128KB threshold
assert!(!obj.use_inlinedata()); // Should be false assert!(!obj.use_inlinedata()); // Should be false
obj.size = 128 * 1024 - 1; obj.size = 128 * 1024 - 1;
assert!(!obj.use_inlinedata()); // Should also be false (always false) assert!(!obj.use_inlinedata()); // Should also be false (always false)
} }
#[test] #[test]
@@ -3244,17 +3237,17 @@ fn test_file_meta_version_header_edge_cases() {
header.ec_m = 0; header.ec_m = 0;
assert!(!header.has_ec()); assert!(!header.has_ec());
// Test matches_not_strict with different signatures but same version_id // Test matches_not_strict with different signatures but same version_id
let mut other = FileMetaVersionHeader::default(); let mut other = FileMetaVersionHeader::default();
let version_id = Some(Uuid::new_v4()); let version_id = Some(Uuid::new_v4());
header.version_id = version_id; header.version_id = version_id;
other.version_id = version_id; other.version_id = version_id;
header.version_type = VersionType::Object; header.version_type = VersionType::Object;
other.version_type = VersionType::Object; other.version_type = VersionType::Object;
header.signature = [1, 2, 3, 4]; header.signature = [1, 2, 3, 4];
other.signature = [5, 6, 7, 8]; other.signature = [5, 6, 7, 8];
// Should match because they have same version_id and type // Should match because they have same version_id and type
assert!(header.matches_not_strict(&other)); assert!(header.matches_not_strict(&other));
// Test sorts_before with same mod_time but different version_id // Test sorts_before with same mod_time but different version_id
let time = OffsetDateTime::from_unix_timestamp(1000).unwrap(); let time = OffsetDateTime::from_unix_timestamp(1000).unwrap();
@@ -3286,12 +3279,12 @@ fn test_file_meta_add_version_edge_cases() {
fi2.mod_time = Some(OffsetDateTime::now_utc()); fi2.mod_time = Some(OffsetDateTime::now_utc());
fm.add_version(fi2).unwrap(); fm.add_version(fi2).unwrap();
// Should still have only one version, but updated // Should still have only one version, but updated
assert_eq!(fm.versions.len(), 1); assert_eq!(fm.versions.len(), 1);
let (_, version) = fm.find_version(version_id).unwrap(); let (_, version) = fm.find_version(version_id).unwrap();
if let Some(obj) = version.object { if let Some(obj) = version.object {
assert_eq!(obj.size, 2048); // Size gets updated when adding same version_id assert_eq!(obj.size, 2048); // Size gets updated when adding same version_id
} }
} }
#[test] #[test]
@@ -3324,12 +3317,11 @@ fn test_file_meta_shard_data_dir_count_edge_cases() {
fi.mod_time = Some(OffsetDateTime::now_utc()); fi.mod_time = Some(OffsetDateTime::now_utc());
fm.add_version(fi).unwrap(); fm.add_version(fi).unwrap();
let count = fm.shard_data_dir_count(&version_id, &data_dir); let count = fm.shard_data_dir_count(&version_id, &data_dir);
assert_eq!(count, 0); // Should be 0 because user_data_dir() requires flag assert_eq!(count, 0); // Should be 0 because user_data_dir() requires flag
// Test with different version_id // Test with different version_id
let other_version_id = Some(Uuid::new_v4()); let other_version_id = Some(Uuid::new_v4());
let count = fm.shard_data_dir_count(&other_version_id, &data_dir); let count = fm.shard_data_dir_count(&other_version_id, &data_dir);
assert_eq!(count, 1); // Should be 1 because the version has matching data_dir and user_data_dir() is true assert_eq!(count, 1); // Should be 1 because the version has matching data_dir and user_data_dir() is true
} }
+4 -8
View File
@@ -5890,7 +5890,7 @@ mod tests {
erasure: ErasureInfo { erasure: ErasureInfo {
data_blocks: 4, data_blocks: 4,
parity_blocks: 2, parity_blocks: 2,
index: 1, // Must be > 0 for is_valid() to return true index: 1, // Must be > 0 for is_valid() to return true
distribution: vec![1, 2, 3, 4, 5, 6], // Must match data_blocks + parity_blocks distribution: vec![1, 2, 3, 4, 5, 6], // Must match data_blocks + parity_blocks
..Default::default() ..Default::default()
}, },
@@ -5902,7 +5902,7 @@ mod tests {
erasure: ErasureInfo { erasure: ErasureInfo {
data_blocks: 6, data_blocks: 6,
parity_blocks: 3, parity_blocks: 3,
index: 1, // Must be > 0 for is_valid() to return true index: 1, // Must be > 0 for is_valid() to return true
distribution: vec![1, 2, 3, 4, 5, 6, 7, 8, 9], // Must match data_blocks + parity_blocks distribution: vec![1, 2, 3, 4, 5, 6, 7, 8, 9], // Must match data_blocks + parity_blocks
..Default::default() ..Default::default()
}, },
@@ -5914,7 +5914,7 @@ mod tests {
erasure: ErasureInfo { erasure: ErasureInfo {
data_blocks: 2, data_blocks: 2,
parity_blocks: 1, parity_blocks: 1,
index: 1, // Must be > 0 for is_valid() to return true index: 1, // Must be > 0 for is_valid() to return true
distribution: vec![1, 2, 3], // Must match data_blocks + parity_blocks distribution: vec![1, 2, 3], // Must match data_blocks + parity_blocks
..Default::default() ..Default::default()
}, },
@@ -6019,11 +6019,7 @@ mod tests {
#[test] #[test]
fn test_join_errs() { fn test_join_errs() {
// Test joining error messages // Test joining error messages
let errs = vec![ let errs = vec![None, Some(Error::from_string("error1")), Some(Error::from_string("error2"))];
None,
Some(Error::from_string("error1")),
Some(Error::from_string("error2")),
];
let joined = join_errs(&errs); let joined = join_errs(&errs);
assert!(joined.contains("<nil>")); assert!(joined.contains("<nil>"));
assert!(joined.contains("error1")); assert!(joined.contains("error1"));
+12 -10
View File
@@ -1153,7 +1153,11 @@ mod tests {
assert_eq!(file_info.get_etag(), None); assert_eq!(file_info.get_etag(), None);
// With etag // With etag
file_info.metadata.as_mut().unwrap().insert("etag".to_string(), "test-etag".to_string()); file_info
.metadata
.as_mut()
.unwrap()
.insert("etag".to_string(), "test-etag".to_string());
assert_eq!(file_info.get_etag(), Some("test-etag".to_string())); assert_eq!(file_info.get_etag(), Some("test-etag".to_string()));
} }
@@ -1282,10 +1286,7 @@ mod tests {
file_info.set_healing(); file_info.set_healing();
assert!(file_info.metadata.is_some()); assert!(file_info.metadata.is_some());
assert_eq!( assert_eq!(file_info.metadata.as_ref().unwrap().get(RUSTFS_HEALING), Some(&"true".to_string()));
file_info.metadata.as_ref().unwrap().get(RUSTFS_HEALING),
Some(&"true".to_string())
);
} }
#[test] #[test]
@@ -1656,10 +1657,11 @@ mod tests {
assert!(!object_info.is_compressed()); assert!(!object_info.is_compressed());
// With compression metadata // With compression metadata
object_info.user_defined.as_mut().unwrap().insert( object_info
format!("{}compression", RESERVED_METADATA_PREFIX), .user_defined
"gzip".to_string() .as_mut()
); .unwrap()
.insert(format!("{}compression", RESERVED_METADATA_PREFIX), "gzip".to_string());
assert!(object_info.is_compressed()); assert!(object_info.is_compressed());
} }
@@ -1866,7 +1868,7 @@ mod tests {
let shard_size = erasure.shard_size(1000); let shard_size = erasure.shard_size(1000);
assert_eq!(shard_size, 1000); // 1000 / 1 = 1000 assert_eq!(shard_size, 1000); // 1000 / 1 = 1000
// Test with zero block size - this will cause division by zero in shard_size // Test with zero block size - this will cause division by zero in shard_size
// So we need to test with non-zero block_size but zero data_blocks was already fixed above // So we need to test with non-zero block_size but zero data_blocks was already fixed above
let erasure = ErasureInfo { let erasure = ErasureInfo {
data_blocks: 4, data_blocks: 4,