mirror of
https://github.com/syncthing/syncthing.git
synced 2024-11-17 02:48:57 -07:00
232 lines
6.6 KiB
Go
232 lines
6.6 KiB
Go
// Copyright (C) 2014 Jakob Borg and Contributors (see the CONTRIBUTORS file).
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify it
|
|
// under the terms of the GNU General Public License as published by the Free
|
|
// Software Foundation, either version 3 of the License, or (at your option)
|
|
// any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but WITHOUT
|
|
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
// more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
// Package files provides a set type to track local/remote files with newness
|
|
// checks. We must do a certain amount of normalization in here. We will get
|
|
// fed paths with either native or wire-format separators and encodings
|
|
// depending on who calls us. We transform paths to wire-format (NFC and
|
|
// slashes) on the way to the database, and transform to native format
|
|
// (varying separator and encoding) on the way back out.
|
|
package files
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/syncthing/syncthing/internal/lamport"
|
|
"github.com/syncthing/syncthing/internal/protocol"
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
)
|
|
|
|
type fileRecord struct {
|
|
File protocol.FileInfo
|
|
Usage int
|
|
Global bool
|
|
}
|
|
|
|
type bitset uint64
|
|
|
|
type Set struct {
|
|
localVersion map[protocol.DeviceID]uint64
|
|
mutex sync.Mutex
|
|
folder string
|
|
db *leveldb.DB
|
|
blockmap *BlockMap
|
|
}
|
|
|
|
func NewSet(folder string, db *leveldb.DB) *Set {
|
|
var s = Set{
|
|
localVersion: make(map[protocol.DeviceID]uint64),
|
|
folder: folder,
|
|
db: db,
|
|
blockmap: NewBlockMap(db, folder),
|
|
}
|
|
|
|
var deviceID protocol.DeviceID
|
|
ldbWithAllFolderTruncated(db, []byte(folder), func(device []byte, f protocol.FileInfoTruncated) bool {
|
|
copy(deviceID[:], device)
|
|
if f.LocalVersion > s.localVersion[deviceID] {
|
|
s.localVersion[deviceID] = f.LocalVersion
|
|
}
|
|
lamport.Default.Tick(f.Version)
|
|
return true
|
|
})
|
|
if debug {
|
|
l.Debugf("loaded localVersion for %q: %#v", folder, s.localVersion)
|
|
}
|
|
clock(s.localVersion[protocol.LocalDeviceID])
|
|
|
|
return &s
|
|
}
|
|
|
|
func (s *Set) Replace(device protocol.DeviceID, fs []protocol.FileInfo) {
|
|
if debug {
|
|
l.Debugf("%s Replace(%v, [%d])", s.folder, device, len(fs))
|
|
}
|
|
normalizeFilenames(fs)
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
s.localVersion[device] = ldbReplace(s.db, []byte(s.folder), device[:], fs)
|
|
if len(fs) == 0 {
|
|
// Reset the local version if all files were removed.
|
|
s.localVersion[device] = 0
|
|
}
|
|
if device == protocol.LocalDeviceID {
|
|
s.blockmap.Drop()
|
|
s.blockmap.Add(fs)
|
|
}
|
|
}
|
|
|
|
func (s *Set) ReplaceWithDelete(device protocol.DeviceID, fs []protocol.FileInfo) {
|
|
if debug {
|
|
l.Debugf("%s ReplaceWithDelete(%v, [%d])", s.folder, device, len(fs))
|
|
}
|
|
normalizeFilenames(fs)
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
if lv := ldbReplaceWithDelete(s.db, []byte(s.folder), device[:], fs); lv > s.localVersion[device] {
|
|
s.localVersion[device] = lv
|
|
}
|
|
if device == protocol.LocalDeviceID {
|
|
s.blockmap.Drop()
|
|
s.blockmap.Add(fs)
|
|
}
|
|
}
|
|
|
|
func (s *Set) Update(device protocol.DeviceID, fs []protocol.FileInfo) {
|
|
if debug {
|
|
l.Debugf("%s Update(%v, [%d])", s.folder, device, len(fs))
|
|
}
|
|
normalizeFilenames(fs)
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
if device == protocol.LocalDeviceID {
|
|
discards := make([]protocol.FileInfo, 0, len(fs))
|
|
updates := make([]protocol.FileInfo, 0, len(fs))
|
|
for _, newFile := range fs {
|
|
existingFile := ldbGet(s.db, []byte(s.folder), device[:], []byte(newFile.Name))
|
|
if existingFile.Version <= newFile.Version {
|
|
discards = append(discards, existingFile)
|
|
updates = append(updates, newFile)
|
|
}
|
|
}
|
|
s.blockmap.Discard(discards)
|
|
s.blockmap.Update(updates)
|
|
}
|
|
if lv := ldbUpdate(s.db, []byte(s.folder), device[:], fs); lv > s.localVersion[device] {
|
|
s.localVersion[device] = lv
|
|
}
|
|
}
|
|
|
|
func (s *Set) WithNeed(device protocol.DeviceID, fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithNeed(%v)", s.folder, device)
|
|
}
|
|
ldbWithNeed(s.db, []byte(s.folder), device[:], false, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) WithNeedTruncated(device protocol.DeviceID, fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithNeedTruncated(%v)", s.folder, device)
|
|
}
|
|
ldbWithNeed(s.db, []byte(s.folder), device[:], true, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) WithHave(device protocol.DeviceID, fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithHave(%v)", s.folder, device)
|
|
}
|
|
ldbWithHave(s.db, []byte(s.folder), device[:], false, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) WithHaveTruncated(device protocol.DeviceID, fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithHaveTruncated(%v)", s.folder, device)
|
|
}
|
|
ldbWithHave(s.db, []byte(s.folder), device[:], true, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) WithGlobal(fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithGlobal()", s.folder)
|
|
}
|
|
ldbWithGlobal(s.db, []byte(s.folder), false, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) WithGlobalTruncated(fn fileIterator) {
|
|
if debug {
|
|
l.Debugf("%s WithGlobalTruncated()", s.folder)
|
|
}
|
|
ldbWithGlobal(s.db, []byte(s.folder), true, nativeFileIterator(fn))
|
|
}
|
|
|
|
func (s *Set) Get(device protocol.DeviceID, file string) protocol.FileInfo {
|
|
f := ldbGet(s.db, []byte(s.folder), device[:], []byte(normalizedFilename(file)))
|
|
f.Name = nativeFilename(f.Name)
|
|
return f
|
|
}
|
|
|
|
func (s *Set) GetGlobal(file string) protocol.FileInfo {
|
|
f := ldbGetGlobal(s.db, []byte(s.folder), []byte(normalizedFilename(file)))
|
|
f.Name = nativeFilename(f.Name)
|
|
return f
|
|
}
|
|
|
|
func (s *Set) Availability(file string) []protocol.DeviceID {
|
|
return ldbAvailability(s.db, []byte(s.folder), []byte(normalizedFilename(file)))
|
|
}
|
|
|
|
func (s *Set) LocalVersion(device protocol.DeviceID) uint64 {
|
|
s.mutex.Lock()
|
|
defer s.mutex.Unlock()
|
|
return s.localVersion[device]
|
|
}
|
|
|
|
// ListFolders returns the folder IDs seen in the database.
|
|
func ListFolders(db *leveldb.DB) []string {
|
|
return ldbListFolders(db)
|
|
}
|
|
|
|
// DropFolder clears out all information related to the given folder from the
|
|
// database.
|
|
func DropFolder(db *leveldb.DB, folder string) {
|
|
ldbDropFolder(db, []byte(folder))
|
|
bm := &BlockMap{
|
|
db: db,
|
|
folder: folder,
|
|
}
|
|
bm.Drop()
|
|
}
|
|
|
|
func normalizeFilenames(fs []protocol.FileInfo) {
|
|
for i := range fs {
|
|
fs[i].Name = normalizedFilename(fs[i].Name)
|
|
}
|
|
}
|
|
|
|
func nativeFileIterator(fn fileIterator) fileIterator {
|
|
return func(fi protocol.FileIntf) bool {
|
|
switch f := fi.(type) {
|
|
case protocol.FileInfo:
|
|
f.Name = nativeFilename(f.Name)
|
|
return fn(f)
|
|
case protocol.FileInfoTruncated:
|
|
f.Name = nativeFilename(f.Name)
|
|
return fn(f)
|
|
default:
|
|
panic("unknown interface type")
|
|
}
|
|
}
|
|
}
|