aboutsummaryrefslogtreecommitdiff
path: root/weed/filer/filer_conf.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/filer/filer_conf.go')
-rw-r--r--weed/filer/filer_conf.go48
1 files changed, 48 insertions, 0 deletions
diff --git a/weed/filer/filer_conf.go b/weed/filer/filer_conf.go
new file mode 100644
index 000000000..079794def
--- /dev/null
+++ b/weed/filer/filer_conf.go
@@ -0,0 +1,48 @@
+package filer
+
+import (
+ "github.com/chrislusf/seaweedfs/weed/glog"
+ "github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+ "github.com/golang/protobuf/proto"
+ "github.com/viant/ptrie"
+)
+
+type FilerConf struct {
+ rules ptrie.Trie
+}
+
+func NewFilerConf(data []byte) (fc *FilerConf) {
+ fc = &FilerConf{
+ rules: ptrie.New(),
+ }
+
+ conf := &filer_pb.FilerConf{}
+ err := proto.UnmarshalText(string(data), conf)
+ if err != nil {
+ glog.Errorf("unable to parse filer conf: %v", err)
+ return
+ }
+
+ fc.doLoadConf(conf)
+ return fc
+}
+
+func (fc *FilerConf) doLoadConf(conf *filer_pb.FilerConf) {
+ for _, location := range conf.Locations {
+ err := fc.rules.Put([]byte(location.LocationPrefix), location)
+ if err != nil {
+ glog.Errorf("put location prefix: %v", err)
+ }
+ }
+}
+
+func (fc *FilerConf) MatchStorageRule(path string) (pathConf *filer_pb.FilerConf_PathConf){
+ fc.rules.MatchPrefix([]byte(path), func(key []byte, value interface{}) bool {
+ pathConf = value.(*filer_pb.FilerConf_PathConf)
+ return true
+ })
+ if pathConf == nil {
+ return &filer_pb.FilerConf_PathConf{}
+ }
+ return pathConf
+}