mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-09 06:39:25 +00:00
houseme
86 lines
3.1 KiB
Rust
86 lines
3.1 KiB
Rust
// Copyright 2024 RustFS Team
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use crate::disk::error::{Error, Result};
|
|
use crate::erasure_coding::BitrotReader;
|
|
use crate::erasure_coding::BitrotWriterWrapper;
|
|
use crate::erasure_coding::decode::ParallelReader;
|
|
use crate::erasure_coding::encode::MultiWriter;
|
|
use bytes::Bytes;
|
|
use tokio::io::AsyncRead;
|
|
use tracing::info;
|
|
|
|
impl super::Erasure {
|
|
pub async fn heal<R>(
|
|
&self,
|
|
writers: &mut [Option<BitrotWriterWrapper>],
|
|
readers: Vec<Option<BitrotReader<R>>>,
|
|
total_length: usize,
|
|
_prefer: &[bool],
|
|
) -> Result<()>
|
|
where
|
|
R: AsyncRead + Unpin + Send + Sync,
|
|
{
|
|
info!(
|
|
"Erasure heal, writers len: {}, readers len: {}, total_length: {}",
|
|
writers.len(),
|
|
readers.len(),
|
|
total_length
|
|
);
|
|
if writers.len() != self.parity_shards + self.data_shards {
|
|
return Err(Error::other("invalid argument"));
|
|
}
|
|
let mut reader = ParallelReader::new(readers, self.clone(), 0, total_length);
|
|
|
|
let start_block = 0;
|
|
let mut end_block = total_length / self.block_size;
|
|
if !total_length.is_multiple_of(self.block_size) {
|
|
end_block += 1;
|
|
}
|
|
|
|
for _ in start_block..end_block {
|
|
let (mut shards, errs) = reader.read().await;
|
|
|
|
// Check if we have enough shards to reconstruct data
|
|
// We need at least data_shards available shards (data + parity combined)
|
|
let available_shards = errs.iter().filter(|e| e.is_none()).count();
|
|
if available_shards < self.data_shards {
|
|
return Err(Error::other(format!(
|
|
"can not reconstruct data: not enough available shards (need {}, have {}) {errs:?}",
|
|
self.data_shards, available_shards
|
|
)));
|
|
}
|
|
|
|
if self.parity_shards > 0 {
|
|
self.decode_data(&mut shards)?;
|
|
}
|
|
|
|
let shards = shards
|
|
.into_iter()
|
|
.map(|s| Bytes::from(s.unwrap_or_default()))
|
|
.collect::<Vec<_>>();
|
|
|
|
// Calculate proper write quorum for heal operation
|
|
// For heal, we only write to disks that need healing, so write quorum should be
|
|
// the number of available writers (disks that need healing)
|
|
let available_writers = writers.iter().filter(|w| w.is_some()).count();
|
|
let write_quorum = available_writers.max(1); // At least 1 writer must succeed
|
|
let mut writers = MultiWriter::new(writers, write_quorum);
|
|
writers.write(shards).await?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|