forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 02:42:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

swarm/storage: sorted references for GetAllReferences

Browse source
This commit is contained in:
Fabio Barone 2019-02-04 11:45:10 -05:00
parent 3dc9065680
commit 7b56fd29cf
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
swarm/storage/filestore.go
View file

@ -19,6 +19,7 @@ package storage
import ( import (
"context" "context"
"io" "io"
"sort"
) )
/* /*
@ -98,8 +99,7 @@ func (f *FileStore) HashSize() int {
} }
// Public API. This endpoint returns all chunk hashes (only) for a given file // Public API. This endpoint returns all chunk hashes (only) for a given file
func (f *FileStore) GetAllReferences(ctx context.Context, data io.Reader, toEncrypt bool) (addrs []Address, err error) { func (f *FileStore) GetAllReferences(ctx context.Context, data io.Reader, toEncrypt bool) (addrs AddressCollection, err error) {
addrs = make([]Address, 0)
// create a special kind of putter, which only will store the references // create a special kind of putter, which only will store the references
putter := &HashExplorer{ putter := &HashExplorer{
hasherStore: NewHasherStore(f.ChunkStore, f.hashFunc, toEncrypt), hasherStore: NewHasherStore(f.ChunkStore, f.hashFunc, toEncrypt),
@ -111,9 +111,12 @@ func (f *FileStore) GetAllReferences(ctx context.Context, data io.Reader, toEncr
return nil, err return nil, err
} }
// collect all references // collect all references
arr := make([]Address, 0)
for _, ref := range putter.References { for _, ref := range putter.References {
addrs = append(addrs, Address(ref)) arr = append(arr, Address(ref))
} }
addrs = NewAddressCollection(len(arr))
sort.Sort(addrs)
return addrs, nil return addrs, nil
} }