2017-12-09 22:54:26 +01:00
// +build !plan9,go1.7
2017-11-12 18:54:25 +01:00
package cache
import (
"fmt"
"io"
"path"
2017-11-20 15:38:28 +01:00
"path/filepath"
2017-11-12 18:54:25 +01:00
"strings"
"sync"
"time"
"os"
2017-11-22 17:32:36 +01:00
"os/signal"
"syscall"
2018-01-11 17:05:41 +01:00
"github.com/ncw/rclone/backend/crypt"
2017-11-12 18:54:25 +01:00
"github.com/ncw/rclone/fs"
2018-01-12 17:30:54 +01:00
"github.com/ncw/rclone/fs/config"
"github.com/ncw/rclone/fs/config/flags"
2018-01-18 21:19:55 +01:00
"github.com/ncw/rclone/fs/config/obscure"
2018-01-12 17:30:54 +01:00
"github.com/ncw/rclone/fs/hash"
"github.com/ncw/rclone/fs/walk"
2018-01-30 21:35:53 +01:00
"github.com/ncw/rclone/lib/atexit"
2017-11-12 18:54:25 +01:00
"github.com/pkg/errors"
"golang.org/x/net/context"
"golang.org/x/time/rate"
)
const (
// DefCacheChunkSize is the default value for chunk size
DefCacheChunkSize = "5M"
2017-12-09 22:54:26 +01:00
// DefCacheTotalChunkSize is the default value for the maximum size of stored chunks
DefCacheTotalChunkSize = "10G"
// DefCacheChunkCleanInterval is the interval at which chunks are cleaned
DefCacheChunkCleanInterval = "1m"
2017-11-12 18:54:25 +01:00
// DefCacheInfoAge is the default value for object info age
DefCacheInfoAge = "6h"
// DefCacheReadRetries is the default value for read retries
2017-12-09 22:54:26 +01:00
DefCacheReadRetries = 10
2017-11-12 18:54:25 +01:00
// DefCacheTotalWorkers is how many workers run in parallel to download chunks
DefCacheTotalWorkers = 4
// DefCacheChunkNoMemory will enable or disable in-memory storage for chunks
DefCacheChunkNoMemory = false
// DefCacheRps limits the number of requests per second to the source FS
DefCacheRps = - 1
// DefCacheWrites will cache file data on writes through the cache
DefCacheWrites = false
2018-01-29 23:05:04 +01:00
// DefCacheTmpWaitTime says how long should files be stored in local cache before being uploaded
DefCacheTmpWaitTime = "15m"
2017-11-12 18:54:25 +01:00
)
// Globals
var (
// Flags
2018-01-12 17:30:54 +01:00
cacheDbPath = flags . StringP ( "cache-db-path" , "" , filepath . Join ( config . CacheDir , "cache-backend" ) , "Directory to cache DB" )
cacheChunkPath = flags . StringP ( "cache-chunk-path" , "" , filepath . Join ( config . CacheDir , "cache-backend" ) , "Directory to cached chunk files" )
cacheDbPurge = flags . BoolP ( "cache-db-purge" , "" , false , "Purge the cache DB before" )
cacheChunkSize = flags . StringP ( "cache-chunk-size" , "" , DefCacheChunkSize , "The size of a chunk" )
cacheTotalChunkSize = flags . StringP ( "cache-total-chunk-size" , "" , DefCacheTotalChunkSize , "The total size which the chunks can take up from the disk" )
cacheChunkCleanInterval = flags . StringP ( "cache-chunk-clean-interval" , "" , DefCacheChunkCleanInterval , "Interval at which chunk cleanup runs" )
cacheInfoAge = flags . StringP ( "cache-info-age" , "" , DefCacheInfoAge , "How much time should object info be stored in cache" )
cacheReadRetries = flags . IntP ( "cache-read-retries" , "" , DefCacheReadRetries , "How many times to retry a read from a cache storage" )
cacheTotalWorkers = flags . IntP ( "cache-workers" , "" , DefCacheTotalWorkers , "How many workers should run in parallel to download chunks" )
cacheChunkNoMemory = flags . BoolP ( "cache-chunk-no-memory" , "" , DefCacheChunkNoMemory , "Disable the in-memory cache for storing chunks during streaming" )
cacheRps = flags . IntP ( "cache-rps" , "" , int ( DefCacheRps ) , "Limits the number of requests per second to the source FS. -1 disables the rate limiter" )
cacheStoreWrites = flags . BoolP ( "cache-writes" , "" , DefCacheWrites , "Will cache file data on writes through the FS" )
2018-01-29 23:05:04 +01:00
cacheTempWritePath = flags . StringP ( "cache-tmp-upload-path" , "" , "" , "Directory to keep temporary files until they are uploaded to the cloud storage" )
cacheTempWaitTime = flags . StringP ( "cache-tmp-wait-time" , "" , DefCacheTmpWaitTime , "How long should files be stored in local cache before being uploaded" )
2017-11-12 18:54:25 +01:00
)
// Register with Fs
func init ( ) {
fs . Register ( & fs . RegInfo {
Name : "cache" ,
Description : "Cache a remote" ,
NewFs : NewFs ,
Options : [ ] fs . Option { {
Name : "remote" ,
Help : "Remote to cache.\nNormally should contain a ':' and a path, eg \"myremote:path/to/dir\",\n\"myremote:bucket\" or maybe \"myremote:\" (not recommended)." ,
2017-12-09 22:54:26 +01:00
} , {
Name : "plex_url" ,
Help : "Optional: The URL of the Plex server" ,
Optional : true ,
} , {
Name : "plex_username" ,
Help : "Optional: The username of the Plex user" ,
Optional : true ,
} , {
Name : "plex_password" ,
Help : "Optional: The password of the Plex user" ,
IsPassword : true ,
Optional : true ,
2017-11-12 18:54:25 +01:00
} , {
Name : "chunk_size" ,
Help : "The size of a chunk. Lower value good for slow connections but can affect seamless reading. \nDefault: " + DefCacheChunkSize ,
Examples : [ ] fs . OptionExample {
{
Value : "1m" ,
Help : "1MB" ,
} , {
Value : "5M" ,
Help : "5 MB" ,
} , {
Value : "10M" ,
Help : "10 MB" ,
} ,
} ,
Optional : true ,
} , {
Name : "info_age" ,
Help : "How much time should object info (file size, file hashes etc) be stored in cache. Use a very high value if you don't plan on changing the source FS from outside the cache. \nAccepted units are: \"s\", \"m\", \"h\".\nDefault: " + DefCacheInfoAge ,
Examples : [ ] fs . OptionExample {
{
Value : "1h" ,
Help : "1 hour" ,
} , {
Value : "24h" ,
Help : "24 hours" ,
} , {
Value : "48h" ,
2017-11-21 10:25:28 +01:00
Help : "48 hours" ,
2017-11-12 18:54:25 +01:00
} ,
} ,
Optional : true ,
} , {
2017-12-09 22:54:26 +01:00
Name : "chunk_total_size" ,
Help : "The maximum size of stored chunks. When the storage grows beyond this size, the oldest chunks will be deleted. \nDefault: " + DefCacheTotalChunkSize ,
2017-11-12 18:54:25 +01:00
Examples : [ ] fs . OptionExample {
{
2017-12-09 22:54:26 +01:00
Value : "500M" ,
Help : "500 MB" ,
2017-11-12 18:54:25 +01:00
} , {
2017-12-09 22:54:26 +01:00
Value : "1G" ,
Help : "1 GB" ,
2017-11-12 18:54:25 +01:00
} , {
2017-12-09 22:54:26 +01:00
Value : "10G" ,
Help : "10 GB" ,
2017-11-12 18:54:25 +01:00
} ,
} ,
Optional : true ,
} } ,
} )
}
// Fs represents a wrapped fs.Fs
type Fs struct {
fs . Fs
2017-12-06 16:14:34 +01:00
wrapper fs . Fs
2017-11-12 18:54:25 +01:00
name string
root string
features * fs . Features // optional features
2018-01-29 23:05:04 +01:00
cache * Persistent
2017-11-12 18:54:25 +01:00
2017-12-09 22:54:26 +01:00
fileAge time . Duration
chunkSize int64
chunkTotalSize int64
chunkCleanInterval time . Duration
readRetries int
totalWorkers int
totalMaxWorkers int
chunkMemory bool
cacheWrites bool
2018-01-29 23:05:04 +01:00
tempWritePath string
tempWriteWait time . Duration
tempFs fs . Fs
2017-12-09 22:54:26 +01:00
lastChunkCleanup time . Time
cleanupMu sync . Mutex
rateLimiter * rate . Limiter
plexConnector * plexConnector
2018-01-29 23:05:04 +01:00
backgroundRunner * backgroundWriter
cleanupChan chan bool
2018-02-10 21:01:05 +01:00
parentsForgetFn [ ] func ( string )
2017-11-12 18:54:25 +01:00
}
2018-01-30 14:35:40 +01:00
// parseRootPath returns a cleaned root path and a nil error or "" and an error when the path is invalid
func parseRootPath ( path string ) ( string , error ) {
return strings . Trim ( path , "/" ) , nil
}
2018-01-29 23:05:04 +01:00
// NewFs constructs a Fs from the path, container:path
2018-01-30 14:35:40 +01:00
func NewFs ( name , rootPath string ) ( fs . Fs , error ) {
2018-01-12 17:30:54 +01:00
remote := config . FileGet ( name , "remote" )
2017-11-12 18:54:25 +01:00
if strings . HasPrefix ( remote , name + ":" ) {
return nil , errors . New ( "can't point cache remote at itself - check the value of the remote setting" )
}
2018-01-30 14:35:40 +01:00
rpath , err := parseRootPath ( rootPath )
if err != nil {
return nil , errors . Wrapf ( err , "failed to clean root path %q" , rootPath )
}
2017-11-12 18:54:25 +01:00
remotePath := path . Join ( remote , rpath )
wrappedFs , wrapErr := fs . NewFs ( remotePath )
2018-01-29 23:05:04 +01:00
if wrapErr != nil && wrapErr != fs . ErrorIsFile {
2017-11-12 18:54:25 +01:00
return nil , errors . Wrapf ( wrapErr , "failed to make remote %q to wrap" , remotePath )
}
2018-01-29 23:05:04 +01:00
var fsErr error
2017-11-12 18:54:25 +01:00
fs . Debugf ( name , "wrapped %v:%v at root %v" , wrappedFs . Name ( ) , wrappedFs . Root ( ) , rpath )
2018-01-29 23:05:04 +01:00
if wrapErr == fs . ErrorIsFile {
fsErr = fs . ErrorIsFile
rpath = cleanPath ( path . Dir ( rpath ) )
}
2018-01-12 17:30:54 +01:00
plexURL := config . FileGet ( name , "plex_url" )
plexToken := config . FileGet ( name , "plex_token" )
2017-11-12 18:54:25 +01:00
var chunkSize fs . SizeSuffix
2018-01-12 17:30:54 +01:00
chunkSizeString := config . FileGet ( name , "chunk_size" , DefCacheChunkSize )
2017-11-12 18:54:25 +01:00
if * cacheChunkSize != DefCacheChunkSize {
chunkSizeString = * cacheChunkSize
}
2018-01-30 14:35:40 +01:00
err = chunkSize . Set ( chunkSizeString )
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
return nil , errors . Wrapf ( err , "failed to understand chunk size %v" , chunkSizeString )
2017-11-12 18:54:25 +01:00
}
2017-12-09 22:54:26 +01:00
var chunkTotalSize fs . SizeSuffix
2018-01-12 17:30:54 +01:00
chunkTotalSizeString := config . FileGet ( name , "chunk_total_size" , DefCacheTotalChunkSize )
2017-12-09 22:54:26 +01:00
if * cacheTotalChunkSize != DefCacheTotalChunkSize {
chunkTotalSizeString = * cacheTotalChunkSize
2017-11-12 18:54:25 +01:00
}
2017-12-09 22:54:26 +01:00
err = chunkTotalSize . Set ( chunkTotalSizeString )
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
return nil , errors . Wrapf ( err , "failed to understand chunk total size %v" , chunkTotalSizeString )
2017-11-12 18:54:25 +01:00
}
2017-12-09 22:54:26 +01:00
chunkCleanIntervalStr := * cacheChunkCleanInterval
chunkCleanInterval , err := time . ParseDuration ( chunkCleanIntervalStr )
2017-11-12 18:54:25 +01:00
if err != nil {
2017-12-09 22:54:26 +01:00
return nil , errors . Wrapf ( err , "failed to understand duration %v" , chunkCleanIntervalStr )
2017-11-12 18:54:25 +01:00
}
2018-01-12 17:30:54 +01:00
infoAge := config . FileGet ( name , "info_age" , DefCacheInfoAge )
2017-12-09 22:54:26 +01:00
if * cacheInfoAge != DefCacheInfoAge {
infoAge = * cacheInfoAge
2017-11-12 18:54:25 +01:00
}
2017-12-09 22:54:26 +01:00
infoDuration , err := time . ParseDuration ( infoAge )
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
return nil , errors . Wrapf ( err , "failed to understand duration %v" , infoAge )
}
waitTime , err := time . ParseDuration ( * cacheTempWaitTime )
if err != nil {
return nil , errors . Wrapf ( err , "failed to understand duration %v" , * cacheTempWaitTime )
2017-11-12 18:54:25 +01:00
}
// configure cache backend
if * cacheDbPurge {
fs . Debugf ( name , "Purging the DB" )
}
f := & Fs {
2017-12-09 22:54:26 +01:00
Fs : wrappedFs ,
name : name ,
root : rpath ,
fileAge : infoDuration ,
chunkSize : int64 ( chunkSize ) ,
chunkTotalSize : int64 ( chunkTotalSize ) ,
chunkCleanInterval : chunkCleanInterval ,
readRetries : * cacheReadRetries ,
totalWorkers : * cacheTotalWorkers ,
totalMaxWorkers : * cacheTotalWorkers ,
chunkMemory : ! * cacheChunkNoMemory ,
cacheWrites : * cacheStoreWrites ,
lastChunkCleanup : time . Now ( ) . Truncate ( time . Hour * 24 * 30 ) ,
2018-01-29 23:05:04 +01:00
tempWritePath : * cacheTempWritePath ,
tempWriteWait : waitTime ,
cleanupChan : make ( chan bool , 1 ) ,
2017-12-09 22:54:26 +01:00
}
if f . chunkTotalSize < ( f . chunkSize * int64 ( f . totalWorkers ) ) {
return nil , errors . Errorf ( "don't set cache-total-chunk-size(%v) less than cache-chunk-size(%v) * cache-workers(%v)" ,
f . chunkTotalSize , f . chunkSize , f . totalWorkers )
2017-11-12 18:54:25 +01:00
}
f . rateLimiter = rate . NewLimiter ( rate . Limit ( float64 ( * cacheRps ) ) , f . totalWorkers )
2017-12-09 22:54:26 +01:00
f . plexConnector = & plexConnector { }
if plexURL != "" {
if plexToken != "" {
f . plexConnector , err = newPlexConnectorWithToken ( f , plexURL , plexToken )
if err != nil {
return nil , errors . Wrapf ( err , "failed to connect to the Plex API %v" , plexURL )
}
} else {
2018-01-12 17:30:54 +01:00
plexUsername := config . FileGet ( name , "plex_username" )
plexPassword := config . FileGet ( name , "plex_password" )
2017-12-09 22:54:26 +01:00
if plexPassword != "" && plexUsername != "" {
2018-01-18 21:19:55 +01:00
decPass , err := obscure . Reveal ( plexPassword )
2017-12-09 22:54:26 +01:00
if err != nil {
decPass = plexPassword
}
f . plexConnector , err = newPlexConnector ( f , plexURL , plexUsername , decPass )
if err != nil {
return nil , errors . Wrapf ( err , "failed to connect to the Plex API %v" , plexURL )
}
}
}
}
2017-11-12 18:54:25 +01:00
dbPath := * cacheDbPath
2017-12-20 21:43:30 +01:00
chunkPath := * cacheChunkPath
// if the dbPath is non default but the chunk path is default, we overwrite the last to follow the same one as dbPath
2018-01-12 17:30:54 +01:00
if dbPath != filepath . Join ( config . CacheDir , "cache-backend" ) &&
chunkPath == filepath . Join ( config . CacheDir , "cache-backend" ) {
2017-12-20 21:43:30 +01:00
chunkPath = dbPath
}
2017-11-20 15:38:28 +01:00
if filepath . Ext ( dbPath ) != "" {
dbPath = filepath . Dir ( dbPath )
2017-11-12 18:54:25 +01:00
}
2017-12-20 21:43:30 +01:00
if filepath . Ext ( chunkPath ) != "" {
chunkPath = filepath . Dir ( chunkPath )
}
2017-11-12 18:54:25 +01:00
err = os . MkdirAll ( dbPath , os . ModePerm )
if err != nil {
return nil , errors . Wrapf ( err , "failed to create cache directory %v" , dbPath )
}
2017-12-20 21:43:30 +01:00
err = os . MkdirAll ( chunkPath , os . ModePerm )
if err != nil {
return nil , errors . Wrapf ( err , "failed to create cache directory %v" , chunkPath )
}
2017-11-12 18:54:25 +01:00
2017-11-20 15:38:28 +01:00
dbPath = filepath . Join ( dbPath , name + ".db" )
2017-12-20 21:43:30 +01:00
chunkPath = filepath . Join ( chunkPath , name )
fs . Infof ( name , "Cache DB path: %v" , dbPath )
fs . Infof ( name , "Cache chunk path: %v" , chunkPath )
f . cache , err = GetPersistent ( dbPath , chunkPath , & Features {
2017-12-09 22:54:26 +01:00
PurgeDb : * cacheDbPurge ,
} )
2017-11-12 18:54:25 +01:00
if err != nil {
2017-11-30 11:27:59 +01:00
return nil , errors . Wrapf ( err , "failed to start cache db" )
2017-11-12 18:54:25 +01:00
}
2017-11-22 17:32:36 +01:00
// Trap SIGINT and SIGTERM to close the DB handle gracefully
c := make ( chan os . Signal , 1 )
2018-01-30 21:35:53 +01:00
signal . Notify ( c , syscall . SIGHUP )
atexit . Register ( func ( ) {
f . StopBackgroundRunners ( )
} )
2017-11-22 17:32:36 +01:00
go func ( ) {
2017-12-19 14:48:48 +01:00
for {
s := <- c
2018-01-30 21:35:53 +01:00
if s == syscall . SIGHUP {
2017-12-19 14:48:48 +01:00
fs . Infof ( f , "Clearing cache from signal" )
f . DirCacheFlush ( )
}
2017-11-22 17:32:36 +01:00
}
} ( )
2017-11-12 18:54:25 +01:00
fs . Infof ( name , "Chunk Memory: %v" , f . chunkMemory )
fs . Infof ( name , "Chunk Size: %v" , fs . SizeSuffix ( f . chunkSize ) )
2017-12-09 22:54:26 +01:00
fs . Infof ( name , "Chunk Total Size: %v" , fs . SizeSuffix ( f . chunkTotalSize ) )
fs . Infof ( name , "Chunk Clean Interval: %v" , f . chunkCleanInterval . String ( ) )
2017-11-12 18:54:25 +01:00
fs . Infof ( name , "Workers: %v" , f . totalWorkers )
fs . Infof ( name , "File Age: %v" , f . fileAge . String ( ) )
2018-01-29 23:05:04 +01:00
if f . cacheWrites {
fs . Infof ( name , "Cache Writes: enabled" )
}
2017-11-12 18:54:25 +01:00
2018-01-29 23:05:04 +01:00
if f . tempWritePath != "" {
err = os . MkdirAll ( f . tempWritePath , os . ModePerm )
if err != nil {
return nil , errors . Wrapf ( err , "failed to create cache directory %v" , f . tempWritePath )
}
f . tempWritePath = filepath . ToSlash ( f . tempWritePath )
f . tempFs , err = fs . NewFs ( f . tempWritePath )
if err != nil {
return nil , errors . Wrapf ( err , "failed to create temp fs: %v" , err )
}
fs . Infof ( name , "Upload Temp Rest Time: %v" , f . tempWriteWait . String ( ) )
fs . Infof ( name , "Upload Temp FS: %v" , f . tempWritePath )
f . backgroundRunner , _ = initBackgroundUploader ( f )
go f . backgroundRunner . run ( )
}
go func ( ) {
for {
time . Sleep ( f . chunkCleanInterval )
select {
case <- f . cleanupChan :
fs . Infof ( f , "stopping cleanup" )
return
default :
fs . Debugf ( f , "starting cleanup" )
f . CleanUpCache ( false )
}
}
} ( )
2017-11-12 18:54:25 +01:00
2018-02-10 21:01:05 +01:00
if doDirChangeNotify := wrappedFs . Features ( ) . DirChangeNotify ; doDirChangeNotify != nil {
doDirChangeNotify ( f . receiveDirChangeNotify , f . chunkCleanInterval )
}
2017-11-12 18:54:25 +01:00
f . features = ( & fs . Features {
CanHaveEmptyDirectories : true ,
DuplicateFiles : false , // storage doesn't permit this
2017-12-06 16:14:34 +01:00
} ) . Fill ( f ) . Mask ( wrappedFs ) . WrapsFs ( f , wrappedFs )
2018-01-29 23:05:04 +01:00
// override only those features that use a temp fs and it doesn't support them
2018-02-10 21:01:05 +01:00
f . features . DirChangeNotify = f . DirChangeNotify
2018-01-29 23:05:04 +01:00
if f . tempWritePath != "" {
if f . tempFs . Features ( ) . Copy == nil {
f . features . Copy = nil
}
if f . tempFs . Features ( ) . Move == nil {
f . features . Move = nil
}
if f . tempFs . Features ( ) . Move == nil {
f . features . Move = nil
}
if f . tempFs . Features ( ) . DirMove == nil {
f . features . DirMove = nil
}
if f . tempFs . Features ( ) . MergeDirs == nil {
f . features . MergeDirs = nil
}
}
// even if the wrapped fs doesn't support it, we still want it
2017-12-18 13:55:37 +01:00
f . features . DirCacheFlush = f . DirCacheFlush
2017-11-12 18:54:25 +01:00
2018-01-29 23:05:04 +01:00
return f , fsErr
2017-11-12 18:54:25 +01:00
}
2018-02-10 21:01:05 +01:00
func ( f * Fs ) receiveDirChangeNotify ( forgetPath string ) {
fs . Debugf ( f , "notify: expiring cache for '%v'" , forgetPath )
// notify upstreams too (vfs)
f . notifyDirChange ( forgetPath )
var cd * Directory
co := NewObject ( f , forgetPath )
err := f . cache . GetObject ( co )
if err == nil {
cd = NewDirectory ( f , cleanPath ( path . Dir ( co . Remote ( ) ) ) )
} else {
cd = NewDirectory ( f , forgetPath )
}
// we list all the cached objects and expire all of them
entries , err := f . cache . GetDirEntries ( cd )
if err != nil {
fs . Debugf ( forgetPath , "notify: ignoring notification on non cached dir" )
return
}
for i := 0 ; i < len ( entries ) ; i ++ {
if co , ok := entries [ i ] . ( * Object ) ; ok {
co . CacheTs = time . Now ( ) . Add ( f . fileAge * - 1 )
err = f . cache . AddObject ( co )
if err != nil {
fs . Errorf ( forgetPath , "notify: error expiring '%v': %v" , co , err )
} else {
fs . Debugf ( forgetPath , "notify: expired %v" , co )
}
}
}
// finally, we expire the dir as well
err = f . cache . ExpireDir ( cd )
if err != nil {
fs . Errorf ( forgetPath , "notify: error expiring '%v': %v" , cd , err )
} else {
fs . Debugf ( forgetPath , "notify: expired '%v'" , cd )
}
}
// notifyDirChange takes a remote (can be dir or entry) and
// tries to determine which is it and notify upstreams of the dir change
func ( f * Fs ) notifyDirChange ( remote string ) {
var cd * Directory
co := NewObject ( f , remote )
err := f . cache . GetObject ( co )
if err == nil {
pd := cleanPath ( path . Dir ( remote ) )
cd = NewDirectory ( f , pd )
} else {
cd = NewDirectory ( f , remote )
}
f . notifyDirChangeUpstream ( cd . Remote ( ) )
}
2018-02-11 21:30:58 +01:00
// notifyDirChangeUpstreamIfNeeded will check if the wrapped remote doesn't notify on dir changes
// or if we use a temp fs
func ( f * Fs ) notifyDirChangeUpstreamIfNeeded ( remote string ) {
if f . Fs . Features ( ) . DirChangeNotify == nil || f . tempWritePath != "" {
f . notifyDirChangeUpstream ( remote )
}
}
2018-02-10 21:01:05 +01:00
// notifyDirChangeUpstream will loop through all the upstreams and notify
// of the provided remote (should be only a dir)
func ( f * Fs ) notifyDirChangeUpstream ( remote string ) {
if len ( f . parentsForgetFn ) > 0 {
for _ , fn := range f . parentsForgetFn {
fn ( remote )
}
}
}
// DirChangeNotify can subsribe multiple callers
// this is coupled with the wrapped fs DirChangeNotify (if it supports it)
// and also notifies other caches (i.e VFS) to clear out whenever something changes
func ( f * Fs ) DirChangeNotify ( notifyFunc func ( string ) , pollInterval time . Duration ) chan bool {
fs . Debugf ( f , "subscribing to DirChangeNotify" )
f . parentsForgetFn = append ( f . parentsForgetFn , notifyFunc )
return make ( chan bool )
}
2017-11-12 18:54:25 +01:00
// Name of the remote (as passed into NewFs)
func ( f * Fs ) Name ( ) string {
return f . name
}
// Root of the remote (as passed into NewFs)
func ( f * Fs ) Root ( ) string {
return f . root
}
// Features returns the optional features of this Fs
func ( f * Fs ) Features ( ) * fs . Features {
return f . features
}
// String returns a description of the FS
func ( f * Fs ) String ( ) string {
2018-01-29 23:05:04 +01:00
return fmt . Sprintf ( "Cache remote %s:%s" , f . name , f . root )
2017-11-12 18:54:25 +01:00
}
// ChunkSize returns the configured chunk size
func ( f * Fs ) ChunkSize ( ) int64 {
return f . chunkSize
}
2018-01-29 23:05:04 +01:00
// InfoAge returns the configured file age
func ( f * Fs ) InfoAge ( ) time . Duration {
return f . fileAge
}
// TempUploadWaitTime returns the configured temp file upload wait time
func ( f * Fs ) TempUploadWaitTime ( ) time . Duration {
return f . tempWriteWait
}
2017-11-12 18:54:25 +01:00
// NewObject finds the Object at remote.
func ( f * Fs ) NewObject ( remote string ) ( fs . Object , error ) {
2018-01-29 23:05:04 +01:00
var err error
fs . Debugf ( f , "new object '%s'" , remote )
2017-11-12 18:54:25 +01:00
co := NewObject ( f , remote )
2018-01-29 23:05:04 +01:00
// search for entry in cache and validate it
err = f . cache . GetObject ( co )
2017-12-18 13:55:37 +01:00
if err != nil {
fs . Debugf ( remote , "find: error: %v" , err )
} else if time . Now ( ) . After ( co . CacheTs . Add ( f . fileAge ) ) {
2018-01-29 23:05:04 +01:00
fs . Debugf ( co , "find: cold object: %+v" , co )
2017-12-18 13:55:37 +01:00
} else {
2018-01-29 23:05:04 +01:00
fs . Debugf ( co , "find: warm object: %v, expiring on: %v" , co , co . CacheTs . Add ( f . fileAge ) )
2017-11-12 18:54:25 +01:00
return co , nil
}
2018-01-29 23:05:04 +01:00
// search for entry in source or temp fs
var obj fs . Object
err = nil
if f . tempWritePath != "" {
obj , err = f . tempFs . NewObject ( remote )
// not found in temp fs
if err != nil {
fs . Debugf ( remote , "find: not found in local cache fs" )
obj , err = f . Fs . NewObject ( remote )
} else {
fs . Debugf ( obj , "find: found in local cache fs" )
}
} else {
obj , err = f . Fs . NewObject ( remote )
}
// not found in either fs
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Debugf ( obj , "find failed: not found in either local or remote fs" )
2017-11-12 18:54:25 +01:00
return nil , err
}
2018-01-29 23:05:04 +01:00
// cache the new entry
co = ObjectFromOriginal ( f , obj ) . persist ( )
fs . Debugf ( co , "find: cached object" )
2017-11-12 18:54:25 +01:00
return co , nil
}
// List the objects and directories in dir into entries
func ( f * Fs ) List ( dir string ) ( entries fs . DirEntries , err error ) {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "list '%s'" , dir )
2017-12-18 13:55:37 +01:00
cd := ShallowDirectory ( f , dir )
2018-01-29 23:05:04 +01:00
// search for cached dir entries and validate them
2017-11-12 18:54:25 +01:00
entries , err = f . cache . GetDirEntries ( cd )
if err != nil {
2017-12-18 13:55:37 +01:00
fs . Debugf ( dir , "list: error: %v" , err )
} else if time . Now ( ) . After ( cd . CacheTs . Add ( f . fileAge ) ) {
fs . Debugf ( dir , "list: cold listing: %v" , cd . CacheTs )
2017-11-12 18:54:25 +01:00
} else if len ( entries ) == 0 {
// TODO: read empty dirs from source?
2017-12-18 13:55:37 +01:00
fs . Debugf ( dir , "list: empty listing" )
2017-11-12 18:54:25 +01:00
} else {
2018-01-29 23:05:04 +01:00
fs . Debugf ( dir , "list: warm %v from cache for: %v, expiring on: %v" , len ( entries ) , cd . abs ( ) , cd . CacheTs . Add ( f . fileAge ) )
fs . Debugf ( dir , "list: cached entries: %v" , entries )
2017-11-12 18:54:25 +01:00
return entries , nil
}
2018-01-29 23:05:04 +01:00
// FIXME need to clean existing cached listing
2017-11-12 18:54:25 +01:00
2018-01-29 23:05:04 +01:00
// we first search any temporary files stored locally
var cachedEntries fs . DirEntries
if f . tempWritePath != "" {
queuedEntries , err := f . cache . searchPendingUploadFromDir ( cd . abs ( ) )
if err != nil {
fs . Errorf ( dir , "list: error getting pending uploads: %v" , err )
} else {
fs . Debugf ( dir , "list: read %v from temp fs" , len ( queuedEntries ) )
fs . Debugf ( dir , "list: temp fs entries: %v" , queuedEntries )
for _ , queuedRemote := range queuedEntries {
queuedEntry , err := f . tempFs . NewObject ( f . cleanRootFromPath ( queuedRemote ) )
if err != nil {
fs . Debugf ( dir , "list: temp file not found in local fs: %v" , err )
continue
}
co := ObjectFromOriginal ( f , queuedEntry ) . persist ( )
fs . Debugf ( co , "list: cached temp object" )
cachedEntries = append ( cachedEntries , co )
}
}
}
// search from the source
2017-11-12 18:54:25 +01:00
entries , err = f . Fs . List ( dir )
if err != nil {
return nil , err
}
2017-12-18 13:55:37 +01:00
fs . Debugf ( dir , "list: read %v from source" , len ( entries ) )
2018-01-29 23:05:04 +01:00
fs . Debugf ( dir , "list: source entries: %v" , entries )
2017-11-12 18:54:25 +01:00
2018-01-29 23:05:04 +01:00
// and then iterate over the ones from source (temp Objects will override source ones)
2017-11-12 18:54:25 +01:00
for _ , entry := range entries {
switch o := entry . ( type ) {
case fs . Object :
2018-01-29 23:05:04 +01:00
// skip over temporary objects (might be uploading)
found := false
for _ , t := range cachedEntries {
if t . Remote ( ) == o . Remote ( ) {
found = true
break
}
}
if found {
continue
}
co := ObjectFromOriginal ( f , o ) . persist ( )
2017-11-12 18:54:25 +01:00
cachedEntries = append ( cachedEntries , co )
2018-01-29 23:05:04 +01:00
fs . Debugf ( dir , "list: cached object: %v" , co )
2017-11-12 18:54:25 +01:00
case fs . Directory :
2018-01-29 23:05:04 +01:00
cdd := DirectoryFromOriginal ( f , o )
err := f . cache . AddDir ( cdd )
if err != nil {
fs . Errorf ( dir , "list: error caching dir from listing %v" , o )
} else {
fs . Debugf ( dir , "list: cached dir: %v" , cdd )
}
cachedEntries = append ( cachedEntries , cdd )
2017-11-12 18:54:25 +01:00
default :
2018-01-29 23:05:04 +01:00
fs . Debugf ( entry , "list: Unknown object type %T" , entry )
2017-11-12 18:54:25 +01:00
}
}
2018-01-29 23:05:04 +01:00
// cache dir meta
t := time . Now ( )
cd . CacheTs = & t
err = f . cache . AddDir ( cd )
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( cd , "list: save error: '%v'" , err )
2017-12-18 13:55:37 +01:00
} else {
2018-01-29 23:05:04 +01:00
fs . Debugf ( dir , "list: cached dir: '%v', cache ts: %v" , cd . abs ( ) , cd . CacheTs )
2017-11-12 18:54:25 +01:00
}
return cachedEntries , nil
}
2018-01-12 17:30:54 +01:00
func ( f * Fs ) recurse ( dir string , list * walk . ListRHelper ) error {
2017-11-12 18:54:25 +01:00
entries , err := f . List ( dir )
if err != nil {
return err
}
for i := 0 ; i < len ( entries ) ; i ++ {
innerDir , ok := entries [ i ] . ( fs . Directory )
if ok {
err := f . recurse ( innerDir . Remote ( ) , list )
if err != nil {
return err
}
}
err := list . Add ( entries [ i ] )
if err != nil {
return err
}
}
return nil
}
// ListR lists the objects and directories of the Fs starting
// from dir recursively into out.
func ( f * Fs ) ListR ( dir string , callback fs . ListRCallback ) ( err error ) {
fs . Debugf ( f , "list recursively from '%s'" , dir )
// we check if the source FS supports ListR
// if it does, we'll use that to get all the entries, cache them and return
do := f . Fs . Features ( ) . ListR
if do != nil {
return do ( dir , func ( entries fs . DirEntries ) error {
// we got called back with a set of entries so let's cache them and call the original callback
for _ , entry := range entries {
switch o := entry . ( type ) {
case fs . Object :
_ = f . cache . AddObject ( ObjectFromOriginal ( f , o ) )
case fs . Directory :
_ = f . cache . AddDir ( DirectoryFromOriginal ( f , o ) )
default :
return errors . Errorf ( "Unknown object type %T" , entry )
}
}
// call the original callback
return callback ( entries )
} )
}
// if we're here, we're gonna do a standard recursive traversal and cache everything
2018-01-12 17:30:54 +01:00
list := walk . NewListRHelper ( callback )
2017-11-12 18:54:25 +01:00
err = f . recurse ( dir , list )
if err != nil {
return err
}
return list . Flush ( )
}
// Mkdir makes the directory (container, bucket)
func ( f * Fs ) Mkdir ( dir string ) error {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "mkdir '%s'" , dir )
2017-11-12 18:54:25 +01:00
err := f . Fs . Mkdir ( dir )
if err != nil {
return err
}
2018-01-29 23:05:04 +01:00
fs . Debugf ( dir , "mkdir: created dir in source fs" )
2017-11-12 18:54:25 +01:00
2017-12-18 13:55:37 +01:00
cd := NewDirectory ( f , cleanPath ( dir ) )
err = f . cache . AddDir ( cd )
if err != nil {
fs . Errorf ( dir , "mkdir: add error: %v" , err )
2018-01-29 23:05:04 +01:00
} else {
fs . Debugf ( cd , "mkdir: added to cache" )
2017-12-18 13:55:37 +01:00
}
2018-01-29 23:05:04 +01:00
// expire parent of new dir
2017-12-18 13:55:37 +01:00
parentCd := NewDirectory ( f , cleanPath ( path . Dir ( dir ) ) )
err = f . cache . ExpireDir ( parentCd )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( parentCd , "mkdir: cache expire error: %v" , err )
} else {
fs . Infof ( parentCd , "mkdir: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-11-12 18:54:25 +01:00
return nil
}
// Rmdir removes the directory (container, bucket) if empty
func ( f * Fs ) Rmdir ( dir string ) error {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "rmdir '%s'" , dir )
if f . tempWritePath != "" {
// pause background uploads
f . backgroundRunner . pause ( )
defer f . backgroundRunner . play ( )
// we check if the source exists on the remote and make the same move on it too if it does
// otherwise, we skip this step
_ , err := f . UnWrap ( ) . List ( dir )
if err == nil {
err := f . Fs . Rmdir ( dir )
if err != nil {
return err
}
fs . Debugf ( dir , "rmdir: removed dir in source fs" )
}
var queuedEntries [ ] * Object
err = walk . Walk ( f . tempFs , dir , true , - 1 , func ( path string , entries fs . DirEntries , err error ) error {
for _ , o := range entries {
if oo , ok := o . ( fs . Object ) ; ok {
co := ObjectFromOriginal ( f , oo )
queuedEntries = append ( queuedEntries , co )
}
}
return nil
} )
if err != nil {
fs . Errorf ( dir , "rmdir: error getting pending uploads: %v" , err )
} else {
fs . Debugf ( dir , "rmdir: read %v from temp fs" , len ( queuedEntries ) )
fs . Debugf ( dir , "rmdir: temp fs entries: %v" , queuedEntries )
if len ( queuedEntries ) > 0 {
fs . Errorf ( dir , "rmdir: temporary dir not empty" )
return fs . ErrorDirectoryNotEmpty
}
}
} else {
err := f . Fs . Rmdir ( dir )
if err != nil {
return err
}
fs . Debugf ( dir , "rmdir: removed dir in source fs" )
2017-11-12 18:54:25 +01:00
}
2017-12-18 13:55:37 +01:00
// remove dir data
d := NewDirectory ( f , dir )
2018-01-29 23:05:04 +01:00
err := f . cache . RemoveDir ( d . abs ( ) )
2017-12-18 13:55:37 +01:00
if err != nil {
fs . Errorf ( dir , "rmdir: remove error: %v" , err )
2018-01-29 23:05:04 +01:00
} else {
fs . Debugf ( d , "rmdir: removed from cache" )
2017-12-18 13:55:37 +01:00
}
// expire parent
parentCd := NewDirectory ( f , cleanPath ( path . Dir ( dir ) ) )
err = f . cache . ExpireDir ( parentCd )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( dir , "rmdir: cache expire error: %v" , err )
} else {
fs . Infof ( parentCd , "rmdir: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-11-12 18:54:25 +01:00
return nil
}
// DirMove moves src, srcRemote to this remote at dstRemote
// using server side move operations.
func ( f * Fs ) DirMove ( src fs . Fs , srcRemote , dstRemote string ) error {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "move dir '%s'/'%s' -> '%s'/'%s'" , src . Root ( ) , srcRemote , f . Root ( ) , dstRemote )
2017-11-12 18:54:25 +01:00
do := f . Fs . Features ( ) . DirMove
if do == nil {
return fs . ErrorCantDirMove
}
srcFs , ok := src . ( * Fs )
if ! ok {
fs . Errorf ( srcFs , "can't move directory - not same remote type" )
return fs . ErrorCantDirMove
}
if srcFs . Fs . Name ( ) != f . Fs . Name ( ) {
fs . Errorf ( srcFs , "can't move directory - not wrapping same remotes" )
return fs . ErrorCantDirMove
}
2018-01-29 23:05:04 +01:00
if f . tempWritePath != "" {
// pause background uploads
f . backgroundRunner . pause ( )
defer f . backgroundRunner . play ( )
// we check if the source exists on the remote and make the same move on it too if it does
// otherwise, we skip this step
_ , err := srcFs . UnWrap ( ) . List ( srcRemote )
if err == nil {
err := do ( srcFs . UnWrap ( ) , srcRemote , dstRemote )
if err != nil {
return err
}
fs . Debugf ( srcRemote , "movedir: dir moved in the source fs" )
}
var queuedEntries [ ] * Object
err = walk . Walk ( f . tempFs , srcRemote , true , - 1 , func ( path string , entries fs . DirEntries , err error ) error {
for _ , o := range entries {
if oo , ok := o . ( fs . Object ) ; ok {
co := ObjectFromOriginal ( f , oo )
queuedEntries = append ( queuedEntries , co )
if co . tempFileStartedUpload ( ) {
fs . Errorf ( co , "can't move - upload has already started. need to finish that" )
return fs . ErrorCantDirMove
}
}
}
return nil
} )
if err != nil {
return err
}
fs . Debugf ( srcRemote , "dirmove: read %v from temp fs" , len ( queuedEntries ) )
fs . Debugf ( srcRemote , "dirmove: temp fs entries: %v" , queuedEntries )
do := f . tempFs . Features ( ) . DirMove
if do == nil {
fs . Errorf ( srcRemote , "dirmove: can't move dir in temp fs" )
return fs . ErrorCantDirMove
}
err = do ( f . tempFs , srcRemote , dstRemote )
if err != nil {
return err
}
err = f . cache . ReconcileTempUploads ( f )
if err != nil {
return err
}
} else {
err := do ( srcFs . UnWrap ( ) , srcRemote , dstRemote )
if err != nil {
return err
}
fs . Debugf ( srcRemote , "movedir: dir moved in the source fs" )
2017-11-12 18:54:25 +01:00
}
2017-12-18 13:55:37 +01:00
// delete src dir from cache along with all chunks
2017-11-12 18:54:25 +01:00
srcDir := NewDirectory ( srcFs , srcRemote )
2018-01-29 23:05:04 +01:00
err := f . cache . RemoveDir ( srcDir . abs ( ) )
2017-12-18 13:55:37 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( srcDir , "dirmove: remove error: %v" , err )
} else {
fs . Debugf ( srcDir , "dirmove: removed cached dir" )
2017-12-18 13:55:37 +01:00
}
// expire src parent
srcParent := NewDirectory ( f , cleanPath ( path . Dir ( srcRemote ) ) )
err = f . cache . ExpireDir ( srcParent )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( srcParent , "dirmove: cache expire error: %v" , err )
} else {
fs . Debugf ( srcParent , "dirmove: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( srcParent . Remote ( ) )
2017-12-18 13:55:37 +01:00
// expire parent dir at the destination path
dstParent := NewDirectory ( f , cleanPath ( path . Dir ( dstRemote ) ) )
err = f . cache . ExpireDir ( dstParent )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( dstParent , "dirmove: cache expire error: %v" , err )
} else {
fs . Debugf ( dstParent , "dirmove: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( dstParent . Remote ( ) )
2017-12-18 13:55:37 +01:00
// TODO: precache dst dir and save the chunks
2017-11-12 18:54:25 +01:00
return nil
}
// cacheReader will split the stream of a reader to be cached at the same time it is read by the original source
func ( f * Fs ) cacheReader ( u io . Reader , src fs . ObjectInfo , originalRead func ( inn io . Reader ) ) {
// create the pipe and tee reader
pr , pw := io . Pipe ( )
tr := io . TeeReader ( u , pw )
// create channel to synchronize
done := make ( chan bool )
defer close ( done )
go func ( ) {
// notify the cache reader that we're complete after the source FS finishes
defer func ( ) {
_ = pw . Close ( )
} ( )
// process original reading
originalRead ( tr )
// signal complete
done <- true
} ( )
go func ( ) {
var offset int64
for {
chunk := make ( [ ] byte , f . chunkSize )
readSize , err := io . ReadFull ( pr , chunk )
// we ignore 3 failures which are ok:
// 1. EOF - original reading finished and we got a full buffer too
// 2. ErrUnexpectedEOF - original reading finished and partial buffer
// 3. ErrClosedPipe - source remote reader was closed (usually means it reached the end) and we need to stop too
// if we have a different error: we're going to error out the original reading too and stop this
if err != nil && err != io . EOF && err != io . ErrUnexpectedEOF && err != io . ErrClosedPipe {
fs . Errorf ( src , "error saving new data in cache. offset: %v, err: %v" , offset , err )
_ = pr . CloseWithError ( err )
break
}
// if we have some bytes we cache them
if readSize > 0 {
chunk = chunk [ : readSize ]
2017-12-09 22:54:26 +01:00
err2 := f . cache . AddChunk ( cleanPath ( path . Join ( f . root , src . Remote ( ) ) ) , chunk , offset )
2017-11-12 18:54:25 +01:00
if err2 != nil {
fs . Errorf ( src , "error saving new data in cache '%v'" , err2 )
_ = pr . CloseWithError ( err2 )
break
}
offset += int64 ( readSize )
}
// stuff should be closed but let's be sure
if err == io . EOF || err == io . ErrUnexpectedEOF || err == io . ErrClosedPipe {
_ = pr . Close ( )
break
}
}
// signal complete
done <- true
} ( )
// wait until both are done
for c := 0 ; c < 2 ; c ++ {
<- done
}
}
2017-11-30 20:16:45 +01:00
type putFn func ( in io . Reader , src fs . ObjectInfo , options ... fs . OpenOption ) ( fs . Object , error )
2017-11-12 18:54:25 +01:00
2017-11-30 20:16:45 +01:00
// put in to the remote path
func ( f * Fs ) put ( in io . Reader , src fs . ObjectInfo , options [ ] fs . OpenOption , put putFn ) ( fs . Object , error ) {
2017-11-12 18:54:25 +01:00
var err error
var obj fs . Object
2018-01-29 23:05:04 +01:00
// queue for upload and store in temp fs if configured
if f . tempWritePath != "" {
obj , err = f . tempFs . Put ( in , src , options ... )
if err != nil {
fs . Errorf ( obj , "put: failed to upload in temp fs: %v" , err )
return nil , err
}
fs . Infof ( obj , "put: uploaded in temp fs" )
err = f . cache . addPendingUpload ( path . Join ( f . Root ( ) , src . Remote ( ) ) , false )
if err != nil {
fs . Errorf ( obj , "put: failed to queue for upload: %v" , err )
return nil , err
}
fs . Infof ( obj , "put: queued for upload" )
// if cache writes is enabled write it first through cache
} else if f . cacheWrites {
2017-11-12 18:54:25 +01:00
f . cacheReader ( in , src , func ( inn io . Reader ) {
2017-11-30 20:16:45 +01:00
obj , err = put ( inn , src , options ... )
2017-11-12 18:54:25 +01:00
} )
2018-01-29 23:05:04 +01:00
if err == nil {
fs . Debugf ( obj , "put: uploaded to remote fs and saved in cache" )
}
// last option: save it directly in remote fs
2017-11-12 18:54:25 +01:00
} else {
2017-11-30 20:16:45 +01:00
obj , err = put ( in , src , options ... )
2018-01-29 23:05:04 +01:00
if err == nil {
fs . Debugf ( obj , "put: uploaded to remote fs" )
}
2017-11-12 18:54:25 +01:00
}
2018-01-29 23:05:04 +01:00
// validate and stop if errors are found
2017-11-12 18:54:25 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( src , "put: error uploading: %v" , err )
2017-11-12 18:54:25 +01:00
return nil , err
}
2018-01-29 23:05:04 +01:00
// cache the new file
2017-11-12 18:54:25 +01:00
cachedObj := ObjectFromOriginal ( f , obj ) . persist ( )
2018-01-29 23:05:04 +01:00
fs . Debugf ( cachedObj , "put: added to cache" )
2017-12-18 13:55:37 +01:00
// expire parent
2018-01-29 23:05:04 +01:00
parentCd := NewDirectory ( f , cleanPath ( path . Dir ( cachedObj . Remote ( ) ) ) )
err = f . cache . ExpireDir ( parentCd )
2017-12-18 13:55:37 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( cachedObj , "put: cache expire error: %v" , err )
} else {
fs . Infof ( parentCd , "put: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-11 21:30:58 +01:00
// advertise to DirChangeNotify
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-11-12 18:54:25 +01:00
return cachedObj , nil
}
2017-11-30 20:16:45 +01:00
// Put in to the remote path with the modTime given of the given size
func ( f * Fs ) Put ( in io . Reader , src fs . ObjectInfo , options ... fs . OpenOption ) ( fs . Object , error ) {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "put data at '%s'" , src . Remote ( ) )
2017-11-30 20:16:45 +01:00
return f . put ( in , src , options , f . Fs . Put )
}
2017-11-12 18:54:25 +01:00
// PutUnchecked uploads the object
func ( f * Fs ) PutUnchecked ( in io . Reader , src fs . ObjectInfo , options ... fs . OpenOption ) ( fs . Object , error ) {
do := f . Fs . Features ( ) . PutUnchecked
if do == nil {
return nil , errors . New ( "can't PutUnchecked" )
}
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "put data unchecked in '%s'" , src . Remote ( ) )
2017-11-30 20:16:45 +01:00
return f . put ( in , src , options , do )
}
2017-11-12 18:54:25 +01:00
2017-11-30 20:16:45 +01:00
// PutStream uploads the object
func ( f * Fs ) PutStream ( in io . Reader , src fs . ObjectInfo , options ... fs . OpenOption ) ( fs . Object , error ) {
do := f . Fs . Features ( ) . PutStream
if do == nil {
return nil , errors . New ( "can't PutStream" )
2017-11-12 18:54:25 +01:00
}
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "put data streaming in '%s'" , src . Remote ( ) )
2017-11-30 20:16:45 +01:00
return f . put ( in , src , options , do )
2017-11-12 18:54:25 +01:00
}
// Copy src to this remote using server side copy operations.
func ( f * Fs ) Copy ( src fs . Object , remote string ) ( fs . Object , error ) {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "copy obj '%s' -> '%s'" , src , remote )
2017-11-12 18:54:25 +01:00
do := f . Fs . Features ( ) . Copy
if do == nil {
2017-11-21 23:38:25 +01:00
fs . Errorf ( src , "source remote (%v) doesn't support Copy" , src . Fs ( ) )
2017-11-12 18:54:25 +01:00
return nil , fs . ErrorCantCopy
}
2018-01-29 23:05:04 +01:00
// the source must be a cached object or we abort
2017-11-12 18:54:25 +01:00
srcObj , ok := src . ( * Object )
if ! ok {
fs . Errorf ( srcObj , "can't copy - not same remote type" )
return nil , fs . ErrorCantCopy
}
2018-01-29 23:05:04 +01:00
// both the source cache fs and this cache fs need to wrap the same remote
2017-11-12 18:54:25 +01:00
if srcObj . CacheFs . Fs . Name ( ) != f . Fs . Name ( ) {
2018-01-29 23:05:04 +01:00
fs . Errorf ( srcObj , "can't copy - not wrapping same remotes" )
2017-11-12 18:54:25 +01:00
return nil , fs . ErrorCantCopy
}
2018-01-29 23:05:04 +01:00
// refresh from source or abort
if err := srcObj . refreshFromSource ( false ) ; err != nil {
fs . Errorf ( f , "can't copy %v - %v" , src , err )
2017-11-12 18:54:25 +01:00
return nil , fs . ErrorCantCopy
}
2018-01-29 23:05:04 +01:00
if srcObj . isTempFile ( ) {
// we check if the feature is stil active
if f . tempWritePath == "" {
fs . Errorf ( srcObj , "can't copy - this is a local cached file but this feature is turned off this run" )
return nil , fs . ErrorCantCopy
}
do = srcObj . ParentFs . Features ( ) . Copy
if do == nil {
fs . Errorf ( src , "parent remote (%v) doesn't support Copy" , srcObj . ParentFs )
return nil , fs . ErrorCantCopy
}
}
2017-11-12 18:54:25 +01:00
obj , err := do ( srcObj . Object , remote )
if err != nil {
fs . Errorf ( srcObj , "error moving in cache: %v" , err )
return nil , err
}
2018-01-29 23:05:04 +01:00
fs . Debugf ( obj , "copy: file copied" )
2017-11-12 18:54:25 +01:00
// persist new
2017-12-18 13:55:37 +01:00
co := ObjectFromOriginal ( f , obj ) . persist ( )
2018-01-29 23:05:04 +01:00
fs . Debugf ( co , "copy: added to cache" )
2017-12-18 13:55:37 +01:00
// expire the destination path
2018-01-29 23:05:04 +01:00
parentCd := NewDirectory ( f , cleanPath ( path . Dir ( co . Remote ( ) ) ) )
err = f . cache . ExpireDir ( parentCd )
2017-12-18 13:55:37 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( parentCd , "copy: cache expire error: %v" , err )
} else {
fs . Infof ( parentCd , "copy: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-12-18 13:55:37 +01:00
// expire src parent
srcParent := NewDirectory ( f , cleanPath ( path . Dir ( src . Remote ( ) ) ) )
err = f . cache . ExpireDir ( srcParent )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( srcParent , "copy: cache expire error: %v" , err )
} else {
fs . Infof ( srcParent , "copy: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( srcParent . Remote ( ) )
2017-11-12 18:54:25 +01:00
2017-12-18 13:55:37 +01:00
return co , nil
2017-11-12 18:54:25 +01:00
}
// Move src to this remote using server side move operations.
func ( f * Fs ) Move ( src fs . Object , remote string ) ( fs . Object , error ) {
2018-01-29 23:05:04 +01:00
fs . Debugf ( f , "moving obj '%s' -> %s" , src , remote )
// if source fs doesn't support move abort
2017-11-12 18:54:25 +01:00
do := f . Fs . Features ( ) . Move
if do == nil {
2017-11-21 23:38:25 +01:00
fs . Errorf ( src , "source remote (%v) doesn't support Move" , src . Fs ( ) )
2017-11-12 18:54:25 +01:00
return nil , fs . ErrorCantMove
}
2018-01-29 23:05:04 +01:00
// the source must be a cached object or we abort
2017-11-12 18:54:25 +01:00
srcObj , ok := src . ( * Object )
if ! ok {
fs . Errorf ( srcObj , "can't move - not same remote type" )
return nil , fs . ErrorCantMove
}
2018-01-29 23:05:04 +01:00
// both the source cache fs and this cache fs need to wrap the same remote
2017-11-12 18:54:25 +01:00
if srcObj . CacheFs . Fs . Name ( ) != f . Fs . Name ( ) {
fs . Errorf ( srcObj , "can't move - not wrapping same remote types" )
return nil , fs . ErrorCantMove
}
2018-01-29 23:05:04 +01:00
// refresh from source or abort
if err := srcObj . refreshFromSource ( false ) ; err != nil {
2017-11-12 18:54:25 +01:00
fs . Errorf ( f , "can't move %v - %v" , src , err )
return nil , fs . ErrorCantMove
}
2018-01-29 23:05:04 +01:00
// if this is a temp object then we perform the changes locally
if srcObj . isTempFile ( ) {
// we check if the feature is stil active
if f . tempWritePath == "" {
fs . Errorf ( srcObj , "can't move - this is a local cached file but this feature is turned off this run" )
return nil , fs . ErrorCantMove
}
// pause background uploads
f . backgroundRunner . pause ( )
defer f . backgroundRunner . play ( )
// started uploads can't be moved until they complete
if srcObj . tempFileStartedUpload ( ) {
fs . Errorf ( srcObj , "can't move - upload has already started. need to finish that" )
return nil , fs . ErrorCantMove
}
do = f . tempFs . Features ( ) . Move
// we must also update the pending queue
err := f . cache . updatePendingUpload ( srcObj . abs ( ) , func ( item * tempUploadInfo ) error {
item . DestPath = path . Join ( f . Root ( ) , remote )
item . AddedOn = time . Now ( )
return nil
} )
if err != nil {
fs . Errorf ( srcObj , "failed to rename queued file for upload: %v" , err )
return nil , fs . ErrorCantMove
}
fs . Debugf ( srcObj , "move: queued file moved to %v" , remote )
}
2017-11-12 18:54:25 +01:00
obj , err := do ( srcObj . Object , remote )
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( srcObj , "error moving: %v" , err )
2017-11-12 18:54:25 +01:00
return nil , err
}
2018-01-29 23:05:04 +01:00
fs . Debugf ( obj , "move: file moved" )
2017-11-12 18:54:25 +01:00
// remove old
2017-12-18 13:55:37 +01:00
err = f . cache . RemoveObject ( srcObj . abs ( ) )
if err != nil {
fs . Errorf ( srcObj , "move: remove error: %v" , err )
2018-01-29 23:05:04 +01:00
} else {
fs . Debugf ( srcObj , "move: removed from cache" )
2017-12-18 13:55:37 +01:00
}
// expire old parent
2018-01-29 23:05:04 +01:00
parentCd := NewDirectory ( f , cleanPath ( path . Dir ( srcObj . Remote ( ) ) ) )
err = f . cache . ExpireDir ( parentCd )
2017-12-18 13:55:37 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( parentCd , "move: parent cache expire error: %v" , err )
} else {
fs . Infof ( parentCd , "move: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-11-12 18:54:25 +01:00
// persist new
2017-12-18 13:55:37 +01:00
cachedObj := ObjectFromOriginal ( f , obj ) . persist ( )
2018-01-29 23:05:04 +01:00
fs . Debugf ( cachedObj , "move: added to cache" )
2017-12-18 13:55:37 +01:00
// expire new parent
2018-01-29 23:05:04 +01:00
parentCd = NewDirectory ( f , cleanPath ( path . Dir ( cachedObj . Remote ( ) ) ) )
err = f . cache . ExpireDir ( parentCd )
2017-12-18 13:55:37 +01:00
if err != nil {
2018-01-29 23:05:04 +01:00
fs . Errorf ( parentCd , "move: expire error: %v" , err )
} else {
fs . Infof ( parentCd , "move: cache expired" )
2017-12-18 13:55:37 +01:00
}
2018-02-10 21:01:05 +01:00
// advertise to DirChangeNotify if wrapped doesn't do that
2018-02-11 21:30:58 +01:00
f . notifyDirChangeUpstreamIfNeeded ( parentCd . Remote ( ) )
2017-11-12 18:54:25 +01:00
return cachedObj , nil
}
// Hashes returns the supported hash sets.
2018-01-12 17:30:54 +01:00
func ( f * Fs ) Hashes ( ) hash . Set {
2017-11-12 18:54:25 +01:00
return f . Fs . Hashes ( )
}
// Purge all files in the root and the root directory
func ( f * Fs ) Purge ( ) error {
fs . Infof ( f , "purging cache" )
f . cache . Purge ( )
do := f . Fs . Features ( ) . Purge
if do == nil {
return nil
}
err := do ( )
if err != nil {
return err
}
return nil
}
// CleanUp the trash in the Fs
func ( f * Fs ) CleanUp ( ) error {
f . CleanUpCache ( false )
do := f . Fs . Features ( ) . CleanUp
if do == nil {
return nil
}
return do ( )
}
// Stats returns stats about the cache storage
func ( f * Fs ) Stats ( ) ( map [ string ] map [ string ] interface { } , error ) {
return f . cache . Stats ( )
}
2018-01-29 23:05:04 +01:00
// openRateLimited will execute a closure under a rate limiter watch
func ( f * Fs ) openRateLimited ( fn func ( ) ( io . ReadCloser , error ) ) ( io . ReadCloser , error ) {
2017-11-12 18:54:25 +01:00
var err error
ctx , cancel := context . WithTimeout ( context . Background ( ) , time . Second * 10 )
defer cancel ( )
start := time . Now ( )
if err = f . rateLimiter . Wait ( ctx ) ; err != nil {
return nil , err
}
elapsed := time . Since ( start )
if elapsed > time . Second * 2 {
fs . Debugf ( f , "rate limited: %s" , elapsed )
}
return fn ( )
}
// CleanUpCache will cleanup only the cache data that is expired
func ( f * Fs ) CleanUpCache ( ignoreLastTs bool ) {
f . cleanupMu . Lock ( )
defer f . cleanupMu . Unlock ( )
2017-12-09 22:54:26 +01:00
if ignoreLastTs || time . Now ( ) . After ( f . lastChunkCleanup . Add ( f . chunkCleanInterval ) ) {
f . cache . CleanChunksBySize ( f . chunkTotalSize )
2017-11-12 18:54:25 +01:00
f . lastChunkCleanup = time . Now ( )
}
}
2018-01-29 23:05:04 +01:00
// StopBackgroundRunners will signall all the runners to stop their work
// can be triggered from a terminate signal or from testing between runs
func ( f * Fs ) StopBackgroundRunners ( ) {
f . cleanupChan <- false
2018-01-30 21:35:53 +01:00
if f . tempWritePath != "" && f . backgroundRunner != nil && f . backgroundRunner . isRunning ( ) {
2018-01-29 23:05:04 +01:00
f . backgroundRunner . close ( )
}
f . cache . Close ( )
fs . Debugf ( f , "Services stopped" )
}
2017-11-12 18:54:25 +01:00
// UnWrap returns the Fs that this Fs is wrapping
func ( f * Fs ) UnWrap ( ) fs . Fs {
return f . Fs
}
2017-12-06 16:14:34 +01:00
// WrapFs returns the Fs that is wrapping this Fs
func ( f * Fs ) WrapFs ( ) fs . Fs {
return f . wrapper
}
// SetWrapper sets the Fs that is wrapping this Fs
func ( f * Fs ) SetWrapper ( wrapper fs . Fs ) {
f . wrapper = wrapper
}
2018-01-29 23:05:04 +01:00
// isWrappedByCrypt checks if this is wrapped by a crypt remote
2017-12-09 22:54:26 +01:00
func ( f * Fs ) isWrappedByCrypt ( ) ( * crypt . Fs , bool ) {
if f . wrapper == nil {
return nil , false
}
c , ok := f . wrapper . ( * crypt . Fs )
return c , ok
}
2018-01-29 23:05:04 +01:00
// cleanRootFromPath trims the root of the current fs from a path
func ( f * Fs ) cleanRootFromPath ( p string ) string {
if f . Root ( ) != "" {
p = p [ len ( f . Root ( ) ) : ] // trim out root
if len ( p ) > 0 { // remove first separator
p = p [ 1 : ]
}
}
return p
}
func ( f * Fs ) isRootInPath ( p string ) bool {
if f . Root ( ) == "" {
return true
}
return strings . HasPrefix ( p , f . Root ( ) + "/" )
}
2017-11-12 18:54:25 +01:00
// DirCacheFlush flushes the dir cache
func ( f * Fs ) DirCacheFlush ( ) {
_ = f . cache . RemoveDir ( "" )
}
2018-01-29 23:05:04 +01:00
// GetBackgroundUploadChannel returns a channel that can be listened to for remote activities that happen
// in the background
func ( f * Fs ) GetBackgroundUploadChannel ( ) chan BackgroundUploadState {
if f . tempWritePath != "" {
return f . backgroundRunner . notifyCh
}
return nil
}
2017-11-12 18:54:25 +01:00
func cleanPath ( p string ) string {
p = path . Clean ( p )
if p == "." || p == "/" {
p = ""
}
return p
}
// Check the interfaces are satisfied
var (
2018-02-10 21:01:05 +01:00
_ fs . Fs = ( * Fs ) ( nil )
_ fs . Purger = ( * Fs ) ( nil )
_ fs . Copier = ( * Fs ) ( nil )
_ fs . Mover = ( * Fs ) ( nil )
_ fs . DirMover = ( * Fs ) ( nil )
_ fs . PutUncheckeder = ( * Fs ) ( nil )
_ fs . PutStreamer = ( * Fs ) ( nil )
_ fs . CleanUpper = ( * Fs ) ( nil )
_ fs . UnWrapper = ( * Fs ) ( nil )
_ fs . Wrapper = ( * Fs ) ( nil )
_ fs . ListRer = ( * Fs ) ( nil )
_ fs . DirChangeNotifier = ( * Fs ) ( nil )
2017-11-12 18:54:25 +01:00
)