From beb0d8397a13db76bfaaf3ecb9789da25c1b480d Mon Sep 17 00:00:00 2001
From: Brian Downs <brian.downs@gmail.com>
Date: Tue, 4 May 2021 10:03:28 -0700
Subject: [PATCH] reference node name when needed

Signed-off-by: Brian Downs <brian.downs@gmail.com>
---
 pkg/etcd/etcd.go | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

diff --git a/pkg/etcd/etcd.go b/pkg/etcd/etcd.go
index 78f0689755..b9b08ff497 100644
--- a/pkg/etcd/etcd.go
+++ b/pkg/etcd/etcd.go
@@ -65,14 +65,13 @@ var (
 )
 
 type ETCD struct {
-	client   *etcd.Client
-	config   *config.Control
-	name     string
-	runtime  *config.ControlRuntime
-	address  string
-	cron     *cron.Cron
-	s3       *s3
-	nodeName string
+	client  *etcd.Client
+	config  *config.Control
+	name    string
+	runtime *config.ControlRuntime
+	address string
+	cron    *cron.Cron
+	s3      *s3
 }
 
 type learnerProgress struct {
@@ -92,8 +91,7 @@ type Members struct {
 // ETCD with an initialized cron value.
 func NewETCD() *ETCD {
 	return &ETCD{
-		cron:     cron.New(),
-		nodeName: os.Getenv("NODE_NAME"),
+		cron: cron.New(),
 	}
 }
 
@@ -948,11 +946,13 @@ func (e *ETCD) listSnapshots(ctx context.Context, snapshotDir string) ([]snapsho
 		return nil, err
 	}
 
+	nodeName := os.Getenv("NODE_NAME")
+
 	for _, f := range files {
 		snapshots = append(snapshots, snapshotFile{
 			Name:     f.Name(),
 			Location: "file://" + filepath.Join(snapshotDir, f.Name()),
-			NodeName: e.nodeName,
+			NodeName: nodeName,
 			CreatedAt: &metav1.Time{
 				Time: f.ModTime(),
 			},
@@ -1022,18 +1022,20 @@ func (e *ETCD) StoreSnapshotData(ctx context.Context) error {
 			snapshotConfigMap.Data = make(map[string]string, 0)
 		}
 
+		nodeName := os.Getenv("NODE_NAME")
+
 		// remove entries for this node only
 		for k, v := range snapshotConfigMap.Data {
 			var sf snapshotFile
 			if err := json.Unmarshal([]byte(v), &sf); err != nil {
 				return err
 			}
-			if sf.NodeName == e.nodeName || sf.NodeName == "s3" {
+			if sf.NodeName == nodeName || sf.NodeName == "s3" {
 				delete(snapshotConfigMap.Data, k)
 			}
 		}
 
-		// this node's entries to the ConfigMap
+		// save this node's entries to the ConfigMap
 		for k, v := range data {
 			snapshotConfigMap.Data[k] = v
 		}