aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/github.com/syndtr/goleveldb/leveldb/storage/file_storage_unix.go
blob: 7e29915379e12ff7a1bd08cd02ecfc53b8bbc0a7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
// Copyright (c) 2012, Suryandaru Triandana <syndtr@gmail.com>
// All rights reserved.
//
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// +build darwin dragonfly freebsd linux netbsd openbsd

package storage

import (
    "os"
    "syscall"
)

type unixFileLock struct {
    f *os.File
}

func (fl *unixFileLock) release() error {
    if err := setFileLock(fl.f, false, false); err != nil {
        return err
    }
    return fl.f.Close()
}

func newFileLock(path string, readOnly bool) (fl fileLock, err error) {
    var flag int
    if readOnly {
        flag = os.O_RDONLY
    } else {
        flag = os.O_RDWR
    }
    f, err := os.OpenFile(path, flag, 0)
    if os.IsNotExist(err) {
        f, err = os.OpenFile(path, flag|os.O_CREATE, 0644)
    }
    if err != nil {
        return
    }
    err = setFileLock(f, readOnly, true)
    if err != nil {
        f.Close()
        return
    }
    fl = &unixFileLock{f: f}
    return
}

func setFileLock(f *os.File, readOnly, lock bool) error {
    how := syscall.LOCK_UN
    if lock {
        if readOnly {
            how = syscall.LOCK_SH
        } else {
            how = syscall.LOCK_EX
        }
    }
    return syscall.Flock(int(f.Fd()), how|syscall.LOCK_NB)
}

func rename(oldpath, newpath string) error {
    return os.Rename(oldpath, newpath)
}

func isErrInvalid(err error) bool {
    if err == os.ErrInvalid {
        return true
    }
    if syserr, ok := err.(*os.SyscallError); ok && syserr.Err == syscall.EINVAL {
        return true
    }
    return false
}

func syncDir(name string) error {
    f, err := os.Open(name)
    if err != nil {
        return err
    }
    defer f.Close()
    if err := f.Sync(); err != nil && !isErrInvalid(err) {
        return err
    }
    return nil
}