Commit 379defa2 authored by Hussein Galal's avatar Hussein Galal Committed by Brad Davidson

reset etcd name on cluster reset (#2284)

* reset etcd name on cluster reset Signed-off-by: 's avatargalal-hussein <hussein.galal.ahmed.11@gmail.com> * gofmt Signed-off-by: 's avatargalal-hussein <hussein.galal.ahmed.11@gmail.com>
parent 3008a048
...@@ -148,8 +148,14 @@ func (e *ETCD) Reset(ctx context.Context, clientAccessInfo *clientaccess.Info) e ...@@ -148,8 +148,14 @@ func (e *ETCD) Reset(ctx context.Context, clientAccessInfo *clientaccess.Info) e
if info.IsDir() { if info.IsDir() {
return fmt.Errorf("etcd: snapshot path is directory: %s", e.config.ClusterResetRestorePath) return fmt.Errorf("etcd: snapshot path is directory: %s", e.config.ClusterResetRestorePath)
} }
return e.Restore(ctx) if err := e.Restore(ctx); err != nil {
return err
}
} }
if err := e.setName(true); err != nil {
return err
}
return e.newCluster(ctx, true) return e.newCluster(ctx, true)
} }
...@@ -272,17 +278,17 @@ func (e *ETCD) Register(ctx context.Context, config *config.Control, l net.Liste ...@@ -272,17 +278,17 @@ func (e *ETCD) Register(ctx context.Context, config *config.Control, l net.Liste
e.config.Datastore.Config.CertFile = e.runtime.ClientETCDCert e.config.Datastore.Config.CertFile = e.runtime.ClientETCDCert
e.config.Datastore.Config.KeyFile = e.runtime.ClientETCDKey e.config.Datastore.Config.KeyFile = e.runtime.ClientETCDKey
if err := e.setName(); err != nil { if err := e.setName(false); err != nil {
return nil, nil, err return nil, nil, err
} }
return l, e.handler(handler), err return l, e.handler(handler), err
} }
func (e *ETCD) setName() error { func (e *ETCD) setName(force bool) error {
fileName := nameFile(e.config) fileName := nameFile(e.config)
data, err := ioutil.ReadFile(fileName) data, err := ioutil.ReadFile(fileName)
if os.IsNotExist(err) { if os.IsNotExist(err) || force {
h, err := os.Hostname() h, err := os.Hostname()
if err != nil { if err != nil {
return err return err
...@@ -615,11 +621,7 @@ func (e *ETCD) Restore(ctx context.Context) error { ...@@ -615,11 +621,7 @@ func (e *ETCD) Restore(ctx context.Context) error {
} else { } else {
return err return err
} }
if err := e.setName(); err != nil { return nil
return err
}
return e.newCluster(ctx, true)
} }
// snapshotRetention iterates through the snapshots and removes the oldest // snapshotRetention iterates through the snapshots and removes the oldest
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment