]> Cypherpunks.ru repositories - gocheese.git/blob - fileutils.go
Update dependencies
[gocheese.git] / fileutils.go
1 /*
2 GoCheese -- Python private package repository and caching proxy
3 Copyright (C) 2019-2021 Sergey Matveev <stargrave@stargrave.org>
4
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 3 of the License.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
16 */
17
18 package main
19
20 import (
21         "os"
22         "path/filepath"
23         "strconv"
24         "time"
25 )
26
27 var NoSync = os.Getenv("GOCHEESE_NO_SYNC") == "1"
28
29 func TempFile(dir string) (*os.File, error) {
30         // Assume that probability of suffix collision is negligible
31         suffix := strconv.FormatInt(time.Now().UnixNano()+int64(os.Getpid()), 16)
32         name := filepath.Join(dir, "nncp"+suffix)
33         return os.OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_EXCL, os.FileMode(0666))
34 }
35
36 func DirSync(dirPath string) error {
37         if NoSync {
38                 return nil
39         }
40         fd, err := os.Open(dirPath)
41         if err != nil {
42                 return err
43         }
44         err = fd.Sync()
45         if err != nil {
46                 fd.Close()
47                 return err
48         }
49         return fd.Close()
50 }
51
52 func WriteFileSync(dirPath, filePath string, data []byte) error {
53         dst, err := TempFile(dirPath)
54         if err != nil {
55                 return err
56         }
57         if _, err = dst.Write(data); err != nil {
58                 os.Remove(dst.Name())
59                 dst.Close()
60                 return err
61         }
62         if !NoSync {
63                 if err = dst.Sync(); err != nil {
64                         os.Remove(dst.Name())
65                         dst.Close()
66                         return err
67                 }
68         }
69         dst.Close()
70         if err = os.Rename(dst.Name(), filePath); err != nil {
71                 return err
72         }
73         return DirSync(dirPath)
74 }