mirror of https://github.com/rclone/rclone.git
cache: move methods used for testing into test file
This commit is contained in:
parent
47e2d5c415
commit
0f20f23651
|
@ -980,15 +980,6 @@ func (b *Persistent) updatePendingUpload(remote string, fn func(item *tempUpload
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetPendingUploadToStarted is a way to mark an entry as started (even if it's not already)
|
|
||||||
// TO BE USED IN TESTING ONLY
|
|
||||||
func (b *Persistent) SetPendingUploadToStarted(remote string) error {
|
|
||||||
return b.updatePendingUpload(remote, func(item *tempUploadInfo) error {
|
|
||||||
item.Started = true
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReconcileTempUploads will recursively look for all the files in the temp directory and add them to the queue
|
// ReconcileTempUploads will recursively look for all the files in the temp directory and add them to the queue
|
||||||
func (b *Persistent) ReconcileTempUploads(ctx context.Context, cacheFs *Fs) error {
|
func (b *Persistent) ReconcileTempUploads(ctx context.Context, cacheFs *Fs) error {
|
||||||
return b.db.Update(func(tx *bolt.Tx) error {
|
return b.db.Update(func(tx *bolt.Tx) error {
|
||||||
|
@ -1036,19 +1027,6 @@ func (b *Persistent) ReconcileTempUploads(ctx context.Context, cacheFs *Fs) erro
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// PurgeTempUploads will remove all the pending uploads from the queue
|
|
||||||
// TO BE USED IN TESTING ONLY
|
|
||||||
func (b *Persistent) PurgeTempUploads() {
|
|
||||||
b.tempQueueMux.Lock()
|
|
||||||
defer b.tempQueueMux.Unlock()
|
|
||||||
|
|
||||||
_ = b.db.Update(func(tx *bolt.Tx) error {
|
|
||||||
_ = tx.DeleteBucket([]byte(tempBucket))
|
|
||||||
_, _ = tx.CreateBucketIfNotExists([]byte(tempBucket))
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close should be called when the program ends gracefully
|
// Close should be called when the program ends gracefully
|
||||||
func (b *Persistent) Close() {
|
func (b *Persistent) Close() {
|
||||||
b.cleanupMux.Lock()
|
b.cleanupMux.Lock()
|
||||||
|
|
|
@ -0,0 +1,23 @@
|
||||||
|
package cache
|
||||||
|
|
||||||
|
import bolt "go.etcd.io/bbolt"
|
||||||
|
|
||||||
|
// PurgeTempUploads will remove all the pending uploads from the queue
|
||||||
|
func (b *Persistent) PurgeTempUploads() {
|
||||||
|
b.tempQueueMux.Lock()
|
||||||
|
defer b.tempQueueMux.Unlock()
|
||||||
|
|
||||||
|
_ = b.db.Update(func(tx *bolt.Tx) error {
|
||||||
|
_ = tx.DeleteBucket([]byte(tempBucket))
|
||||||
|
_, _ = tx.CreateBucketIfNotExists([]byte(tempBucket))
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetPendingUploadToStarted is a way to mark an entry as started (even if it's not already)
|
||||||
|
func (b *Persistent) SetPendingUploadToStarted(remote string) error {
|
||||||
|
return b.updatePendingUpload(remote, func(item *tempUploadInfo) error {
|
||||||
|
item.Started = true
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
Loading…
Reference in New Issue