refactor: add new methods to Insights

Signed-off-by: Mike Beaumont <[email protected]>
kumahq · Oct 26, 2023 · 9928dbb · 9928dbb
1 parent e29729f
commit 9928dbb
Show file tree

Hide file tree

Showing 9 changed files with 95 additions and 48 deletions.
diff --git a/api/generic/insights.go b/api/generic/insights.go
@@ -9,7 +9,10 @@ import (
 type Insight interface {
 	proto.Message
 	IsOnline() bool
+	// TODO Deprecated: bad
 	GetLastSubscription() Subscription
+	GetSubscription(id string) Subscription
+	GetOnlineSubscriptions() []Subscription
 	UpdateSubscription(Subscription) error
 }
 

diff --git a/api/mesh/v1alpha1/dataplane_insight_helpers.go b/api/mesh/v1alpha1/dataplane_insight_helpers.go
@@ -56,13 +56,23 @@ func (x *DataplaneInsight) IsOnline() bool {
 	return false
 }
 
-func (x *DataplaneInsight) GetSubscription(id string) (int, *DiscoverySubscription) {
-	for i, s := range x.GetSubscriptions() {
+func (x *DataplaneInsight) GetOnlineSubscriptions() []generic.Subscription {
+	var subs []generic.Subscription
+	for _, s := range x.GetSubscriptions() {
+		if s.ConnectTime != nil && s.DisconnectTime == nil {
+			subs = append(subs, s)
+		}
+	}
+	return subs
+}
+
+func (x *DataplaneInsight) GetSubscription(id string) generic.Subscription {
+	for _, s := range x.GetSubscriptions() {
 		if s.Id == id {
-			return i, s
+			return s
 		}
 	}
-	return -1, nil
+	return nil
 }
 
 func (x *DataplaneInsight) UpdateCert(generation time.Time, expiration time.Time, issuedBackend string, supportedBackends []string) error {
@@ -93,13 +103,14 @@ func (x *DataplaneInsight) UpdateSubscription(s generic.Subscription) error {
 	if !ok {
 		return errors.Errorf("invalid type %T for DataplaneInsight", s)
 	}
-	i, old := x.GetSubscription(discoverySubscription.Id)
-	if old != nil {
-		x.Subscriptions[i] = discoverySubscription
-	} else {
-		x.finalizeSubscriptions()
-		x.Subscriptions = append(x.Subscriptions, discoverySubscription)
+	for i, sub := range x.GetSubscriptions() {
+		if sub.GetId() == discoverySubscription.Id {
+			x.Subscriptions[i] = discoverySubscription
+			return nil
+		}
 	}
+	x.finalizeSubscriptions()
+	x.Subscriptions = append(x.Subscriptions, discoverySubscription)
 	return nil
 }
 

diff --git a/api/mesh/v1alpha1/dataplane_insight_helpers_test.go b/api/mesh/v1alpha1/dataplane_insight_helpers_test.go
@@ -120,8 +120,8 @@ var _ = Describe("DataplaneHelpers", func() {
 				})).To(Succeed())
 
 				// then
-				_, subscription := dataplaneInsight.GetSubscription("2")
-				Expect(subscription.DisconnectTime).ToNot(BeNil())
+				subscription := dataplaneInsight.GetSubscription("2")
+				Expect(subscription.(*DiscoverySubscription).DisconnectTime).ToNot(BeNil())
 			})
 
 			It("should return error for wrong subscription type", func() {

diff --git a/api/mesh/v1alpha1/zone_ingress_insight_helpers.go b/api/mesh/v1alpha1/zone_ingress_insight_helpers.go
@@ -9,13 +9,13 @@ import (
 
 var _ generic.Insight = &ZoneIngressInsight{}
 
-func (x *ZoneIngressInsight) GetSubscription(id string) (int, *DiscoverySubscription) {
-	for i, s := range x.GetSubscriptions() {
+func (x *ZoneIngressInsight) GetSubscription(id string) generic.Subscription {
+	for _, s := range x.GetSubscriptions() {
 		if s.Id == id {
-			return i, s
+			return s
 		}
 	}
-	return -1, nil
+	return nil
 }
 
 func (x *ZoneIngressInsight) UpdateSubscription(s generic.Subscription) error {
@@ -26,13 +26,14 @@ func (x *ZoneIngressInsight) UpdateSubscription(s generic.Subscription) error {
 	if !ok {
 		return errors.Errorf("invalid type %T for ZoneIngressInsight", s)
 	}
-	i, old := x.GetSubscription(discoverySubscription.Id)
-	if old != nil {
-		x.Subscriptions[i] = discoverySubscription
-	} else {
-		x.finalizeSubscriptions()
-		x.Subscriptions = append(x.Subscriptions, discoverySubscription)
+	for i, sub := range x.GetSubscriptions() {
+		if sub.GetId() == discoverySubscription.Id {
+			x.Subscriptions[i] = discoverySubscription
+			return nil
+		}
 	}
+	x.finalizeSubscriptions()
+	x.Subscriptions = append(x.Subscriptions, discoverySubscription)
 	return nil
 }
 
@@ -57,6 +58,16 @@ func (x *ZoneIngressInsight) IsOnline() bool {
 	return false
 }
 
+func (x *ZoneIngressInsight) GetOnlineSubscriptions() []generic.Subscription {
+	var subs []generic.Subscription
+	for _, s := range x.GetSubscriptions() {
+		if s.ConnectTime != nil && s.DisconnectTime == nil {
+			subs = append(subs, s)
+		}
+	}
+	return subs
+}
+
 func (x *ZoneIngressInsight) GetLastSubscription() generic.Subscription {
 	if len(x.GetSubscriptions()) == 0 {
 		return (*DiscoverySubscription)(nil)

diff --git a/api/mesh/v1alpha1/zone_ingress_insight_helpers_test.go b/api/mesh/v1alpha1/zone_ingress_insight_helpers_test.go
@@ -38,8 +38,8 @@ var _ = Describe("Zone Ingress Insights", func() {
 			})).To(Succeed())
 
 			// then
-			_, subscription := zoneInsight.GetSubscription("2")
-			Expect(subscription.DisconnectTime).ToNot(BeNil())
+			subscription := zoneInsight.GetSubscription("2")
+			Expect(subscription.(*mesh_proto.DiscoverySubscription).DisconnectTime).ToNot(BeNil())
 		})
 
 		It("should return error for wrong subscription type", func() {

diff --git a/api/mesh/v1alpha1/zoneegressinsight_helpers.go b/api/mesh/v1alpha1/zoneegressinsight_helpers.go
@@ -9,13 +9,13 @@ import (
 
 var _ generic.Insight = &ZoneEgressInsight{}
 
-func (x *ZoneEgressInsight) GetSubscription(id string) (int, *DiscoverySubscription) {
-	for i, s := range x.GetSubscriptions() {
+func (x *ZoneEgressInsight) GetSubscription(id string) generic.Subscription {
+	for _, s := range x.GetSubscriptions() {
 		if s.Id == id {
-			return i, s
+			return s
 		}
 	}
-	return -1, nil
+	return nil
 }
 
 func (x *ZoneEgressInsight) UpdateSubscription(s generic.Subscription) error {
@@ -26,13 +26,14 @@ func (x *ZoneEgressInsight) UpdateSubscription(s generic.Subscription) error {
 	if !ok {
 		return errors.Errorf("invalid type %T for ZoneEgressInsight", s)
 	}
-	i, old := x.GetSubscription(discoverySubscription.Id)
-	if old != nil {
-		x.Subscriptions[i] = discoverySubscription
-	} else {
-		x.finalizeSubscriptions()
-		x.Subscriptions = append(x.Subscriptions, discoverySubscription)
+	for i, sub := range x.GetSubscriptions() {
+		if sub.GetId() == discoverySubscription.Id {
+			x.Subscriptions[i] = discoverySubscription
+			return nil
+		}
 	}
+	x.finalizeSubscriptions()
+	x.Subscriptions = append(x.Subscriptions, discoverySubscription)
 	return nil
 }
 
@@ -57,6 +58,16 @@ func (x *ZoneEgressInsight) IsOnline() bool {
 	return false
 }
 
+func (x *ZoneEgressInsight) GetOnlineSubscriptions() []generic.Subscription {
+	var subs []generic.Subscription
+	for _, s := range x.GetSubscriptions() {
+		if s.ConnectTime != nil && s.DisconnectTime == nil {
+			subs = append(subs, s)
+		}
+	}
+	return subs
+}
+
 func (x *ZoneEgressInsight) GetLastSubscription() generic.Subscription {
 	if len(x.GetSubscriptions()) == 0 {
 		return (*DiscoverySubscription)(nil)

diff --git a/api/mesh/v1alpha1/zoneegressinsight_helpers_test.go b/api/mesh/v1alpha1/zoneegressinsight_helpers_test.go
@@ -38,8 +38,8 @@ var _ = Describe("Zone Egress Insights", func() {
 			})).To(Succeed())
 
 			// then
-			_, subscription := zoneInsight.GetSubscription("2")
-			Expect(subscription.DisconnectTime).ToNot(BeNil())
+			subscription := zoneInsight.GetSubscription("2")
+			Expect(subscription.(*mesh_proto.DiscoverySubscription)).ToNot(BeNil())
 		})
 
 		It("should return error for wrong subscription type", func() {

diff --git a/api/system/v1alpha1/zone_insight_helpers.go b/api/system/v1alpha1/zone_insight_helpers.go
@@ -19,13 +19,13 @@ func NewSubscriptionStatus() *KDSSubscriptionStatus {
 	}
 }
 
-func (x *ZoneInsight) GetSubscription(id string) (int, *KDSSubscription) {
-	for i, s := range x.GetSubscriptions() {
+func (x *ZoneInsight) GetSubscription(id string) generic.Subscription {
+	for _, s := range x.GetSubscriptions() {
 		if s.Id == id {
-			return i, s
+			return s
 		}
 	}
-	return -1, nil
+	return nil
 }
 
 func (x *ZoneInsight) GetLastSubscription() generic.Subscription {
@@ -44,6 +44,16 @@ func (x *ZoneInsight) IsOnline() bool {
 	return false
 }
 
+func (x *ZoneInsight) GetOnlineSubscriptions() []generic.Subscription {
+	var subs []generic.Subscription
+	for _, s := range x.GetSubscriptions() {
+		if s.ConnectTime != nil && s.DisconnectTime == nil {
+			subs = append(subs, s)
+		}
+	}
+	return subs
+}
+
 func (x *KDSSubscription) SetDisconnectTime(time time.Time) {
 	x.DisconnectTime = timestamppb.New(time)
 }
@@ -64,13 +74,14 @@ func (x *ZoneInsight) UpdateSubscription(s generic.Subscription) error {
 	if !ok {
 		return errors.Errorf("invalid type %T for ZoneInsight", s)
 	}
-	i, old := x.GetSubscription(kdsSubscription.Id)
-	if old != nil {
-		x.Subscriptions[i] = kdsSubscription
-	} else {
-		x.finalizeSubscriptions()
-		x.Subscriptions = append(x.Subscriptions, kdsSubscription)
+	for i, sub := range x.GetSubscriptions() {
+		if sub.GetId() == kdsSubscription.Id {
+			x.Subscriptions[i] = kdsSubscription
+			return nil
+		}
 	}
+	x.finalizeSubscriptions()
+	x.Subscriptions = append(x.Subscriptions, kdsSubscription)
 	return nil
 }
 

diff --git a/api/system/v1alpha1/zone_insight_helpers_test.go b/api/system/v1alpha1/zone_insight_helpers_test.go
@@ -38,8 +38,8 @@ var _ = Describe("Zone Insights", func() {
 			})).To(Succeed())
 
 			// then
-			_, subscription := zoneInsight.GetSubscription("2")
-			Expect(subscription.DisconnectTime).ToNot(BeNil())
+			subscription := zoneInsight.GetSubscription("2")
+			Expect(subscription.(*system_proto.KDSSubscription).DisconnectTime).ToNot(BeNil())
 		})
 
 		It("should return error for wrong subscription type", func() {