X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=fileutils.go;h=b37b99ce4c79aa01c2e56534c9a2887767e0e6a8;hb=60834a0713d5dcc6a9911511cb8618ce7358c824;hp=70d3f1123840718924d05f4817cdabdcf6034aa6;hpb=b036ee436eb9bd8889734232a22d3f24be5c9ee2;p=gocheese.git diff --git a/fileutils.go b/fileutils.go index 70d3f11..b37b99c 100644 --- a/fileutils.go +++ b/fileutils.go @@ -1,6 +1,6 @@ /* GoCheese -- Python private package repository and caching proxy -Copyright (C) 2019 Sergey Matveev +Copyright (C) 2019-2021 Sergey Matveev 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 @@ -18,12 +18,16 @@ along with this program. If not, see . package main import ( + "log" + "net/http" "os" "path/filepath" "strconv" "time" ) +var NoSync = os.Getenv("GOCHEESE_NO_SYNC") == "1" + func TempFile(dir string) (*os.File, error) { // Assume that probability of suffix collision is negligible suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16) @@ -32,6 +36,9 @@ func TempFile(dir string) (*os.File, error) { } func DirSync(dirPath string) error { + if NoSync { + return nil + } fd, err := os.Open(dirPath) if err != nil { return err @@ -44,7 +51,7 @@ func DirSync(dirPath string) error { return fd.Close() } -func WriteFileSync(dirPath, filePath string, data []byte) error { +func WriteFileSync(dirPath, filePath string, data []byte, mtime time.Time) error { dst, err := TempFile(dirPath) if err != nil { return err @@ -54,14 +61,32 @@ func WriteFileSync(dirPath, filePath string, data []byte) error { dst.Close() return err } - if err = dst.Sync(); err != nil { - os.Remove(dst.Name()) - dst.Close() - return err + if !NoSync { + if err = dst.Sync(); err != nil { + os.Remove(dst.Name()) + dst.Close() + return err + } } dst.Close() + if err = os.Chtimes(dst.Name(), mtime, mtime); err != nil { + return err + } if err = os.Rename(dst.Name(), filePath); err != nil { return err } return DirSync(dirPath) } + +func mkdirForPkg(w http.ResponseWriter, r *http.Request, pkgName string) bool { + path := filepath.Join(*Root, pkgName) + if _, err := os.Stat(path); os.IsNotExist(err) { + if err = os.Mkdir(path, os.FileMode(0777)); err != nil { + log.Println("error", r.RemoteAddr, "mkdir", pkgName, err) + http.Error(w, err.Error(), http.StatusInternalServerError) + return false + } + log.Println(r.RemoteAddr, "mkdir", pkgName) + } + return true +}