mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
parent
c449da6ff9
commit
bd7107bd4b
52
wal/wal.go
52
wal/wal.go
@ -69,7 +69,11 @@ var (
|
|||||||
// A just opened WAL is in read mode, and ready for reading records.
|
// A just opened WAL is in read mode, and ready for reading records.
|
||||||
// The WAL will be ready for appending after reading out all the previous records.
|
// The WAL will be ready for appending after reading out all the previous records.
|
||||||
type WAL struct {
|
type WAL struct {
|
||||||
dir string // the living directory of the underlay files
|
dir string // the living directory of the underlay files
|
||||||
|
|
||||||
|
// dirFile is a fd for the wal directory for syncing on Rename
|
||||||
|
dirFile *os.File
|
||||||
|
|
||||||
metadata []byte // metadata recorded at the head of each WAL
|
metadata []byte // metadata recorded at the head of each WAL
|
||||||
state raftpb.HardState // hardstate recorded at the head of WAL
|
state raftpb.HardState // hardstate recorded at the head of WAL
|
||||||
|
|
||||||
@ -108,10 +112,10 @@ func Create(dirpath string, metadata []byte) (*WAL, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if _, err := f.Seek(0, os.SEEK_END); err != nil {
|
if _, err = f.Seek(0, os.SEEK_END); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if err := fileutil.Preallocate(f.File, SegmentSizeBytes, true); err != nil {
|
if err = fileutil.Preallocate(f.File, SegmentSizeBytes, true); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -121,17 +125,33 @@ func Create(dirpath string, metadata []byte) (*WAL, error) {
|
|||||||
encoder: newEncoder(f, 0),
|
encoder: newEncoder(f, 0),
|
||||||
}
|
}
|
||||||
w.locks = append(w.locks, f)
|
w.locks = append(w.locks, f)
|
||||||
if err := w.saveCrc(0); err != nil {
|
if err = w.saveCrc(0); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if err := w.encoder.encode(&walpb.Record{Type: metadataType, Data: metadata}); err != nil {
|
if err = w.encoder.encode(&walpb.Record{Type: metadataType, Data: metadata}); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if err := w.SaveSnapshot(walpb.Snapshot{}); err != nil {
|
if err = w.SaveSnapshot(walpb.Snapshot{}); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return w.renameWal(tmpdirpath)
|
if w, err = w.renameWal(tmpdirpath); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// directory was renamed; sync parent dir to persist rename
|
||||||
|
pdir, perr := fileutil.OpenDir(path.Dir(w.dir))
|
||||||
|
if perr != nil {
|
||||||
|
return nil, perr
|
||||||
|
}
|
||||||
|
if perr = fileutil.Fsync(pdir); perr != nil {
|
||||||
|
return nil, perr
|
||||||
|
}
|
||||||
|
if perr = pdir.Close(); err != nil {
|
||||||
|
return nil, perr
|
||||||
|
}
|
||||||
|
|
||||||
|
return w, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Open opens the WAL at the given snap.
|
// Open opens the WAL at the given snap.
|
||||||
@ -141,7 +161,14 @@ func Create(dirpath string, metadata []byte) (*WAL, error) {
|
|||||||
// the given snap. The WAL cannot be appended to before reading out all of its
|
// the given snap. The WAL cannot be appended to before reading out all of its
|
||||||
// previous records.
|
// previous records.
|
||||||
func Open(dirpath string, snap walpb.Snapshot) (*WAL, error) {
|
func Open(dirpath string, snap walpb.Snapshot) (*WAL, error) {
|
||||||
return openAtIndex(dirpath, snap, true)
|
w, err := openAtIndex(dirpath, snap, true)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if w.dirFile, err = fileutil.OpenDir(w.dir); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return w, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// OpenForRead only opens the wal files for read.
|
// OpenForRead only opens the wal files for read.
|
||||||
@ -373,6 +400,10 @@ func (w *WAL) cut() error {
|
|||||||
if err = os.Rename(newTail.Name(), fpath); err != nil {
|
if err = os.Rename(newTail.Name(), fpath); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if err = fileutil.Fsync(w.dirFile); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
newTail.Close()
|
newTail.Close()
|
||||||
|
|
||||||
if newTail, err = fileutil.LockFile(fpath, os.O_WRONLY, fileutil.PrivateFileMode); err != nil {
|
if newTail, err = fileutil.LockFile(fpath, os.O_WRONLY, fileutil.PrivateFileMode); err != nil {
|
||||||
@ -475,6 +506,11 @@ func (w *WAL) Close() error {
|
|||||||
plog.Errorf("failed to unlock during closing wal: %s", err)
|
plog.Errorf("failed to unlock during closing wal: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if err := w.dirFile.Close(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -16,7 +16,11 @@
|
|||||||
|
|
||||||
package wal
|
package wal
|
||||||
|
|
||||||
import "os"
|
import (
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"github.com/coreos/etcd/pkg/fileutil"
|
||||||
|
)
|
||||||
|
|
||||||
func (w *WAL) renameWal(tmpdirpath string) (*WAL, error) {
|
func (w *WAL) renameWal(tmpdirpath string) (*WAL, error) {
|
||||||
// On non-Windows platforms, hold the lock while renaming. Releasing
|
// On non-Windows platforms, hold the lock while renaming. Releasing
|
||||||
@ -34,5 +38,7 @@ func (w *WAL) renameWal(tmpdirpath string) (*WAL, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
w.fp = newFilePipeline(w.dir, SegmentSizeBytes)
|
w.fp = newFilePipeline(w.dir, SegmentSizeBytes)
|
||||||
return w, nil
|
df, err := fileutil.OpenDir(w.dir)
|
||||||
|
w.dirFile = df
|
||||||
|
return w, err
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user