42bd42df5a
To do so the BlockMap struct has been removed. It behaves like any other prefixed part of the database, but was not integrated in the recent keyer refactor. Now the database is only flushed when files are in a consistent state.
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
// Copyright (C) 2014 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
package db
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
|
|
"github.com/syncthing/syncthing/lib/osutil"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb/util"
|
|
)
|
|
|
|
var blockFinder *BlockFinder
|
|
|
|
type BlockFinder struct {
|
|
db *instance
|
|
}
|
|
|
|
func NewBlockFinder(db *Lowlevel) *BlockFinder {
|
|
if blockFinder != nil {
|
|
return blockFinder
|
|
}
|
|
|
|
return &BlockFinder{
|
|
db: newInstance(db),
|
|
}
|
|
}
|
|
|
|
func (f *BlockFinder) String() string {
|
|
return fmt.Sprintf("BlockFinder@%p", f)
|
|
}
|
|
|
|
// Iterate takes an iterator function which iterates over all matching blocks
|
|
// for the given hash. The iterator function has to return either true (if
|
|
// they are happy with the block) or false to continue iterating for whatever
|
|
// reason. The iterator finally returns the result, whether or not a
|
|
// satisfying block was eventually found.
|
|
func (f *BlockFinder) Iterate(folders []string, hash []byte, iterFn func(string, string, int32) bool) bool {
|
|
t := f.db.newReadOnlyTransaction()
|
|
defer t.close()
|
|
|
|
var key []byte
|
|
for _, folder := range folders {
|
|
key = f.db.keyer.GenerateBlockMapKey(key, []byte(folder), hash, nil)
|
|
iter := t.NewIterator(util.BytesPrefix(key), nil)
|
|
|
|
for iter.Next() && iter.Error() == nil {
|
|
file := string(f.db.keyer.NameFromBlockMapKey(iter.Key()))
|
|
index := int32(binary.BigEndian.Uint32(iter.Value()))
|
|
if iterFn(folder, osutil.NativeFilename(file), index) {
|
|
iter.Release()
|
|
return true
|
|
}
|
|
}
|
|
iter.Release()
|
|
}
|
|
return false
|
|
}
|