gitlab-com-seaweedfs/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2024-01-19 02:48:24 +00:00
Code Issues Projects Releases Wiki Activity

refactoring

Browse source
This commit is contained in:
Chris Lu 2020-12-06 21:54:34 -08:00
parent 9acda432fe
commit 78f3ab439c
2 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
weed/s3iam/s3iam_filer_store.go
View file

@ -33,7 +33,7 @@ func (ifs *IAMFilerStore) LoadIAMConfig(config *iam_pb.S3ApiConfiguration) error
if err != nil {
return err
}
err = ifs.loadIAMConfigFromEntry(resp.Entry, config)
err = ifs.loadIAMConfigFromBytes(resp.Entry.Content, config)
if err != nil {
return err
}
@ -79,8 +79,8 @@ func (ifs *IAMFilerStore) SaveIAMConfig(config *iam_pb.S3ApiConfiguration) error
return nil
}
func (ifs *IAMFilerStore) loadIAMConfigFromEntry(entry *filer_pb.Entry, config *iam_pb.S3ApiConfiguration) error {
if err := proto.Unmarshal(entry.Content, config); err != nil {
func (ifs *IAMFilerStore) loadIAMConfigFromBytes(content []byte, config *iam_pb.S3ApiConfiguration) error {
if err := proto.Unmarshal(content, config); err != nil {
return err
}
return nil

7
weed/s3iam/s3iam_filer_store_test.go
View file

@ -1,9 +1,9 @@
package s3iam
import (
"github.com/golang/protobuf/proto"
"testing"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
"github.com/chrislusf/seaweedfs/weed/pb/iam_pb"
"github.com/stretchr/testify/assert"
@ -51,11 +51,10 @@ func TestS3Conf(t *testing.T) {
},
},
}
entry := filer_pb.Entry{}
err := ifs.saveIAMConfigToEntry(&entry, s3Conf)
content, _ := proto.Marshal(config)
assert.Equal(t, err, nil)
s3ConfSaved := &iam_pb.S3ApiConfiguration{}
err = ifs.loadIAMConfigFromEntry(&entry, s3ConfSaved)
err = ifs.loadIAMConfigFromBytes(content, s3ConfSaved)
assert.Equal(t, err, nil)
assert.Equal(t, "some_name", s3ConfSaved.Identities[0].Name)