123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854 |
- // Copyright 2015 The etcd Authors
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- package wal
- import (
- "bytes"
- "io"
- "io/ioutil"
- "os"
- "path"
- "path/filepath"
- "reflect"
- "testing"
- "github.com/coreos/etcd/pkg/fileutil"
- "github.com/coreos/etcd/pkg/pbutil"
- "github.com/coreos/etcd/raft/raftpb"
- "github.com/coreos/etcd/wal/walpb"
- )
- func TestNew(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, []byte("somedata"))
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- if g := filepath.Base(w.tail().Name()); g != walName(0, 0) {
- t.Errorf("name = %+v, want %+v", g, walName(0, 0))
- }
- defer w.Close()
- // file is preallocated to segment size; only read data written by wal
- off, err := w.tail().Seek(0, io.SeekCurrent)
- if err != nil {
- t.Fatal(err)
- }
- gd := make([]byte, off)
- f, err := os.Open(filepath.Join(p, filepath.Base(w.tail().Name())))
- if err != nil {
- t.Fatal(err)
- }
- defer f.Close()
- if _, err = io.ReadFull(f, gd); err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- var wb bytes.Buffer
- e := newEncoder(&wb, 0, 0)
- err = e.encode(&walpb.Record{Type: crcType, Crc: 0})
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- err = e.encode(&walpb.Record{Type: metadataType, Data: []byte("somedata")})
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- r := &walpb.Record{
- Type: snapshotType,
- Data: pbutil.MustMarshal(&walpb.Snapshot{}),
- }
- if err = e.encode(r); err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- e.flush()
- if !bytes.Equal(gd, wb.Bytes()) {
- t.Errorf("data = %v, want %v", gd, wb.Bytes())
- }
- }
- func TestNewForInitedDir(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- os.Create(filepath.Join(p, walName(0, 0)))
- if _, err = Create(p, nil); err == nil || err != os.ErrExist {
- t.Errorf("err = %v, want %v", err, os.ErrExist)
- }
- }
- func TestOpenAtIndex(t *testing.T) {
- dir, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(dir)
- f, err := os.Create(filepath.Join(dir, walName(0, 0)))
- if err != nil {
- t.Fatal(err)
- }
- f.Close()
- w, err := Open(dir, walpb.Snapshot{})
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- if g := filepath.Base(w.tail().Name()); g != walName(0, 0) {
- t.Errorf("name = %+v, want %+v", g, walName(0, 0))
- }
- if w.seq() != 0 {
- t.Errorf("seq = %d, want %d", w.seq(), 0)
- }
- w.Close()
- wname := walName(2, 10)
- f, err = os.Create(filepath.Join(dir, wname))
- if err != nil {
- t.Fatal(err)
- }
- f.Close()
- w, err = Open(dir, walpb.Snapshot{Index: 5})
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- if g := filepath.Base(w.tail().Name()); g != wname {
- t.Errorf("name = %+v, want %+v", g, wname)
- }
- if w.seq() != 2 {
- t.Errorf("seq = %d, want %d", w.seq(), 2)
- }
- w.Close()
- emptydir, err := ioutil.TempDir(os.TempDir(), "waltestempty")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(emptydir)
- if _, err = Open(emptydir, walpb.Snapshot{}); err != ErrFileNotFound {
- t.Errorf("err = %v, want %v", err, ErrFileNotFound)
- }
- }
- // TestVerify tests that Verify throws a non-nil error when the WAL is corrupted.
- // The test creates a WAL directory and cuts out multiple WAL files. Then
- // it corrupts one of the files by completely truncating it.
- func TestVerify(t *testing.T) {
- walDir, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(walDir)
- // create WAL
- w, err := Create(walDir, nil)
- if err != nil {
- t.Fatal(err)
- }
- defer w.Close()
- // make 5 separate files
- for i := 0; i < 5; i++ {
- es := []raftpb.Entry{{Index: uint64(i), Data: []byte("waldata" + string(i+1))}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if err = w.cut(); err != nil {
- t.Fatal(err)
- }
- }
- // to verify the WAL is not corrupted at this point
- err = Verify(walDir, walpb.Snapshot{})
- if err != nil {
- t.Errorf("expected a nil error, got %v", err)
- }
- walFiles, err := ioutil.ReadDir(walDir)
- if err != nil {
- t.Fatal(err)
- }
- // corrupt the WAL by truncating one of the WAL files completely
- err = os.Truncate(path.Join(walDir, walFiles[2].Name()), 0)
- if err != nil {
- t.Fatal(err)
- }
- err = Verify(walDir, walpb.Snapshot{})
- if err == nil {
- t.Error("expected a non-nil error, got nil")
- }
- }
- // TODO: split it into smaller tests for better readability
- func TestCut(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, nil)
- if err != nil {
- t.Fatal(err)
- }
- defer w.Close()
- state := raftpb.HardState{Term: 1}
- if err = w.Save(state, nil); err != nil {
- t.Fatal(err)
- }
- if err = w.cut(); err != nil {
- t.Fatal(err)
- }
- wname := walName(1, 1)
- if g := filepath.Base(w.tail().Name()); g != wname {
- t.Errorf("name = %s, want %s", g, wname)
- }
- es := []raftpb.Entry{{Index: 1, Term: 1, Data: []byte{1}}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if err = w.cut(); err != nil {
- t.Fatal(err)
- }
- snap := walpb.Snapshot{Index: 2, Term: 1}
- if err = w.SaveSnapshot(snap); err != nil {
- t.Fatal(err)
- }
- wname = walName(2, 2)
- if g := filepath.Base(w.tail().Name()); g != wname {
- t.Errorf("name = %s, want %s", g, wname)
- }
- // check the state in the last WAL
- // We do check before closing the WAL to ensure that Cut syncs the data
- // into the disk.
- f, err := os.Open(filepath.Join(p, wname))
- if err != nil {
- t.Fatal(err)
- }
- defer f.Close()
- nw := &WAL{
- decoder: newDecoder(f),
- start: snap,
- }
- _, gst, _, err := nw.ReadAll()
- if err != nil {
- t.Fatal(err)
- }
- if !reflect.DeepEqual(gst, state) {
- t.Errorf("state = %+v, want %+v", gst, state)
- }
- }
- func TestSaveWithCut(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, []byte("metadata"))
- if err != nil {
- t.Fatal(err)
- }
- state := raftpb.HardState{Term: 1}
- if err = w.Save(state, nil); err != nil {
- t.Fatal(err)
- }
- bigData := make([]byte, 500)
- strdata := "Hello World!!"
- copy(bigData, strdata)
- // set a lower value for SegmentSizeBytes, else the test takes too long to complete
- restoreLater := SegmentSizeBytes
- const EntrySize int = 500
- SegmentSizeBytes = 2 * 1024
- defer func() { SegmentSizeBytes = restoreLater }()
- var index uint64 = 0
- for totalSize := 0; totalSize < int(SegmentSizeBytes); totalSize += EntrySize {
- ents := []raftpb.Entry{{Index: index, Term: 1, Data: bigData}}
- if err = w.Save(state, ents); err != nil {
- t.Fatal(err)
- }
- index++
- }
- w.Close()
- neww, err := Open(p, walpb.Snapshot{})
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- defer neww.Close()
- wname := walName(1, index)
- if g := filepath.Base(neww.tail().Name()); g != wname {
- t.Errorf("name = %s, want %s", g, wname)
- }
- _, newhardstate, entries, err := neww.ReadAll()
- if err != nil {
- t.Fatal(err)
- }
- if !reflect.DeepEqual(newhardstate, state) {
- t.Errorf("Hard State = %+v, want %+v", newhardstate, state)
- }
- if len(entries) != int(SegmentSizeBytes/int64(EntrySize)) {
- t.Errorf("Number of entries = %d, expected = %d", len(entries), int(SegmentSizeBytes/int64(EntrySize)))
- }
- for _, oneent := range entries {
- if !bytes.Equal(oneent.Data, bigData) {
- t.Errorf("the saved data does not match at Index %d : found: %s , want :%s", oneent.Index, oneent.Data, bigData)
- }
- }
- }
- func TestRecover(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, []byte("metadata"))
- if err != nil {
- t.Fatal(err)
- }
- if err = w.SaveSnapshot(walpb.Snapshot{}); err != nil {
- t.Fatal(err)
- }
- ents := []raftpb.Entry{{Index: 1, Term: 1, Data: []byte{1}}, {Index: 2, Term: 2, Data: []byte{2}}}
- if err = w.Save(raftpb.HardState{}, ents); err != nil {
- t.Fatal(err)
- }
- sts := []raftpb.HardState{{Term: 1, Vote: 1, Commit: 1}, {Term: 2, Vote: 2, Commit: 2}}
- for _, s := range sts {
- if err = w.Save(s, nil); err != nil {
- t.Fatal(err)
- }
- }
- w.Close()
- if w, err = Open(p, walpb.Snapshot{}); err != nil {
- t.Fatal(err)
- }
- metadata, state, entries, err := w.ReadAll()
- if err != nil {
- t.Fatal(err)
- }
- if !bytes.Equal(metadata, []byte("metadata")) {
- t.Errorf("metadata = %s, want %s", metadata, "metadata")
- }
- if !reflect.DeepEqual(entries, ents) {
- t.Errorf("ents = %+v, want %+v", entries, ents)
- }
- // only the latest state is recorded
- s := sts[len(sts)-1]
- if !reflect.DeepEqual(state, s) {
- t.Errorf("state = %+v, want %+v", state, s)
- }
- w.Close()
- }
- func TestSearchIndex(t *testing.T) {
- tests := []struct {
- names []string
- index uint64
- widx int
- wok bool
- }{
- {
- []string{
- "0000000000000000-0000000000000000.wal",
- "0000000000000001-0000000000001000.wal",
- "0000000000000002-0000000000002000.wal",
- },
- 0x1000, 1, true,
- },
- {
- []string{
- "0000000000000001-0000000000004000.wal",
- "0000000000000002-0000000000003000.wal",
- "0000000000000003-0000000000005000.wal",
- },
- 0x4000, 1, true,
- },
- {
- []string{
- "0000000000000001-0000000000002000.wal",
- "0000000000000002-0000000000003000.wal",
- "0000000000000003-0000000000005000.wal",
- },
- 0x1000, -1, false,
- },
- }
- for i, tt := range tests {
- idx, ok := searchIndex(tt.names, tt.index)
- if idx != tt.widx {
- t.Errorf("#%d: idx = %d, want %d", i, idx, tt.widx)
- }
- if ok != tt.wok {
- t.Errorf("#%d: ok = %v, want %v", i, ok, tt.wok)
- }
- }
- }
- func TestScanWalName(t *testing.T) {
- tests := []struct {
- str string
- wseq, windex uint64
- wok bool
- }{
- {"0000000000000000-0000000000000000.wal", 0, 0, true},
- {"0000000000000000.wal", 0, 0, false},
- {"0000000000000000-0000000000000000.snap", 0, 0, false},
- }
- for i, tt := range tests {
- s, index, err := parseWalName(tt.str)
- if g := err == nil; g != tt.wok {
- t.Errorf("#%d: ok = %v, want %v", i, g, tt.wok)
- }
- if s != tt.wseq {
- t.Errorf("#%d: seq = %d, want %d", i, s, tt.wseq)
- }
- if index != tt.windex {
- t.Errorf("#%d: index = %d, want %d", i, index, tt.windex)
- }
- }
- }
- func TestRecoverAfterCut(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- md, err := Create(p, []byte("metadata"))
- if err != nil {
- t.Fatal(err)
- }
- for i := 0; i < 10; i++ {
- if err = md.SaveSnapshot(walpb.Snapshot{Index: uint64(i)}); err != nil {
- t.Fatal(err)
- }
- es := []raftpb.Entry{{Index: uint64(i)}}
- if err = md.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if err = md.cut(); err != nil {
- t.Fatal(err)
- }
- }
- md.Close()
- if err := os.Remove(filepath.Join(p, walName(4, 4))); err != nil {
- t.Fatal(err)
- }
- for i := 0; i < 10; i++ {
- w, err := Open(p, walpb.Snapshot{Index: uint64(i)})
- if err != nil {
- if i <= 4 {
- if err != ErrFileNotFound {
- t.Errorf("#%d: err = %v, want %v", i, err, ErrFileNotFound)
- }
- } else {
- t.Errorf("#%d: err = %v, want nil", i, err)
- }
- continue
- }
- metadata, _, entries, err := w.ReadAll()
- if err != nil {
- t.Errorf("#%d: err = %v, want nil", i, err)
- continue
- }
- if !bytes.Equal(metadata, []byte("metadata")) {
- t.Errorf("#%d: metadata = %s, want %s", i, metadata, "metadata")
- }
- for j, e := range entries {
- if e.Index != uint64(j+i+1) {
- t.Errorf("#%d: ents[%d].Index = %+v, want %+v", i, j, e.Index, j+i+1)
- }
- }
- w.Close()
- }
- }
- func TestOpenAtUncommittedIndex(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, nil)
- if err != nil {
- t.Fatal(err)
- }
- if err = w.SaveSnapshot(walpb.Snapshot{}); err != nil {
- t.Fatal(err)
- }
- if err = w.Save(raftpb.HardState{}, []raftpb.Entry{{Index: 0}}); err != nil {
- t.Fatal(err)
- }
- w.Close()
- w, err = Open(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- // commit up to index 0, try to read index 1
- if _, _, _, err = w.ReadAll(); err != nil {
- t.Errorf("err = %v, want nil", err)
- }
- w.Close()
- }
- // TestOpenForRead tests that OpenForRead can load all files.
- // The tests creates WAL directory, and cut out multiple WAL files. Then
- // it releases the lock of part of data, and excepts that OpenForRead
- // can read out all files even if some are locked for write.
- func TestOpenForRead(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- // create WAL
- w, err := Create(p, nil)
- if err != nil {
- t.Fatal(err)
- }
- defer w.Close()
- // make 10 separate files
- for i := 0; i < 10; i++ {
- es := []raftpb.Entry{{Index: uint64(i)}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if err = w.cut(); err != nil {
- t.Fatal(err)
- }
- }
- // release the lock to 5
- unlockIndex := uint64(5)
- w.ReleaseLockTo(unlockIndex)
- // All are available for read
- w2, err := OpenForRead(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- defer w2.Close()
- _, _, ents, err := w2.ReadAll()
- if err != nil {
- t.Fatalf("err = %v, want nil", err)
- }
- if g := ents[len(ents)-1].Index; g != 9 {
- t.Errorf("last index read = %d, want %d", g, 9)
- }
- }
- func TestSaveEmpty(t *testing.T) {
- var buf bytes.Buffer
- var est raftpb.HardState
- w := WAL{
- encoder: newEncoder(&buf, 0, 0),
- }
- if err := w.saveState(&est); err != nil {
- t.Errorf("err = %v, want nil", err)
- }
- if len(buf.Bytes()) != 0 {
- t.Errorf("buf.Bytes = %d, want 0", len(buf.Bytes()))
- }
- }
- func TestReleaseLockTo(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- // create WAL
- w, err := Create(p, nil)
- defer func() {
- if err = w.Close(); err != nil {
- t.Fatal(err)
- }
- }()
- if err != nil {
- t.Fatal(err)
- }
- // release nothing if no files
- err = w.ReleaseLockTo(10)
- if err != nil {
- t.Errorf("err = %v, want nil", err)
- }
- // make 10 separate files
- for i := 0; i < 10; i++ {
- es := []raftpb.Entry{{Index: uint64(i)}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if err = w.cut(); err != nil {
- t.Fatal(err)
- }
- }
- // release the lock to 5
- unlockIndex := uint64(5)
- w.ReleaseLockTo(unlockIndex)
- // expected remaining are 4,5,6,7,8,9,10
- if len(w.locks) != 7 {
- t.Errorf("len(w.locks) = %d, want %d", len(w.locks), 7)
- }
- for i, l := range w.locks {
- var lockIndex uint64
- _, lockIndex, err = parseWalName(filepath.Base(l.Name()))
- if err != nil {
- t.Fatal(err)
- }
- if lockIndex != uint64(i+4) {
- t.Errorf("#%d: lockindex = %d, want %d", i, lockIndex, uint64(i+4))
- }
- }
- // release the lock to 15
- unlockIndex = uint64(15)
- w.ReleaseLockTo(unlockIndex)
- // expected remaining is 10
- if len(w.locks) != 1 {
- t.Errorf("len(w.locks) = %d, want %d", len(w.locks), 1)
- }
- _, lockIndex, err := parseWalName(filepath.Base(w.locks[0].Name()))
- if err != nil {
- t.Fatal(err)
- }
- if lockIndex != uint64(10) {
- t.Errorf("lockindex = %d, want %d", lockIndex, 10)
- }
- }
- // TestTailWriteNoSlackSpace ensures that tail writes append if there's no preallocated space.
- func TestTailWriteNoSlackSpace(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- // create initial WAL
- w, err := Create(p, []byte("metadata"))
- if err != nil {
- t.Fatal(err)
- }
- // write some entries
- for i := 1; i <= 5; i++ {
- es := []raftpb.Entry{{Index: uint64(i), Term: 1, Data: []byte{byte(i)}}}
- if err = w.Save(raftpb.HardState{Term: 1}, es); err != nil {
- t.Fatal(err)
- }
- }
- // get rid of slack space by truncating file
- off, serr := w.tail().Seek(0, io.SeekCurrent)
- if serr != nil {
- t.Fatal(serr)
- }
- if terr := w.tail().Truncate(off); terr != nil {
- t.Fatal(terr)
- }
- w.Close()
- // open, write more
- w, err = Open(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- _, _, ents, rerr := w.ReadAll()
- if rerr != nil {
- t.Fatal(rerr)
- }
- if len(ents) != 5 {
- t.Fatalf("got entries %+v, expected 5 entries", ents)
- }
- // write more entries
- for i := 6; i <= 10; i++ {
- es := []raftpb.Entry{{Index: uint64(i), Term: 1, Data: []byte{byte(i)}}}
- if err = w.Save(raftpb.HardState{Term: 1}, es); err != nil {
- t.Fatal(err)
- }
- }
- w.Close()
- // confirm all writes
- w, err = Open(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- _, _, ents, rerr = w.ReadAll()
- if rerr != nil {
- t.Fatal(rerr)
- }
- if len(ents) != 10 {
- t.Fatalf("got entries %+v, expected 10 entries", ents)
- }
- w.Close()
- }
- // TestRestartCreateWal ensures that an interrupted WAL initialization is clobbered on restart
- func TestRestartCreateWal(t *testing.T) {
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- // make temporary directory so it looks like initialization is interrupted
- tmpdir := filepath.Clean(p) + ".tmp"
- if err = os.Mkdir(tmpdir, fileutil.PrivateDirMode); err != nil {
- t.Fatal(err)
- }
- if _, err = os.OpenFile(filepath.Join(tmpdir, "test"), os.O_WRONLY|os.O_CREATE, fileutil.PrivateFileMode); err != nil {
- t.Fatal(err)
- }
- w, werr := Create(p, []byte("abc"))
- if werr != nil {
- t.Fatal(werr)
- }
- w.Close()
- if Exist(tmpdir) {
- t.Fatalf("got %q exists, expected it to not exist", tmpdir)
- }
- if w, err = OpenForRead(p, walpb.Snapshot{}); err != nil {
- t.Fatal(err)
- }
- defer w.Close()
- if meta, _, _, rerr := w.ReadAll(); rerr != nil || string(meta) != "abc" {
- t.Fatalf("got error %v and meta %q, expected nil and %q", rerr, meta, "abc")
- }
- }
- // TestOpenOnTornWrite ensures that entries past the torn write are truncated.
- func TestOpenOnTornWrite(t *testing.T) {
- maxEntries := 40
- clobberIdx := 20
- overwriteEntries := 5
- p, err := ioutil.TempDir(os.TempDir(), "waltest")
- if err != nil {
- t.Fatal(err)
- }
- defer os.RemoveAll(p)
- w, err := Create(p, nil)
- defer func() {
- if err = w.Close(); err != nil && err != os.ErrInvalid {
- t.Fatal(err)
- }
- }()
- if err != nil {
- t.Fatal(err)
- }
- // get offset of end of each saved entry
- offsets := make([]int64, maxEntries)
- for i := range offsets {
- es := []raftpb.Entry{{Index: uint64(i)}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- if offsets[i], err = w.tail().Seek(0, io.SeekCurrent); err != nil {
- t.Fatal(err)
- }
- }
- fn := filepath.Join(p, filepath.Base(w.tail().Name()))
- w.Close()
- // clobber some entry with 0's to simulate a torn write
- f, ferr := os.OpenFile(fn, os.O_WRONLY, fileutil.PrivateFileMode)
- if ferr != nil {
- t.Fatal(ferr)
- }
- defer f.Close()
- _, err = f.Seek(offsets[clobberIdx], io.SeekStart)
- if err != nil {
- t.Fatal(err)
- }
- zeros := make([]byte, offsets[clobberIdx+1]-offsets[clobberIdx])
- _, err = f.Write(zeros)
- if err != nil {
- t.Fatal(err)
- }
- f.Close()
- w, err = Open(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- // seek up to clobbered entry
- _, _, _, err = w.ReadAll()
- if err != nil {
- t.Fatal(err)
- }
- // write a few entries past the clobbered entry
- for i := 0; i < overwriteEntries; i++ {
- // Index is different from old, truncated entries
- es := []raftpb.Entry{{Index: uint64(i + clobberIdx), Data: []byte("new")}}
- if err = w.Save(raftpb.HardState{}, es); err != nil {
- t.Fatal(err)
- }
- }
- w.Close()
- // read back the entries, confirm number of entries matches expectation
- w, err = OpenForRead(p, walpb.Snapshot{})
- if err != nil {
- t.Fatal(err)
- }
- _, _, ents, rerr := w.ReadAll()
- if rerr != nil {
- // CRC error? the old entries were likely never truncated away
- t.Fatal(rerr)
- }
- wEntries := (clobberIdx - 1) + overwriteEntries
- if len(ents) != wEntries {
- t.Fatalf("expected len(ents) = %d, got %d", wEntries, len(ents))
- }
- }
|