From a70c5ac33c4c405d656de8287f64cc00708d5a1e Mon Sep 17 00:00:00 2001 From: Swapnil Mhamane Date: Tue, 6 Aug 2019 14:24:27 +0530 Subject: [PATCH] Add experimental faiil below revision flag Main motivation behind this is to help extenal entity configure revision in case of bucket change or object store prefix change Signed-off-by: Swapnil Mhamane --- cmd/initializer.go | 3 +- cmd/types.go | 3 +- pkg/initializer/initializer.go | 4 +-- pkg/initializer/validator/datavalidator.go | 10 ++++-- .../validator/datavalidator_test.go | 32 ++++++++++++------- pkg/initializer/validator/types.go | 2 +- pkg/server/httpAPI.go | 17 +++++++++- test/e2e/integration/cloud_backup_test.go | 4 +-- 8 files changed, 53 insertions(+), 22 deletions(-) diff --git a/cmd/initializer.go b/cmd/initializer.go index 810a9832d..e03a30de3 100644 --- a/cmd/initializer.go +++ b/cmd/initializer.go @@ -81,7 +81,7 @@ func NewInitializeCommand(stopCh <-chan struct{}) *cobra.Command { } } etcdInitializer := initializer.NewInitializer(options, snapstoreConfig, logger) - err = etcdInitializer.Initialize(mode) + err = etcdInitializer.Initialize(mode, failBelowRevision) if err != nil { logger.Fatalf("initializer failed. %v", err) } @@ -90,6 +90,7 @@ func NewInitializeCommand(stopCh <-chan struct{}) *cobra.Command { initializeEtcdFlags(initializeCmd) initializeSnapstoreFlags(initializeCmd) initializeValidatorFlags(initializeCmd) + initializeCmd.Flags().Int64Var(&failBelowRevision, "experimental-fail-below-revision", 0, "revision below which validation fails") return initializeCmd } diff --git a/cmd/types.go b/cmd/types.go index d9299d2c6..1189b7cae 100644 --- a/cmd/types.go +++ b/cmd/types.go @@ -68,7 +68,8 @@ var ( snapstoreTempDir string //initializer flags - validationMode string + validationMode string + failBelowRevision int64 ) var emptyStruct struct{} diff --git a/pkg/initializer/initializer.go b/pkg/initializer/initializer.go index 797cb6296..dae4d6064 100644 --- a/pkg/initializer/initializer.go +++ b/pkg/initializer/initializer.go @@ -42,9 +42,9 @@ const ( // * Check if Latest snapshot available. // - Try to perform an Etcd data restoration from the latest snapshot. // - No snapshots are available, start etcd as a fresh installation. -func (e *EtcdInitializer) Initialize(mode validator.Mode) error { +func (e *EtcdInitializer) Initialize(mode validator.Mode, failBelowRevision int64) error { start := time.Now() - dataDirStatus, err := e.Validator.Validate(mode) + dataDirStatus, err := e.Validator.Validate(mode, failBelowRevision) if err != nil && dataDirStatus != validator.DataDirectoryNotExist { metrics.ValidationDurationSeconds.With(prometheus.Labels{metrics.LabelSucceeded: metrics.ValueSucceededFalse}).Observe(time.Now().Sub(start).Seconds()) err = fmt.Errorf("error while initializing: %v", err) diff --git a/pkg/initializer/validator/datavalidator.go b/pkg/initializer/validator/datavalidator.go index 6fc9520a2..80f0f2ea3 100644 --- a/pkg/initializer/validator/datavalidator.go +++ b/pkg/initializer/validator/datavalidator.go @@ -76,7 +76,7 @@ func (d *DataValidator) backendPath() string { return filepath.Join(d.snapDir(), // - If data directory structure is invalid return DataDirectoryInvStruct status. // * Check for data corruption. // - return data directory corruption status. -func (d *DataValidator) Validate(mode Mode) (DataDirStatus, error) { +func (d *DataValidator) Validate(mode Mode, failBelowRevision int64) (DataDirStatus, error) { dataDir := d.Config.DataDir dirExists, err := directoryExist(dataDir) if err != nil { @@ -98,7 +98,7 @@ func (d *DataValidator) Validate(mode Mode) (DataDirStatus, error) { if d.Config.SnapstoreConfig != nil { d.Logger.Info("Checking for revision consistency...") - if err = checkRevisionConsistency(d.backendPath(), *d.Config.SnapstoreConfig); err != nil { + if err = checkRevisionConsistency(d.backendPath(), *d.Config.SnapstoreConfig, failBelowRevision); err != nil { d.Logger.Infof("Etcd revision inconsistent with latest snapshot revision: %v", err) return RevisionConsistencyError, nil } @@ -338,12 +338,16 @@ func verifyDB(path string) error { } // checkRevisionConsistency compares the latest revisions on the etcd db file and the latest snapshot to verify that the etcd revision is not lesser than snapshot revision. -func checkRevisionConsistency(dbPath string, config snapstore.Config) error { +func checkRevisionConsistency(dbPath string, config snapstore.Config, failBelowRevision int64) error { etcdRevision, err := getLatestEtcdRevision(dbPath) if err != nil { return fmt.Errorf("unable to get current etcd revision from backend db file: %v", err) } + if etcdRevision < failBelowRevision { + return fmt.Errorf("current etcd revision (%d) is less than fail below revision (%d): possible data loss", etcdRevision, failBelowRevision) + } + store, err := snapstore.GetSnapstore(&config) if err != nil { return fmt.Errorf("unable to fetch snapstore: %v", err) diff --git a/pkg/initializer/validator/datavalidator_test.go b/pkg/initializer/validator/datavalidator_test.go index 2bbdb37d6..f6d2fdec3 100644 --- a/pkg/initializer/validator/datavalidator_test.go +++ b/pkg/initializer/validator/datavalidator_test.go @@ -45,7 +45,7 @@ var _ = Describe("Running Datavalidator", func() { tempDir := fmt.Sprintf("%s.%s", restoreDataDir, "temp") err = os.Rename(restoreDataDir, tempDir) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).Should(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(DataDirectoryNotExist), Equal(DataDirectoryError))) err = os.Rename(tempDir, restoreDataDir) @@ -61,7 +61,7 @@ var _ = Describe("Running Datavalidator", func() { tempDir := fmt.Sprintf("%s.%s", memberDir, "temp") err = os.Rename(memberDir, tempDir) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(DataDirectoryInvStruct), Equal(DataDirectoryError))) err = os.Rename(tempDir, memberDir) @@ -75,7 +75,7 @@ var _ = Describe("Running Datavalidator", func() { tempDir := fmt.Sprintf("%s.%s", snapDir, "temp") err = os.Rename(snapDir, tempDir) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(DataDirectoryInvStruct), Equal(DataDirectoryError))) err = os.Rename(tempDir, snapDir) @@ -88,7 +88,7 @@ var _ = Describe("Running Datavalidator", func() { tempDir := fmt.Sprintf("%s.%s", walDir, "temp") err = os.Rename(walDir, tempDir) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(DataDirectoryInvStruct), Equal(DataDirectoryError))) err = os.Rename(tempDir, walDir) @@ -104,7 +104,7 @@ var _ = Describe("Running Datavalidator", func() { Expect(err).ShouldNot(HaveOccurred()) err = os.Mkdir(walDir, 0700) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Sanity) + dataDirStatus, err := validator.Validate(Sanity, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(Equal(DataDirectoryValid)) err = os.RemoveAll(walDir) @@ -138,7 +138,7 @@ var _ = Describe("Running Datavalidator", func() { // newEtcdRevision: current revision number on etcd db Expect(etcdRevision).To(BeNumerically(">=", newEtcdRevision)) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(RevisionConsistencyError), Equal(DataDirectoryError))) @@ -174,7 +174,7 @@ var _ = Describe("Running Datavalidator", func() { _, err = file.Write(byteSlice) Expect(err).ShouldNot(HaveOccurred()) - dataDirStatus, err := validator.Validate(Full) + dataDirStatus, err := validator.Validate(Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(int(dataDirStatus)).Should(SatisfyAny(Equal(DataDirectoryCorrupt), Equal(DataDirectoryError), Equal(RevisionConsistencyError))) @@ -187,10 +187,20 @@ var _ = Describe("Running Datavalidator", func() { }) }) Context("with clean data directory", func() { - It("should return DataDirStatus as DataDirectoryValid, and nil error", func() { - dataDirStatus, err := validator.Validate(Full) - Expect(err).ShouldNot(HaveOccurred()) - Expect(int(dataDirStatus)).Should(Equal(DataDirectoryValid)) + Context("with fail below revision configured to low value", func() { + It("should return DataDirStatus as DataDirectoryValid, and nil error", func() { + dataDirStatus, err := validator.Validate(Full, 0) + Expect(err).ShouldNot(HaveOccurred()) + Expect(int(dataDirStatus)).Should(Equal(DataDirectoryValid)) + }) + }) + + Context("with fail below revision configured to high value", func() { + It("should return DataDirStatus as RevisionConsistencyError and nil error", func() { + dataDirStatus, err := validator.Validate(Full, 1000000) + Expect(err).ShouldNot(HaveOccurred()) + Expect(int(dataDirStatus)).Should(Equal(RevisionConsistencyError)) + }) }) }) }) diff --git a/pkg/initializer/validator/types.go b/pkg/initializer/validator/types.go index 56e069b1d..cb21e3a30 100644 --- a/pkg/initializer/validator/types.go +++ b/pkg/initializer/validator/types.go @@ -65,5 +65,5 @@ type DataValidator struct { // Validator is the interface for data validation actions. type Validator interface { - Validate(Mode) error + Validate(Mode, int64) error } diff --git a/pkg/server/httpAPI.go b/pkg/server/httpAPI.go index 8960c5bf3..996e70453 100644 --- a/pkg/server/httpAPI.go +++ b/pkg/server/httpAPI.go @@ -18,6 +18,7 @@ import ( "fmt" "net/http" "net/http/pprof" + "strconv" "sync" "sync/atomic" @@ -150,6 +151,20 @@ func (h *HTTPHandler) serveInitialize(rw http.ResponseWriter, req *http.Request) <-h.AckCh } + failBelowRevisionStr := req.URL.Query().Get("failbelowrevision") + h.Logger.Infof("Validation failBelowRevision: %s", failBelowRevisionStr) + var failBelowRevision int64 + if len(failBelowRevisionStr) != 0 { + var err error + failBelowRevision, err = strconv.ParseInt(failBelowRevisionStr, 10, 64) + if err != nil { + h.initializationStatusMutex.Lock() + defer h.initializationStatusMutex.Unlock() + h.Logger.Errorf("Failed initialization due wrong parameter value `failbelowrevision`: %v", err) + h.initializationStatus = initializationStatusFailed + return + } + } switch modeVal := req.URL.Query().Get("mode"); modeVal { case string(validator.Full): mode = validator.Full @@ -159,7 +174,7 @@ func (h *HTTPHandler) serveInitialize(rw http.ResponseWriter, req *http.Request) mode = validator.Full } h.Logger.Infof("Validation mode: %s", mode) - err := h.EtcdInitializer.Initialize(mode) + err := h.EtcdInitializer.Initialize(mode, failBelowRevision) h.initializationStatusMutex.Lock() defer h.initializationStatusMutex.Unlock() if err != nil { diff --git a/test/e2e/integration/cloud_backup_test.go b/test/e2e/integration/cloud_backup_test.go index c08098cc7..c0c2cdb67 100644 --- a/test/e2e/integration/cloud_backup_test.go +++ b/test/e2e/integration/cloud_backup_test.go @@ -216,7 +216,7 @@ var _ = Describe("CloudBackup", func() { SnapstoreConfig: snapstoreConfig, }, } - dataDirStatus, err := dataValidator.Validate(validator.Full) + dataDirStatus, err := dataValidator.Validate(validator.Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(dataDirStatus).Should(Equal(validator.DataDirStatus(validator.DataDirectoryValid))) }) @@ -248,7 +248,7 @@ var _ = Describe("CloudBackup", func() { SnapstoreConfig: snapstoreConfig, }, } - dataDirStatus, err := dataValidator.Validate(validator.Full) + dataDirStatus, err := dataValidator.Validate(validator.Full, 0) Expect(err).ShouldNot(HaveOccurred()) Expect(dataDirStatus).Should(SatisfyAny(Equal(validator.DataDirStatus(validator.DataDirectoryCorrupt)), Equal(validator.DataDirStatus(validator.RevisionConsistencyError)))) })