seaweedfs/weed/filer/filer_conf.go

150 lines
3.5 KiB
Go
Raw Normal View History

2020-11-15 08:26:05 +00:00
package filer
import (
2020-11-16 05:48:17 +00:00
"bytes"
2020-11-15 22:06:03 +00:00
"context"
"io"
2020-11-15 22:06:03 +00:00
2020-11-15 08:26:05 +00:00
"github.com/chrislusf/seaweedfs/weed/glog"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
2020-11-15 22:06:03 +00:00
"github.com/chrislusf/seaweedfs/weed/util"
2020-11-16 05:48:17 +00:00
"github.com/golang/protobuf/jsonpb"
2020-11-15 08:26:05 +00:00
"github.com/viant/ptrie"
)
2020-11-15 22:06:03 +00:00
const (
DirectoryEtcRoot = "/etc"
DirectoryEtcSeaweedFS = "/etc/seaweedfs"
FilerConfName = "filer.conf"
IamConfigDirecotry = "/etc/iam"
IamIdentityFile = "identity.json"
2021-04-10 18:57:45 +00:00
IamPoliciesFile = "policies.json"
2020-11-15 22:06:03 +00:00
)
2020-11-15 08:26:05 +00:00
type FilerConf struct {
rules ptrie.Trie
}
2020-11-15 22:06:03 +00:00
func NewFilerConf() (fc *FilerConf) {
2020-11-15 08:26:05 +00:00
fc = &FilerConf{
rules: ptrie.New(),
}
2020-11-15 22:06:03 +00:00
return fc
}
2020-11-16 00:59:28 +00:00
func (fc *FilerConf) loadFromFiler(filer *Filer) (err error) {
filerConfPath := util.NewFullPath(DirectoryEtcSeaweedFS, FilerConfName)
2020-11-15 22:06:03 +00:00
entry, err := filer.FindEntry(context.Background(), filerConfPath)
if err != nil {
if err == filer_pb.ErrNotFound {
return nil
}
glog.Errorf("read filer conf entry %s: %v", filerConfPath, err)
return
}
2020-11-30 12:34:04 +00:00
if len(entry.Content) > 0 {
return fc.LoadFromBytes(entry.Content)
}
return fc.loadFromChunks(filer, entry.Content, entry.Chunks)
2020-11-15 22:06:03 +00:00
}
func (fc *FilerConf) loadFromChunks(filer *Filer, content []byte, chunks []*filer_pb.FileChunk) (err error) {
if len(content) == 0 {
content, err = filer.readEntry(chunks)
if err != nil {
glog.Errorf("read filer conf content: %v", err)
return
}
2020-11-15 22:06:03 +00:00
}
return fc.LoadFromBytes(content)
2020-11-15 22:06:03 +00:00
}
func (fc *FilerConf) LoadFromBytes(data []byte) (err error) {
2020-11-15 08:26:05 +00:00
conf := &filer_pb.FilerConf{}
2020-11-16 05:48:17 +00:00
if err := jsonpb.Unmarshal(bytes.NewReader(data), conf); err != nil {
2020-12-07 03:47:06 +00:00
return err
2020-11-15 08:26:05 +00:00
}
2020-11-15 22:06:03 +00:00
return fc.doLoadConf(conf)
2020-11-15 08:26:05 +00:00
}
2020-11-15 22:06:03 +00:00
func (fc *FilerConf) doLoadConf(conf *filer_pb.FilerConf) (err error) {
2020-11-15 08:26:05 +00:00
for _, location := range conf.Locations {
err = fc.AddLocationConf(location)
2020-11-15 08:26:05 +00:00
if err != nil {
2020-11-15 22:06:03 +00:00
// this is not recoverable
return nil
2020-11-15 08:26:05 +00:00
}
}
2020-11-15 22:06:03 +00:00
return nil
2020-11-15 08:26:05 +00:00
}
func (fc *FilerConf) AddLocationConf(locConf *filer_pb.FilerConf_PathConf) (err error) {
err = fc.rules.Put([]byte(locConf.LocationPrefix), locConf)
if err != nil {
glog.Errorf("put location prefix: %v", err)
}
return
}
func (fc *FilerConf) DeleteLocationConf(locationPrefix string) {
rules := ptrie.New()
fc.rules.Walk(func(key []byte, value interface{}) bool {
if string(key) == locationPrefix {
return true
}
rules.Put(key, value)
return true
})
fc.rules = rules
return
}
2020-11-16 00:59:28 +00:00
func (fc *FilerConf) MatchStorageRule(path string) (pathConf *filer_pb.FilerConf_PathConf) {
pathConf = &filer_pb.FilerConf_PathConf{}
2020-11-15 08:26:05 +00:00
fc.rules.MatchPrefix([]byte(path), func(key []byte, value interface{}) bool {
t := value.(*filer_pb.FilerConf_PathConf)
mergePathConf(pathConf, t)
2020-11-15 08:26:05 +00:00
return true
})
return pathConf
}
// merge if values in b is not empty, merge them into a
func mergePathConf(a, b *filer_pb.FilerConf_PathConf) {
a.Collection = util.Nvl(b.Collection, a.Collection)
a.Replication = util.Nvl(b.Replication, a.Replication)
a.Ttl = util.Nvl(b.Ttl, a.Ttl)
2020-12-13 19:59:32 +00:00
if b.DiskType != "" {
a.DiskType = b.DiskType
}
a.Fsync = b.Fsync || a.Fsync
if b.VolumeGrowthCount > 0 {
a.VolumeGrowthCount = b.VolumeGrowthCount
}
}
func (fc *FilerConf) ToProto() *filer_pb.FilerConf {
m := &filer_pb.FilerConf{}
fc.rules.Walk(func(key []byte, value interface{}) bool {
pathConf := value.(*filer_pb.FilerConf_PathConf)
m.Locations = append(m.Locations, pathConf)
return true
})
return m
}
func (fc *FilerConf) ToText(writer io.Writer) error {
2020-11-16 05:48:17 +00:00
m := jsonpb.Marshaler{
EmitDefaults: false,
Indent: " ",
}
return m.Marshal(writer, fc.ToProto())
}