mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
130 lines
3.9 KiB
Go
130 lines
3.9 KiB
Go
package shell
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/filer"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/storage/super_block"
|
|
)
|
|
|
|
func init() {
|
|
Commands = append(Commands, &commandFsConfigure{})
|
|
}
|
|
|
|
type commandFsConfigure struct {
|
|
}
|
|
|
|
func (c *commandFsConfigure) Name() string {
|
|
return "fs.configure"
|
|
}
|
|
|
|
func (c *commandFsConfigure) Help() string {
|
|
return `configure and apply storage options for each location
|
|
|
|
# see the current configuration file content
|
|
fs.configure
|
|
|
|
# trying the changes and see the possible configuration file content
|
|
fs.configure -locationPrfix=/my/folder -collection=abc
|
|
fs.configure -locationPrfix=/my/folder -collection=abc -ttl=7d
|
|
|
|
# example: configure adding only 1 physical volume for each bucket collection
|
|
fs.configure -locationPrfix=/buckets/ -volumeGrowthCount=1
|
|
|
|
# apply the changes
|
|
fs.configure -locationPrfix=/my/folder -collection=abc -apply
|
|
|
|
# delete the changes
|
|
fs.configure -locationPrfix=/my/folder -delete -apply
|
|
|
|
`
|
|
}
|
|
|
|
func (c *commandFsConfigure) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
|
|
|
|
fsConfigureCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
|
|
locationPrefix := fsConfigureCommand.String("locationPrefix", "", "path prefix, required to update the path-specific configuration")
|
|
collection := fsConfigureCommand.String("collection", "", "assign writes to this collection")
|
|
replication := fsConfigureCommand.String("replication", "", "assign writes with this replication")
|
|
ttl := fsConfigureCommand.String("ttl", "", "assign writes with this ttl")
|
|
diskType := fsConfigureCommand.String("diskType", "", "[hdd|ssd] choose between hard drive or solid state drive")
|
|
fsync := fsConfigureCommand.Bool("fsync", false, "fsync for the writes")
|
|
volumeGrowthCount := fsConfigureCommand.Int("volumeGrowthCount", 0, "the number of physical volumes to add if no writable volumes")
|
|
isDelete := fsConfigureCommand.Bool("delete", false, "delete the configuration by locationPrefix")
|
|
apply := fsConfigureCommand.Bool("apply", false, "update and apply filer configuration")
|
|
if err = fsConfigureCommand.Parse(args); err != nil {
|
|
return nil
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
if err = commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error {
|
|
return filer.ReadEntry(commandEnv.MasterClient, client, filer.DirectoryEtcSeaweedFS, filer.FilerConfName, &buf)
|
|
}); err != nil && err != filer_pb.ErrNotFound {
|
|
return err
|
|
}
|
|
|
|
fc := filer.NewFilerConf()
|
|
if buf.Len() > 0 {
|
|
if err = fc.LoadFromBytes(buf.Bytes()); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if *locationPrefix != "" {
|
|
locConf := &filer_pb.FilerConf_PathConf{
|
|
LocationPrefix: *locationPrefix,
|
|
Collection: *collection,
|
|
Replication: *replication,
|
|
Ttl: *ttl,
|
|
Fsync: *fsync,
|
|
DiskType: *diskType,
|
|
VolumeGrowthCount: uint32(*volumeGrowthCount),
|
|
}
|
|
|
|
// check collection
|
|
if *collection != "" && strings.HasPrefix(*locationPrefix, "/buckets/") {
|
|
return fmt.Errorf("one s3 bucket goes to one collection and not customizable.")
|
|
}
|
|
|
|
// check replication
|
|
if *replication != "" {
|
|
rp, err := super_block.NewReplicaPlacementFromString(*replication)
|
|
if err != nil {
|
|
return fmt.Errorf("parse replication %s: %v", *replication, err)
|
|
}
|
|
if *volumeGrowthCount%rp.GetCopyCount() != 0 {
|
|
return fmt.Errorf("volumeGrowthCount %d should be devided by replication copy count %d", *volumeGrowthCount, rp.GetCopyCount())
|
|
}
|
|
}
|
|
|
|
// save it
|
|
if *isDelete {
|
|
fc.DeleteLocationConf(*locationPrefix)
|
|
} else {
|
|
fc.AddLocationConf(locConf)
|
|
}
|
|
}
|
|
|
|
buf.Reset()
|
|
fc.ToText(&buf)
|
|
|
|
fmt.Fprintf(writer, string(buf.Bytes()))
|
|
fmt.Fprintln(writer)
|
|
|
|
if *apply {
|
|
|
|
if err := filer.SaveAs(commandEnv.option.FilerHost, int(commandEnv.option.FilerPort), filer.DirectoryEtcSeaweedFS, filer.FilerConfName, "text/plain; charset=utf-8", &buf); err != nil {
|
|
return err
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|