mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-06-02 16:50:25 +02:00
able to configure the quota for a bucket
This commit is contained in:
parent
b1063162b6
commit
606667f205
|
@ -43,6 +43,7 @@ type Entry struct {
|
||||||
HardLinkCounter int32
|
HardLinkCounter int32
|
||||||
Content []byte
|
Content []byte
|
||||||
Remote *filer_pb.RemoteEntry
|
Remote *filer_pb.RemoteEntry
|
||||||
|
Quota int64
|
||||||
}
|
}
|
||||||
|
|
||||||
func (entry *Entry) Size() uint64 {
|
func (entry *Entry) Size() uint64 {
|
||||||
|
@ -70,6 +71,7 @@ func (entry *Entry) ShallowClone() *Entry {
|
||||||
newEntry.HardLinkCounter = entry.HardLinkCounter
|
newEntry.HardLinkCounter = entry.HardLinkCounter
|
||||||
newEntry.Content = entry.Content
|
newEntry.Content = entry.Content
|
||||||
newEntry.Remote = entry.Remote
|
newEntry.Remote = entry.Remote
|
||||||
|
newEntry.Quota = entry.Quota
|
||||||
|
|
||||||
return newEntry
|
return newEntry
|
||||||
}
|
}
|
||||||
|
@ -96,6 +98,7 @@ func (entry *Entry) ToExistingProtoEntry(message *filer_pb.Entry) {
|
||||||
message.HardLinkCounter = entry.HardLinkCounter
|
message.HardLinkCounter = entry.HardLinkCounter
|
||||||
message.Content = entry.Content
|
message.Content = entry.Content
|
||||||
message.RemoteEntry = entry.Remote
|
message.RemoteEntry = entry.Remote
|
||||||
|
message.Quota = entry.Quota
|
||||||
}
|
}
|
||||||
|
|
||||||
func FromPbEntryToExistingEntry(message *filer_pb.Entry, fsEntry *Entry) {
|
func FromPbEntryToExistingEntry(message *filer_pb.Entry, fsEntry *Entry) {
|
||||||
|
@ -106,6 +109,7 @@ func FromPbEntryToExistingEntry(message *filer_pb.Entry, fsEntry *Entry) {
|
||||||
fsEntry.HardLinkCounter = message.HardLinkCounter
|
fsEntry.HardLinkCounter = message.HardLinkCounter
|
||||||
fsEntry.Content = message.Content
|
fsEntry.Content = message.Content
|
||||||
fsEntry.Remote = message.RemoteEntry
|
fsEntry.Remote = message.RemoteEntry
|
||||||
|
fsEntry.Quota = message.Quota
|
||||||
}
|
}
|
||||||
|
|
||||||
func (entry *Entry) ToProtoFullEntry() *filer_pb.FullEntry {
|
func (entry *Entry) ToProtoFullEntry() *filer_pb.FullEntry {
|
||||||
|
|
|
@ -118,6 +118,9 @@ func EqualEntry(a, b *Entry) bool {
|
||||||
if !proto.Equal(a.Remote, b.Remote) {
|
if !proto.Equal(a.Remote, b.Remote) {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
if a.Quota != b.Quota {
|
||||||
|
return false
|
||||||
|
}
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
93
weed/shell/command_s3_bucket_quota.go
Normal file
93
weed/shell/command_s3_bucket_quota.go
Normal file
|
@ -0,0 +1,93 @@
|
||||||
|
package shell
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
||||||
|
"io"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
Commands = append(Commands, &commandS3BucketQuota{})
|
||||||
|
}
|
||||||
|
|
||||||
|
type commandS3BucketQuota struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *commandS3BucketQuota) Name() string {
|
||||||
|
return "s3.bucket.quota"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *commandS3BucketQuota) Help() string {
|
||||||
|
return `set/remove/enable/disable quota for a bucket
|
||||||
|
|
||||||
|
Example:
|
||||||
|
s3.bucket.quota -name=<bucket_name> -operation=set -sizeMB=1024
|
||||||
|
`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *commandS3BucketQuota) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
|
||||||
|
|
||||||
|
bucketCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
|
||||||
|
bucketName := bucketCommand.String("name", "", "bucket name")
|
||||||
|
operationName := bucketCommand.String("op", "set", "operation name [set|remove|enable|disable]")
|
||||||
|
sizeMB := bucketCommand.Int64("sizeMB", 0, "bucket quota size in MiB")
|
||||||
|
if err = bucketCommand.Parse(args); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if *bucketName == "" {
|
||||||
|
return fmt.Errorf("empty bucket name")
|
||||||
|
}
|
||||||
|
|
||||||
|
err = commandEnv.WithFilerClient(false, func(client filer_pb.SeaweedFilerClient) error {
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
resp, err := client.GetFilerConfiguration(ctx, &filer_pb.GetFilerConfigurationRequest{})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("get filer configuration: %v", err)
|
||||||
|
}
|
||||||
|
filerBucketsPath := resp.DirBuckets
|
||||||
|
|
||||||
|
lookupResp, err := client.LookupDirectoryEntry(ctx, &filer_pb.LookupDirectoryEntryRequest{
|
||||||
|
Directory: filerBucketsPath,
|
||||||
|
Name: *bucketName,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("did not find bucket %s: %v", *bucketName, err)
|
||||||
|
}
|
||||||
|
bucketEntry := lookupResp.Entry
|
||||||
|
|
||||||
|
switch *operationName {
|
||||||
|
case "set":
|
||||||
|
bucketEntry.Quota = *sizeMB * 1024 * 1024
|
||||||
|
case "remove":
|
||||||
|
bucketEntry.Quota = 0
|
||||||
|
case "enable":
|
||||||
|
if bucketEntry.Quota < 0 {
|
||||||
|
bucketEntry.Quota = -bucketEntry.Quota
|
||||||
|
}
|
||||||
|
case "disable":
|
||||||
|
if bucketEntry.Quota > 0 {
|
||||||
|
bucketEntry.Quota = -bucketEntry.Quota
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := filer_pb.UpdateEntry(client, &filer_pb.UpdateEntryRequest{
|
||||||
|
Directory: filerBucketsPath,
|
||||||
|
Entry: bucketEntry,
|
||||||
|
}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
println("updated quota for bucket", *bucketName)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
return err
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in a new issue