Skip to content

Commit

Permalink
ConfigListener needs to be manually restarted after HotRodRolling upg…
Browse files Browse the repository at this point in the history
…rades. Closes #2160
  • Loading branch information
ryanemerson committed Oct 4, 2024
1 parent b40df50 commit 41adc0c
Show file tree
Hide file tree
Showing 2 changed files with 82 additions and 2 deletions.
9 changes: 7 additions & 2 deletions launcher/listener/listener.go
Original file line number Diff line number Diff line change
Expand Up @@ -57,9 +57,12 @@ func New(ctx context.Context, p Parameters) {
}

infinispan := &v1.Infinispan{}
if err = k8s.Client.Get(ctx, types.NamespacedName{Namespace: p.Namespace, Name: p.Cluster}, infinispan); err != nil {
log.Fatalf("unable to load Infinispan cluster %s in Namespace %s: %v", p.Cluster, p.Namespace, err)
loadInfinispan := func() {
if err = k8s.Client.Get(ctx, types.NamespacedName{Namespace: p.Namespace, Name: p.Cluster}, infinispan); err != nil {
log.Fatalf("unable to load Infinispan cluster %s in Namespace %s: %v", p.Cluster, p.Namespace, err)
}
}
loadInfinispan()

secret := &corev1.Secret{}
if err = k8s.Client.Get(ctx, types.NamespacedName{Namespace: p.Namespace, Name: infinispan.GetAdminSecretName()}, secret); err != nil {
Expand Down Expand Up @@ -131,6 +134,8 @@ func New(ctx context.Context, p Parameters) {
containerSse.ReconnectStrategy = backoff.NewConstantBackOff(time.Second)
containerSse.ReconnectNotify = func(e error, t time.Duration) {
log.Warnf("Cache stream connection lost. Reconnecting: %v", e)
// Reload the Infinispan CR as the StatefulSet may have been updated due to a Hot Rod rolling upgrade
loadInfinispan()
}

go func() {
Expand Down
75 changes: 75 additions & 0 deletions test/e2e/infinispan/upgrade_operand_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,9 @@ import (
"testing"

ispnv1 "github.com/infinispan/infinispan-operator/api/v1"
"github.com/infinispan/infinispan-operator/pkg/infinispan/version"
"github.com/infinispan/infinispan-operator/pkg/kubernetes"
"github.com/infinispan/infinispan-operator/pkg/mime"
"github.com/infinispan/infinispan-operator/pkg/reconcile/pipeline/infinispan/handler/provision"
tutils "github.com/infinispan/infinispan-operator/test/e2e/utils"
"k8s.io/utils/pointer"
Expand Down Expand Up @@ -173,6 +175,79 @@ func TestOperandCVEGracefulShutdown(t *testing.T) {
assert.EqualValues(t, 1, ss.Status.ObservedGeneration)
}

func TestOperandHotRodRolling(t *testing.T) {
defer testKube.CleanNamespaceAndLogOnPanic(t, tutils.Namespace)
versionManager := tutils.VersionManager()

// Create Infinispan Cluster using first Operand and then upgrade to the next Operand that is not marked as a CVE as
// we want to ensure that a StatefulSet rolling upgrade does not occur.
// The actual Operand versions deployed should not impact the test as we're only verifying the HR upgrade procedure
// which is controlled by the Operator
replicas := 1
operands := versionManager.Operands
startingOperand := operands[0]
var upgradeOperand *version.Operand
for i := 1; i < len(operands); i++ {
op := operands[i]
if !op.CVE {
upgradeOperand = operands[i]
break
}
}
assert.NotNil(t, upgradeOperand)

ispn := tutils.DefaultSpec(t, testKube, func(i *ispnv1.Infinispan) {
i.Spec.ConfigListener.Enabled = true
i.Spec.Replicas = int32(replicas)
i.Spec.Version = startingOperand.Ref()
i.Spec.Upgrades = &ispnv1.InfinispanUpgradesSpec{
Type: ispnv1.UpgradeTypeHotRodRolling,
}
})

originalStatefulSetName := ispn.Name
testKube.CreateInfinispan(ispn, tutils.Namespace)
testKube.WaitForInfinispanPods(replicas, tutils.SinglePodTimeout, ispn.Name, tutils.Namespace)
testKube.WaitForInfinispanCondition(ispn.Name, ispn.Namespace, ispnv1.ConditionWellFormed)
testKube.WaitForInfinispanState(ispn.Name, ispn.Namespace, func(i *ispnv1.Infinispan) bool {
return i.IsConditionTrue(ispnv1.ConditionWellFormed) &&
i.Status.Operand.Version == startingOperand.Ref() &&
i.Status.Operand.Image == startingOperand.Image &&
i.Status.Operand.Phase == ispnv1.OperandPhaseRunning &&
i.Status.StatefulSetName == originalStatefulSetName
})

tutils.ExpectNoError(
testKube.UpdateInfinispan(ispn, func() {
ispn.Spec.Version = upgradeOperand.Ref()
ispn.Default()
}),
)

newStatefulSetName := ispn.Name + "-1"
testKube.WaitForStateFulSetRemoval(originalStatefulSetName, tutils.Namespace)
testKube.WaitForInfinispanPodsCreatedBy(0, tutils.SinglePodTimeout, originalStatefulSetName, tutils.Namespace)
testKube.WaitForStateFulSet(newStatefulSetName, tutils.Namespace)
testKube.WaitForInfinispanPodsCreatedBy(replicas, tutils.SinglePodTimeout, newStatefulSetName, tutils.Namespace)
testKube.WaitForInfinispanCondition(ispn.Name, ispn.Namespace, ispnv1.ConditionWellFormed)
testKube.WaitForInfinispanState(ispn.Name, ispn.Namespace, func(i *ispnv1.Infinispan) bool {
return i.IsConditionTrue(ispnv1.ConditionWellFormed) &&
i.Status.Operand.Version == upgradeOperand.Ref() &&
i.Status.Operand.Image == upgradeOperand.Image &&
i.Status.Operand.Phase == ispnv1.OperandPhaseRunning &&
i.Status.StatefulSetName == newStatefulSetName
})

// Ensure that the Cache ConfigListener is still able to receive updates after the upgrade has completed by creating
// a cache on the server and waiting for it's corresponding Cache CR to be created.
client := tutils.HTTPClientForClusterWithVersionManager(ispn, testKube, versionManager)
cacheName := "config-listener"
cacheConfig := "localCache: \n memory: \n maxCount: \"100\"\n"
cacheHelper := tutils.NewCacheHelper(cacheName, client)
cacheHelper.Create(cacheConfig, mime.ApplicationYaml)
testKube.WaitForCacheConditionReady(cacheName, ispn.Name, tutils.Namespace)
}

// TestOperandCVEHotRodRolling tests that Operands marked as CVE releases, with the same upstream version as the currently
// installed operand, only result in a StatefulSet rolling upgrade
func TestOperandCVEHotRodRolling(t *testing.T) {
Expand Down

0 comments on commit 41adc0c

Please sign in to comment.