seaweedfs/weed/filer/filer_conf.go

149 lines
3.4 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/golang/protobuf/proto"
"github.com/viant/ptrie"
)
2020-11-15 22:06:03 +00:00
const (
2020-11-16 00:59:28 +00:00
DirectoryEtc = "/etc"
2020-11-15 22:06:03 +00:00
FilerConfName = "filer.conf"
)
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) {
2020-11-15 22:06:03 +00:00
filerConfPath := util.NewFullPath(DirectoryEtc, FilerConfName)
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
}
return fc.loadFromChunks(filer, entry.Chunks)
}
func (fc *FilerConf) loadFromChunks(filer *Filer, chunks []*filer_pb.FileChunk) (err error) {
data, err := filer.readEntry(chunks)
if err != nil {
glog.Errorf("read filer conf content: %v", err)
return
}
return fc.LoadFromBytes(data)
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 {
err = proto.UnmarshalText(string(data), conf)
if err != nil {
glog.Errorf("unable to parse filer conf: %v", err)
// this is not recoverable
return nil
}
2020-11-15 22:06:03 +00:00
return nil
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)
if b.DiskType != filer_pb.FilerConf_PathConf_NONE {
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())
}