Chris Lu
4 years ago
5 changed files with 100 additions and 4 deletions
-
5go.mod
-
14go.sum
-
48weed/filer/filer_conf.go
-
31weed/filer/filer_conf_test.go
-
6weed/pb/filer_pb/filer_pb_helper.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 |
||||
|
} |
||||
@ -0,0 +1,31 @@ |
|||||
|
package filer |
||||
|
|
||||
|
import ( |
||||
|
"testing" |
||||
|
|
||||
|
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" |
||||
|
"github.com/stretchr/testify/assert" |
||||
|
"github.com/viant/ptrie" |
||||
|
) |
||||
|
|
||||
|
func TestFilerConf(t *testing.T) { |
||||
|
|
||||
|
fc := &FilerConf{ |
||||
|
rules: ptrie.New(), |
||||
|
} |
||||
|
conf := &filer_pb.FilerConf{Locations: []*filer_pb.FilerConf_PathConf{ |
||||
|
{ |
||||
|
LocationPrefix: "/buckets/abc", |
||||
|
Collection: "abc", |
||||
|
}, |
||||
|
{ |
||||
|
LocationPrefix: "/buckets/abcd", |
||||
|
Collection: "abcd", |
||||
|
}, |
||||
|
}} |
||||
|
fc.doLoadConf(conf) |
||||
|
|
||||
|
assert.Equal(t, "abc", fc.MatchStorageRule("/buckets/abc/jasdf").Collection) |
||||
|
assert.Equal(t, "abcd", fc.MatchStorageRule("/buckets/abcd/jasdf").Collection) |
||||
|
|
||||
|
} |
||||
Write
Preview
Loading…
Cancel
Save
Reference in new issue