87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
|
// +build !windows,!plan9
|
||
|
|
|
||
|
|
package bolt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"syscall"
|
||
|
|
"time"
|
||
|
|
"unsafe"
|
||
|
|
)
|
||
|
|
|
||
|
|
// flock acquires an advisory lock on a file descriptor.
|
||
|
|
func flock(f *os.File, exclusive bool, timeout time.Duration) error {
|
||
|
|
var t time.Time
|
||
|
|
for {
|
||
|
|
// If we're beyond our timeout then return an error.
|
||
|
|
// This can only occur after we've attempted a flock once.
|
||
|
|
if t.IsZero() {
|
||
|
|
t = time.Now()
|
||
|
|
} else if timeout > 0 && time.Since(t) > timeout {
|
||
|
|
return ErrTimeout
|
||
|
|
}
|
||
|
|
flag := syscall.LOCK_SH
|
||
|
|
if exclusive {
|
||
|
|
flag = syscall.LOCK_EX
|
||
|
|
}
|
||
|
|
|
||
|
|
// Otherwise attempt to obtain an exclusive lock.
|
||
|
|
err := syscall.Flock(int(f.Fd()), flag|syscall.LOCK_NB)
|
||
|
|
if err == nil {
|
||
|
|
return nil
|
||
|
|
} else if err != syscall.EWOULDBLOCK {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Wait for a bit and try again.
|
||
|
|
time.Sleep(50 * time.Millisecond)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// funlock releases an advisory lock on a file descriptor.
|
||
|
|
func funlock(f *os.File) error {
|
||
|
|
return syscall.Flock(int(f.Fd()), syscall.LOCK_UN)
|
||
|
|
}
|
||
|
|
|
||
|
|
// mmap memory maps a DB's data file.
|
||
|
|
func mmap(db *DB, sz int) error {
|
||
|
|
// Truncate and fsync to ensure file size metadata is flushed.
|
||
|
|
// https://github.com/boltdb/bolt/issues/284
|
||
|
|
if !db.NoGrowSync && !db.readOnly {
|
||
|
|
if err := db.file.Truncate(int64(sz)); err != nil {
|
||
|
|
return fmt.Errorf("file resize error: %s", err)
|
||
|
|
}
|
||
|
|
if err := db.file.Sync(); err != nil {
|
||
|
|
return fmt.Errorf("file sync error: %s", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Map the data file to memory.
|
||
|
|
b, err := syscall.Mmap(int(db.file.Fd()), 0, sz, syscall.PROT_READ, syscall.MAP_SHARED)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save the original byte slice and convert to a byte array pointer.
|
||
|
|
db.dataref = b
|
||
|
|
db.data = (*[maxMapSize]byte)(unsafe.Pointer(&b[0]))
|
||
|
|
db.datasz = sz
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// munmap unmaps a DB's data file from memory.
|
||
|
|
func munmap(db *DB) error {
|
||
|
|
// Ignore the unmap if we have no mapped data.
|
||
|
|
if db.dataref == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Unmap using the original byte slice.
|
||
|
|
err := syscall.Munmap(db.dataref)
|
||
|
|
db.dataref = nil
|
||
|
|
db.data = nil
|
||
|
|
db.datasz = 0
|
||
|
|
return err
|
||
|
|
}
|