diff --git a/src/adminserver/systemcfg/systemcfg.go b/src/adminserver/systemcfg/systemcfg.go index 213a60647..432e9e044 100644 --- a/src/adminserver/systemcfg/systemcfg.go +++ b/src/adminserver/systemcfg/systemcfg.go @@ -262,13 +262,16 @@ func Init() (err error) { return err } db := GetDatabaseFromCfg(envCfgs) - // Initialize the schema, then register the DB. - if err := dao.UpgradeSchema(db); err != nil { - return err - } if err := dao.InitDatabase(db); err != nil { return err } + if err := dao.UpgradeSchema(db); err != nil { + return err + } + if err := dao.CheckSchemaVersion(); err != nil { + return err + } + if err := initCfgStore(); err != nil { return err } diff --git a/src/common/dao/base.go b/src/common/dao/base.go index 3cab37527..6b013e354 100644 --- a/src/common/dao/base.go +++ b/src/common/dao/base.go @@ -81,6 +81,13 @@ func InitDatabase(database *models.Database) error { if err := db.Register(); err != nil { return err } + + log.Info("Register database completed") + return nil +} + +// CheckSchemaVersion checks that whether the schema version matches with the expected one +func CheckSchemaVersion() error { version, err := GetSchemaVersion() if err != nil { return err @@ -89,8 +96,6 @@ func InitDatabase(database *models.Database) error { return fmt.Errorf("unexpected database schema version, expected %s, got %s", SchemaVersion, version.Version) } - - log.Info("Register database completed") return nil } diff --git a/src/common/utils/utils.go b/src/common/utils/utils.go index f6bf117bc..4a5210ad0 100644 --- a/src/common/utils/utils.go +++ b/src/common/utils/utils.go @@ -88,15 +88,17 @@ func TestTCPConn(addr string, timeout, interval int) error { cancel := make(chan int) go func() { + n := 1 for { select { case <-cancel: break default: - conn, err := net.DialTimeout("tcp", addr, time.Duration(timeout)*time.Second) + conn, err := net.DialTimeout("tcp", addr, time.Duration(n)*time.Second) if err != nil { log.Errorf("failed to connect to tcp://%s, retry after %d seconds :%v", addr, interval, err) + n = n * 2 time.Sleep(time.Duration(interval) * time.Second) continue }