Merge pull request #392 from pace/failover-context

failover: Pass context to OnActive and OnPassive

Author: monstermunchkin

maintenance/failover/failover.go

@@ -41,11 +41,11 @@ const Label = "github.com.pace.bricks.activepassive"
 type ActivePassive struct {
 	// OnActive will be called in case the current processes
 	// is elected to be the active one
-	OnActive func()
+	OnActive func(ctx context.Context)
 
 	// OnPassive will be called in case the current process is
 	// the passive one
-	OnPassive func()
+	OnPassive func(ctx context.Context)
 
 	// OnStop is called after the ActivePassive process stops
 	OnStop func()
@@ -203,15 +203,15 @@ func (a *ActivePassive) label(s status) string {
 func (a *ActivePassive) becomeActive(ctx context.Context) {
 	if a.setState(ctx, ACTIVE) {
 		if a.OnActive != nil {
-			a.OnActive()
+			a.OnActive(ctx)
 		}
 	}
 }
 
 func (a *ActivePassive) becomePassive(ctx context.Context) {
 	if a.setState(ctx, PASSIVE) {
 		if a.OnPassive != nil {
-			a.OnPassive()
+			a.OnPassive(ctx)
 		}
 	}
 }