parent
92a37e5113
commit
554a244b47
@ -0,0 +1,153 @@ |
||||
/* |
||||
* Minimalist Object Storage, (C) 2015 Minio, Inc. |
||||
* |
||||
* 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 donut |
||||
|
||||
import ( |
||||
"errors" |
||||
"os" |
||||
"path" |
||||
"syscall" |
||||
|
||||
"io/ioutil" |
||||
|
||||
"github.com/minio/minio/pkg/iodine" |
||||
) |
||||
|
||||
// internal disk struct
|
||||
type disk struct { |
||||
root string |
||||
order int |
||||
filesystem map[string]string |
||||
} |
||||
|
||||
// NewDisk - instantiate new disk
|
||||
func NewDisk(diskPath string, diskOrder int) (Disk, error) { |
||||
if diskPath == "" || diskOrder < 0 { |
||||
return nil, iodine.New(errors.New("invalid argument"), nil) |
||||
} |
||||
s := syscall.Statfs_t{} |
||||
err := syscall.Statfs(diskPath, &s) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
st, err := os.Stat(diskPath) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
if !st.IsDir() { |
||||
return nil, iodine.New(syscall.ENOTDIR, nil) |
||||
} |
||||
d := disk{ |
||||
root: diskPath, |
||||
order: diskOrder, |
||||
filesystem: make(map[string]string), |
||||
} |
||||
if fsType := d.getFSType(s.Type); fsType != "UNKNOWN" { |
||||
d.filesystem["FSType"] = fsType |
||||
d.filesystem["MountPoint"] = d.root |
||||
return d, nil |
||||
} |
||||
return nil, iodine.New(errors.New("unsupported filesystem"), nil) |
||||
} |
||||
|
||||
// GetPath - get root disk path
|
||||
func (d disk) GetPath() string { |
||||
return d.root |
||||
} |
||||
|
||||
// GetOrder - get order of disk present in graph
|
||||
func (d disk) GetOrder() int { |
||||
return d.order |
||||
} |
||||
|
||||
// GetFSInfo - get disk filesystem and its usage information
|
||||
func (d disk) GetFSInfo() map[string]string { |
||||
s := syscall.Statfs_t{} |
||||
err := syscall.Statfs(d.root, &s) |
||||
if err != nil { |
||||
return nil |
||||
} |
||||
d.filesystem["Total"] = d.formatBytes(uint64(s.Bsize) * s.Blocks) |
||||
d.filesystem["Free"] = d.formatBytes(uint64(s.Bsize) * s.Bfree) |
||||
return d.filesystem |
||||
} |
||||
|
||||
// MakeDir - make a directory inside disk root path
|
||||
func (d disk) MakeDir(dirname string) error { |
||||
return os.MkdirAll(path.Join(d.root, dirname), 0700) |
||||
} |
||||
|
||||
// ListDir - list a directory inside disk root path, get only directories
|
||||
func (d disk) ListDir(dirname string) ([]os.FileInfo, error) { |
||||
contents, err := ioutil.ReadDir(path.Join(d.root, dirname)) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
var directories []os.FileInfo |
||||
for _, content := range contents { |
||||
// Include only directories, ignore everything else
|
||||
if content.IsDir() { |
||||
directories = append(directories, content) |
||||
} |
||||
} |
||||
return directories, nil |
||||
} |
||||
|
||||
// ListFiles - list a directory inside disk root path, get only files
|
||||
func (d disk) ListFiles(dirname string) ([]os.FileInfo, error) { |
||||
contents, err := ioutil.ReadDir(path.Join(d.root, dirname)) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
var files []os.FileInfo |
||||
for _, content := range contents { |
||||
// Include only regular files, ignore everything else
|
||||
if content.Mode().IsRegular() { |
||||
files = append(files, content) |
||||
} |
||||
} |
||||
return files, nil |
||||
} |
||||
|
||||
// MakeFile - create a file inside disk root path
|
||||
func (d disk) MakeFile(filename string) (*os.File, error) { |
||||
if filename == "" { |
||||
return nil, iodine.New(errors.New("Invalid argument"), nil) |
||||
} |
||||
filePath := path.Join(d.root, filename) |
||||
// Create directories if they don't exist
|
||||
if err := os.MkdirAll(path.Dir(filePath), 0700); err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
dataFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE, 0600) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
return dataFile, nil |
||||
} |
||||
|
||||
// OpenFile - read a file inside disk root path
|
||||
func (d disk) OpenFile(filename string) (*os.File, error) { |
||||
if filename == "" { |
||||
return nil, iodine.New(errors.New("Invalid argument"), nil) |
||||
} |
||||
dataFile, err := os.Open(path.Join(d.root, filename)) |
||||
if err != nil { |
||||
return nil, iodine.New(err, nil) |
||||
} |
||||
return dataFile, nil |
||||
} |
@ -0,0 +1,58 @@ |
||||
/* |
||||
* Minimalist Object Storage, (C) 2015 Minio, Inc. |
||||
* |
||||
* 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 donut |
||||
|
||||
import ( |
||||
"fmt" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
/// This file contains all the internal functions used inside Disk interface
|
||||
|
||||
// formatBytes - Convert bytes to human readable string. Like a 2 MB, 64.2 KB, 52 B
|
||||
func (d disk) formatBytes(i uint64) (result string) { |
||||
switch { |
||||
case i > (1024 * 1024 * 1024 * 1024): |
||||
result = fmt.Sprintf("%.02f TB", float64(i)/1024/1024/1024/1024) |
||||
case i > (1024 * 1024 * 1024): |
||||
result = fmt.Sprintf("%.02f GB", float64(i)/1024/1024/1024) |
||||
case i > (1024 * 1024): |
||||
result = fmt.Sprintf("%.02f MB", float64(i)/1024/1024) |
||||
case i > 1024: |
||||
result = fmt.Sprintf("%.02f KB", float64(i)/1024) |
||||
default: |
||||
result = fmt.Sprintf("%d B", i) |
||||
} |
||||
result = strings.Trim(result, " ") |
||||
return |
||||
} |
||||
|
||||
// fsType2StrinMap - list of filesystems supported by donut
|
||||
var fsType2StringMap = map[string]string{ |
||||
"11": "HFS", |
||||
} |
||||
|
||||
// getFSType - get filesystem type
|
||||
func (d disk) getFSType(fsType uint32) string { |
||||
fsTypeHex := strconv.FormatUint(uint64(fsType), 16) |
||||
fsTypeString, ok := fsType2StringMap[fsTypeHex] |
||||
if ok == false { |
||||
return "UNKNOWN" |
||||
} |
||||
return fsTypeString |
||||
} |
Loading…
Reference in new issue