Adding the context to bmc.Close().

bmc-toolbox · Apr 5, 2021 · 99a478f · 99a478f
1 parent cb4ebf6
commit 99a478f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/pkg/butler/configure.go b/pkg/butler/configure.go
@@ -1,6 +1,7 @@
 package butler
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"time"
@@ -80,7 +81,7 @@ func (b *Butler) configureAsset(config []byte, asset *asset.Asset) (err error) {
 		c := configure.NewBmcConfigurator(bmc, asset, b.Config.Resources, renderedConfig, b.Config, b.StopChan, log)
 		c.Apply()
 
-		bmc.Close()
+		bmc.Close(context.TODO())
 	case devices.Cmc:
 		chassis := client.(devices.Cmc)
 

diff --git a/pkg/butler/configure/server.go b/pkg/butler/configure/server.go
@@ -1,6 +1,7 @@
 package configure
 
 import (
+	"context"
 	"strings"
 
 	"github.com/bmc-toolbox/bmcbutler/pkg/asset"
@@ -191,7 +192,7 @@ func (b *Bmc) Apply() {
 		}).Info("BMC to be reset.")
 
 		// Close the current connection - so we don't leave connections hanging.
-		b.bmc.Close()
+		b.bmc.Close(context.TODO())
 
 		//// reset BMC using SSH.
 		_, err := b.bmc.PowerCycleBmc()

diff --git a/pkg/butler/execute.go b/pkg/butler/execute.go
@@ -1,6 +1,7 @@
 package butler
 
 import (
+	"context"
 	"errors"
 	"fmt"
 
@@ -70,7 +71,7 @@ func (b *Butler) executeCommand(command string, asset *asset.Asset) (err error)
 			}).Debug("Command executed.")
 
 		}
-		bmc.Close()
+		bmc.Close(context.TODO())
 	case devices.Cmc:
 		chassis := client.(devices.Cmc)
 		//b.executeCommandChassis(chassis, command)