doesnm/quickshare
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(fileinfostore): support GetSharing

Browse source
This commit is contained in:
hexxa 2021-08-11 15:29:53 +08:00 • committed by Hexxa
parent c78692df52
commit e74c3c34e3
2 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/fileinfostore/file_info_store.go
View file

@ -13,6 +13,7 @@ const (
type IFileInfoStore interface { type IFileInfoStore interface {
AddSharing(dirPath string) error AddSharing(dirPath string) error
DelSharing(dirPath string) error DelSharing(dirPath string) error
GetSharing(dirPath string) (bool, bool)
ListSharings(prefix string) (map[string]bool, error) ListSharings(prefix string) (map[string]bool, error)
} }
@ -47,6 +48,10 @@ func (us *FileInfoStore) DelSharing(dirPath string) error {
return us.store.DelBoolIn(SharingNs, dirPath) return us.store.DelBoolIn(SharingNs, dirPath)
} }
func (us *FileInfoStore) GetSharing(dirPath string) (bool, bool) {
return us.store.GetBoolIn(SharingNs, dirPath)
}
func (us *FileInfoStore) ListSharings(prefix string) (map[string]bool, error) { func (us *FileInfoStore) ListSharings(prefix string) (map[string]bool, error) {
return us.store.ListBoolsByPrefixIn(prefix, SharingNs) return us.store.ListBoolsByPrefixIn(prefix, SharingNs)
} }

8
src/fileinfostore/file_info_store_test.go
View file

@ -30,6 +30,10 @@ func TestUserStores(t *testing.T) {
if !sharingMap[sharingDir] { if !sharingMap[sharingDir] {
t.Fatalf("sharing(%s) not found", sharingDir) t.Fatalf("sharing(%s) not found", sharingDir)
} }
mustTrue, exist := store.GetSharing(sharingDir)
if !mustTrue || !exist {
t.Fatalf("get sharing(%t %t) should exist", mustTrue, exist)
}
} }
for _, dirPath := range dirPaths { for _, dirPath := range dirPaths {
@ -47,6 +51,10 @@ func TestUserStores(t *testing.T) {
if sharingMap[dirPath] { if sharingMap[dirPath] {
t.Fatalf("sharing(%s) should not exist", dirPath) t.Fatalf("sharing(%s) should not exist", dirPath)
} }
_, exist := store.GetSharing(dirPath)
if exist {
t.Fatalf("get sharing(%t) should not exit", exist)
}
} }
} }