fenix/photocatalog
1
0
Fork 0
mirror of https://github.com/derfenix/photocatalog.git synced 2026-03-11 21:35:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Restore COW on copy for linux

Browse Source
This commit is contained in:
derFenix
2025-01-04 02:23:54 +03:00
parent e7c515c718
commit f73d612666
2 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
application/config.go
View File

@@ -33,7 +33,7 @@ func (c *Config) Validate() error {
return fmt.Errorf("target dir is required") return fmt.Errorf("target dir is required")
} }
if !slices.Contains([]Mode{ModeHardlink, ModeSymlink, ModeMove, ModeSymlink}, c.Mode) { if !slices.Contains([]Mode{ModeHardlink, ModeSymlink, ModeMove, ModeCopy}, c.Mode) {
return fmt.Errorf("invalid mode %s", c.Mode) return fmt.Errorf("invalid mode %s", c.Mode)
} }

19
internal/organizer/modes/copy.go
View File

@@ -5,12 +5,18 @@ import (
"io" "io"
"log" "log"
"os" "os"
"runtime"
"sync/atomic"
"golang.org/x/sys/unix"
) )
var cowDisabled = atomic.Bool{}
type Copy struct{} type Copy struct{}
func (c Copy) PlaceIt(sourcePath, targetPath string, mode os.FileMode) error { func (c Copy) PlaceIt(sourcePath, targetPath string, mode os.FileMode) error {
targetFile, err := os.OpenFile(targetPath, os.O_TRUNC|os.O_RDWR|os.O_CREATE, mode) targetFile, err := os.OpenFile(targetPath, os.O_TRUNC|os.O_WRONLY|os.O_CREATE, mode)
if err != nil { if err != nil {
return fmt.Errorf("open target file: %w", err) return fmt.Errorf("open target file: %w", err)
} }
@@ -28,6 +34,17 @@ func (c Copy) PlaceIt(sourcePath, targetPath string, mode os.FileMode) error {
_ = sourceFile.Close() _ = sourceFile.Close()
}() }()
// Try to do a COW.
if runtime.GOOS == "linux" && !cowDisabled.Load() {
if err := unix.IoctlFileClone(int(targetFile.Fd()+1), int(sourceFile.Fd())); err == nil {
return nil
} else {
log.Println(fmt.Errorf("COW attempt for %s failed: %w", targetPath, err))
log.Println("Disabling COW until restart")
cowDisabled.Store(true)
}
}
copySize, err := io.Copy(targetFile, sourceFile) copySize, err := io.Copy(targetFile, sourceFile)
if err != nil { if err != nil {
return fmt.Errorf("copy source file: %w", err) return fmt.Errorf("copy source file: %w", err)