2016-09-09 21:53:09 +02:00
|
|
|
/*
|
2019-04-09 20:39:42 +02:00
|
|
|
* MinIO Cloud Storage, (C) 2016 MinIO, Inc.
|
2016-09-09 21:53:09 +02:00
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package cmd
|
|
|
|
|
2016-09-16 22:06:49 +02:00
|
|
|
import (
|
2019-01-17 13:58:18 +01:00
|
|
|
"io"
|
2016-09-16 22:06:49 +02:00
|
|
|
"sync"
|
|
|
|
)
|
2016-09-10 00:32:08 +02:00
|
|
|
|
2016-09-09 21:53:09 +02:00
|
|
|
// naughtyDisk wraps a POSIX disk and returns programmed errors
|
|
|
|
// specified by the developer. The purpose is to simulate errors
|
2016-10-05 21:48:07 +02:00
|
|
|
// that are hard to simulate in practice like DiskNotFound.
|
2016-09-09 21:53:09 +02:00
|
|
|
// Programmed errors are stored in errors field.
|
|
|
|
type naughtyDisk struct {
|
|
|
|
// The real disk
|
2018-02-16 02:45:57 +01:00
|
|
|
disk StorageAPI
|
2016-09-09 21:53:09 +02:00
|
|
|
// Programmed errors: API call number => error to return
|
|
|
|
errors map[int]error
|
|
|
|
// The error to return when no error value is programmed
|
|
|
|
defaultErr error
|
|
|
|
// The current API call number
|
|
|
|
callNR int
|
2016-09-16 22:06:49 +02:00
|
|
|
// Data protection
|
|
|
|
mu sync.Mutex
|
2016-09-09 21:53:09 +02:00
|
|
|
}
|
|
|
|
|
2018-02-16 02:45:57 +01:00
|
|
|
func newNaughtyDisk(d StorageAPI, errs map[int]error, defaultErr error) *naughtyDisk {
|
2016-09-09 21:53:09 +02:00
|
|
|
return &naughtyDisk{disk: d, errors: errs, defaultErr: defaultErr}
|
|
|
|
}
|
|
|
|
|
2016-10-05 21:48:07 +02:00
|
|
|
func (d *naughtyDisk) String() string {
|
2016-10-20 04:59:48 +02:00
|
|
|
return d.disk.String()
|
2016-10-05 21:48:07 +02:00
|
|
|
}
|
|
|
|
|
2018-02-16 02:45:57 +01:00
|
|
|
func (d *naughtyDisk) IsOnline() bool {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err == errDiskNotFound
|
2016-11-24 00:48:10 +01:00
|
|
|
}
|
2018-02-16 02:45:57 +01:00
|
|
|
return d.disk.IsOnline()
|
2016-11-24 00:48:10 +01:00
|
|
|
}
|
|
|
|
|
2020-01-13 22:09:10 +01:00
|
|
|
func (*naughtyDisk) Hostname() string {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2016-11-24 00:48:10 +01:00
|
|
|
func (d *naughtyDisk) Close() (err error) {
|
|
|
|
if err = d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.Close()
|
|
|
|
}
|
|
|
|
|
2016-09-09 21:53:09 +02:00
|
|
|
func (d *naughtyDisk) calcError() (err error) {
|
2016-09-16 22:06:49 +02:00
|
|
|
d.mu.Lock()
|
|
|
|
defer d.mu.Unlock()
|
2016-09-09 21:53:09 +02:00
|
|
|
d.callNR++
|
|
|
|
if err, ok := d.errors[d.callNR]; ok {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if d.defaultErr != nil {
|
|
|
|
return d.defaultErr
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-10-25 19:37:53 +02:00
|
|
|
func (d *naughtyDisk) SetDiskID(id string) {
|
|
|
|
}
|
|
|
|
|
2019-12-12 15:02:37 +01:00
|
|
|
func (d *naughtyDisk) CrawlAndGetDataUsage(endCh <-chan struct{}) (info DataUsageInfo, err error) {
|
|
|
|
return d.disk.CrawlAndGetDataUsage(endCh)
|
|
|
|
}
|
|
|
|
|
2018-05-23 12:11:29 +02:00
|
|
|
func (d *naughtyDisk) DiskInfo() (info DiskInfo, err error) {
|
2016-09-10 00:32:08 +02:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return info, err
|
|
|
|
}
|
|
|
|
return d.disk.DiskInfo()
|
|
|
|
}
|
|
|
|
|
2019-12-24 01:31:03 +01:00
|
|
|
func (d *naughtyDisk) MakeVolBulk(volumes ...string) (err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.MakeVolBulk(volumes...)
|
|
|
|
}
|
|
|
|
|
2016-09-09 21:53:09 +02:00
|
|
|
func (d *naughtyDisk) MakeVol(volume string) (err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.MakeVol(volume)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) ListVols() (vols []VolInfo, err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return d.disk.ListVols()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) StatVol(volume string) (volInfo VolInfo, err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return VolInfo{}, err
|
|
|
|
}
|
|
|
|
return d.disk.StatVol(volume)
|
|
|
|
}
|
|
|
|
func (d *naughtyDisk) DeleteVol(volume string) (err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.DeleteVol(volume)
|
|
|
|
}
|
|
|
|
|
2019-05-14 22:49:10 +02:00
|
|
|
func (d *naughtyDisk) Walk(volume, path, marker string, recursive bool, leafFile string, readMetadataFn readMetadataFunc, endWalkCh chan struct{}) (chan FileInfo, error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return d.disk.Walk(volume, path, marker, recursive, leafFile, readMetadataFn, endWalkCh)
|
|
|
|
}
|
|
|
|
|
2019-04-23 23:54:28 +02:00
|
|
|
func (d *naughtyDisk) ListDir(volume, path string, count int, leafFile string) (entries []string, err error) {
|
2016-09-09 21:53:09 +02:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return []string{}, err
|
|
|
|
}
|
2019-04-23 23:54:28 +02:00
|
|
|
return d.disk.ListDir(volume, path, count, leafFile)
|
2016-09-09 21:53:09 +02:00
|
|
|
}
|
|
|
|
|
2017-09-25 20:32:56 +02:00
|
|
|
func (d *naughtyDisk) ReadFile(volume string, path string, offset int64, buf []byte, verifier *BitrotVerifier) (n int64, err error) {
|
2016-09-09 21:53:09 +02:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
2017-09-25 20:32:56 +02:00
|
|
|
return d.disk.ReadFile(volume, path, offset, buf, verifier)
|
2017-05-16 23:21:52 +02:00
|
|
|
}
|
|
|
|
|
2019-01-17 13:58:18 +01:00
|
|
|
func (d *naughtyDisk) ReadFileStream(volume, path string, offset, length int64) (io.ReadCloser, error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return d.disk.ReadFileStream(volume, path, offset, length)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) CreateFile(volume, path string, size int64, reader io.Reader) error {
|
2016-10-29 21:44:44 +02:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-01-17 13:58:18 +01:00
|
|
|
return d.disk.CreateFile(volume, path, size, reader)
|
2016-10-29 21:44:44 +02:00
|
|
|
}
|
|
|
|
|
2016-09-09 21:53:09 +02:00
|
|
|
func (d *naughtyDisk) AppendFile(volume, path string, buf []byte) error {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.AppendFile(volume, path, buf)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) RenameFile(srcVolume, srcPath, dstVolume, dstPath string) error {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.RenameFile(srcVolume, srcPath, dstVolume, dstPath)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) StatFile(volume string, path string) (file FileInfo, err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return FileInfo{}, err
|
|
|
|
}
|
|
|
|
return d.disk.StatFile(volume, path)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *naughtyDisk) DeleteFile(volume string, path string) (err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return d.disk.DeleteFile(volume, path)
|
|
|
|
}
|
|
|
|
|
2019-05-13 21:25:49 +02:00
|
|
|
func (d *naughtyDisk) DeleteFileBulk(volume string, paths []string) ([]error, error) {
|
|
|
|
errs := make([]error, len(paths))
|
|
|
|
for idx, path := range paths {
|
|
|
|
errs[idx] = d.disk.DeleteFile(volume, path)
|
|
|
|
}
|
|
|
|
return errs, nil
|
|
|
|
}
|
|
|
|
|
2019-05-22 22:47:15 +02:00
|
|
|
func (d *naughtyDisk) WriteAll(volume string, path string, reader io.Reader) (err error) {
|
2018-11-14 15:18:35 +01:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-05-22 22:47:15 +02:00
|
|
|
return d.disk.WriteAll(volume, path, reader)
|
2018-11-14 15:18:35 +01:00
|
|
|
}
|
|
|
|
|
2016-09-09 21:53:09 +02:00
|
|
|
func (d *naughtyDisk) ReadAll(volume string, path string) (buf []byte, err error) {
|
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return d.disk.ReadAll(volume, path)
|
|
|
|
}
|
2019-07-08 22:51:18 +02:00
|
|
|
|
2019-09-11 22:49:53 +02:00
|
|
|
func (d *naughtyDisk) VerifyFile(volume, path string, size int64, algo BitrotAlgorithm, sum []byte, shardSize int64) error {
|
2019-07-08 22:51:18 +02:00
|
|
|
if err := d.calcError(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-09-11 22:49:53 +02:00
|
|
|
return d.disk.VerifyFile(volume, path, size, algo, sum, shardSize)
|
2019-07-08 22:51:18 +02:00
|
|
|
}
|