bingoohuang
4 years ago
10 changed files with 147 additions and 77 deletions
-
2unmaintained/repeated_vacuum/repeated_vacuum.go
-
9weed/command/server.go
-
26weed/command/volume.go
-
2weed/server/volume_server.go
-
10weed/storage/disk_location.go
-
4weed/storage/store.go
-
22weed/util/bytes.go
-
26weed/util/bytes_test.go
-
90weed/util/minfreespace.go
-
29weed/util/minfreespace_test.go
@ -0,0 +1,90 @@ |
|||||
|
package util |
||||
|
|
||||
|
import ( |
||||
|
"errors" |
||||
|
"fmt" |
||||
|
"github.com/chrislusf/seaweedfs/weed/glog" |
||||
|
"strconv" |
||||
|
"strings" |
||||
|
) |
||||
|
|
||||
|
// MinFreeSpaceType is the type of MinFreeSpace.
|
||||
|
type MinFreeSpaceType int |
||||
|
|
||||
|
const ( |
||||
|
// AsPercent set the MinFreeSpaceType to a percentage value from 0 to 100.
|
||||
|
AsPercent MinFreeSpaceType = iota |
||||
|
// AsBytes set the MinFreeSpaceType to a absolute value bytes.
|
||||
|
AsBytes |
||||
|
) |
||||
|
|
||||
|
// MinFreeSpace is type that defines the limit for the minimum free space.
|
||||
|
type MinFreeSpace struct { |
||||
|
Type MinFreeSpaceType |
||||
|
Bytes uint64 |
||||
|
Percent float32 |
||||
|
Raw string |
||||
|
} |
||||
|
|
||||
|
// IsLow tells whether the free space is low or not.
|
||||
|
func (s MinFreeSpace) IsLow(freeBytes uint64, freePercent float32) (yes bool, desc string) { |
||||
|
switch s.Type { |
||||
|
case AsPercent: |
||||
|
yes = freePercent < s.Percent |
||||
|
op := IfElse(yes, "<", ">=") |
||||
|
return yes, fmt.Sprintf("disk free %.2f%% %s required %.2f%%", freePercent, op, s.Percent) |
||||
|
case AsBytes: |
||||
|
yes = freeBytes < s.Bytes |
||||
|
op := IfElse(yes, "<", ">=") |
||||
|
return yes, fmt.Sprintf("disk free %s %s required %s", |
||||
|
BytesToHumanReadable(freeBytes), op, BytesToHumanReadable(s.Bytes)) |
||||
|
} |
||||
|
|
||||
|
return false, "" |
||||
|
} |
||||
|
|
||||
|
// String returns a string representation of MinFreeSpace.
|
||||
|
func (s MinFreeSpace) String() string { |
||||
|
switch s.Type { |
||||
|
case AsPercent: |
||||
|
return fmt.Sprintf("%.2f%%", s.Percent) |
||||
|
default: |
||||
|
return s.Raw |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// MustParseMinFreeSpace parses comma-separated argument for min free space setting.
|
||||
|
// minFreeSpace has the high priority than minFreeSpacePercent if it is set.
|
||||
|
func MustParseMinFreeSpace(minFreeSpace string, minFreeSpacePercent string) (spaces []MinFreeSpace) { |
||||
|
ss := strings.Split(EmptyTo(minFreeSpace, minFreeSpacePercent), ",") |
||||
|
for _, freeString := range ss { |
||||
|
if vv, e := ParseMinFreeSpace(freeString); e == nil { |
||||
|
spaces = append(spaces, *vv) |
||||
|
} else { |
||||
|
glog.Fatalf("The value specified in -minFreeSpace not a valid value %s", freeString) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return spaces |
||||
|
} |
||||
|
|
||||
|
var ErrMinFreeSpaceBadValue = errors.New("minFreeSpace is invalid") |
||||
|
|
||||
|
// ParseMinFreeSpace parses min free space expression s as percentage like 1,10 or human readable size like 10G
|
||||
|
func ParseMinFreeSpace(s string) (*MinFreeSpace, error) { |
||||
|
if percent, e := strconv.ParseFloat(s, 32); e == nil { |
||||
|
if percent < 0 || percent > 100 { |
||||
|
return nil, ErrMinFreeSpaceBadValue |
||||
|
} |
||||
|
return &MinFreeSpace{Type: AsPercent, Percent: float32(percent), Raw: s}, nil |
||||
|
} |
||||
|
|
||||
|
if directSize, e := ParseBytes(s); e == nil { |
||||
|
if directSize <= 100 { |
||||
|
return nil, ErrMinFreeSpaceBadValue |
||||
|
} |
||||
|
return &MinFreeSpace{Type: AsBytes, Bytes: directSize, Raw: s}, nil |
||||
|
} |
||||
|
|
||||
|
return nil, ErrMinFreeSpaceBadValue |
||||
|
} |
@ -0,0 +1,29 @@ |
|||||
|
package util |
||||
|
|
||||
|
import "testing" |
||||
|
|
||||
|
func TestParseMinFreeSpace(t *testing.T) { |
||||
|
tests := []struct { |
||||
|
in string |
||||
|
ok bool |
||||
|
value *MinFreeSpace |
||||
|
}{ |
||||
|
{in: "42", ok: true, value: &MinFreeSpace{Type: AsPercent, Percent: 42, Raw: "42"}}, |
||||
|
{in: "-1", ok: false, value: nil}, |
||||
|
{in: "101", ok: false, value: nil}, |
||||
|
{in: "100B", ok: false, value: nil}, |
||||
|
{in: "100Ki", ok: true, value: &MinFreeSpace{Type: AsBytes, Bytes: 100 * 1024, Raw: "100Ki"}}, |
||||
|
{in: "100GiB", ok: true, value: &MinFreeSpace{Type: AsBytes, Bytes: 100 * 1024 * 1024 * 1024, Raw: "100GiB"}}, |
||||
|
{in: "42M", ok: true, value: &MinFreeSpace{Type: AsBytes, Bytes: 42 * 1000 * 1000, Raw: "42M"}}, |
||||
|
} |
||||
|
|
||||
|
for _, p := range tests { |
||||
|
got, err := ParseMinFreeSpace(p.in) |
||||
|
if p.ok != (err == nil) { |
||||
|
t.Errorf("failed to test %v", p.in) |
||||
|
} |
||||
|
if p.ok && err == nil && *got != *p.value { |
||||
|
t.Errorf("failed to test %v", p.in) |
||||
|
} |
||||
|
} |
||||
|
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue