mirror of
https://github.com/minio/minio.git
synced 2024-12-24 22:25:54 -05:00
e84f533c6c
wait groups are necessary with io.Pipes() to avoid races when a blocking function may not be expected and a Write() -> Close() before Read() races on each other. We should avoid such situations.. Co-authored-by: Klaus Post <klauspost@gmail.com>
60 lines
2.0 KiB
Go
60 lines
2.0 KiB
Go
// Copyright (c) 2015-2021 MinIO, Inc.
|
|
//
|
|
// This file is part of MinIO Object Storage stack
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/minio/minio/cmd/logger"
|
|
"github.com/minio/minio/pkg/bpool"
|
|
xioutil "github.com/minio/minio/pkg/ioutil"
|
|
)
|
|
|
|
// Heal heals the shard files on non-nil writers. Note that the quorum passed is 1
|
|
// as healing should continue even if it has been successful healing only one shard file.
|
|
func (e Erasure) Heal(ctx context.Context, readers []io.ReaderAt, writers []io.Writer, size int64, bp *bpool.BytePoolCap) error {
|
|
r, w := xioutil.WaitPipe()
|
|
go func() {
|
|
_, err := e.Decode(ctx, w, readers, 0, size, size, nil)
|
|
w.CloseWithError(err)
|
|
}()
|
|
|
|
// Fetch buffer for I/O, returns from the pool if not allocates a new one and returns.
|
|
var buffer []byte
|
|
switch {
|
|
case size == 0:
|
|
buffer = make([]byte, 1) // Allocate atleast a byte to reach EOF
|
|
case size >= e.blockSize:
|
|
buffer = bp.Get()
|
|
defer bp.Put(buffer)
|
|
case size < e.blockSize:
|
|
// No need to allocate fully blockSizeV1 buffer if the incoming data is smaller.
|
|
buffer = make([]byte, size, 2*size+int64(e.parityBlocks+e.dataBlocks-1))
|
|
}
|
|
|
|
// quorum is 1 because CreateFile should continue writing as long as we are writing to even 1 disk.
|
|
n, err := e.Encode(ctx, r, writers, buffer, 1)
|
|
if err == nil && n != size {
|
|
logger.LogIf(ctx, errLessData)
|
|
err = errLessData
|
|
}
|
|
r.CloseWithError(err)
|
|
return err
|
|
}
|