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 <[email protected]>
gardener · Aug 6, 2019 · a70c5ac · a70c5ac
1 parent a17381e
commit a70c5ac
Show file tree

Hide file tree

Showing 8 changed files with 53 additions and 22 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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))))
 			})