From eef7e79ec38bfd8c49b22d7f5295179a03dd79de Mon Sep 17 00:00:00 2001
From: mtryfoss <morten@tryfoss.no>
Date: Tue, 30 Jun 2020 21:34:13 +0200
Subject: [PATCH] Add goroutine to check if Asterisk entity ID has changed.
 (#33)

* Add goroutine to check if Asterisk entity ID has changed.

* Add comment
---
 proxy/types.go   |  2 ++
 server/server.go | 28 ++++++++++++++++++++++++++++
 2 files changed, 30 insertions(+)

diff --git a/proxy/types.go b/proxy/types.go
index 6ec6c12..ac6faf1 100644
--- a/proxy/types.go
+++ b/proxy/types.go
@@ -10,6 +10,8 @@ import (
 
 // AnnouncementInterval is the amount of time to wait between periodic service availability announcements
 var AnnouncementInterval = time.Minute
+// EntityCheckInterval is the interval between checks against Asterisk entity ID
+var EntityCheckInterval = time.Second * 10
 
 // Announcement describes the structure of an ARI proxy's announcement of availability on the network.  These are sent periodically and upon request (by a Ping).
 type Announcement struct {
diff --git a/server/server.go b/server/server.go
index 22cd5e0..34e651e 100644
--- a/server/server.go
+++ b/server/server.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"time"
+	"os"
 
 	"github.com/CyCoreSystems/ari-proxy/v5/proxy"
 	"github.com/CyCoreSystems/ari-proxy/v5/server/dialog"
@@ -220,6 +221,9 @@ func (s *Server) listen(ctx context.Context) error {
 	// Run the event handler
 	go s.runEventHandler(ctx)
 
+	// Run the entity check handler
+	go s.runEntityChecker(ctx)
+
 	// TODO: run the dialog cleanup routine (remove bindings for entities which no longer exist)
 	// go s.runDialogCleaner(ctx)
 
@@ -233,6 +237,30 @@ func (s *Server) listen(ctx context.Context) error {
 	return ctx.Err()
 }
 
+// runEntityChecker runs the periodic check againt Asterisk entity id
+func (s *Server) runEntityChecker(ctx context.Context) {
+	ticker := time.NewTicker(proxy.EntityCheckInterval)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-ctx.Done():
+			return
+		case <-ticker.C:
+			info, err := s.ari.Asterisk().Info(nil)
+			if err != nil {
+				s.Log.Error("failed to get info from Asterisk", "error", err)
+				continue
+			}
+			if s.AsteriskID != info.SystemInfo.EntityID {
+				s.Log.Warn("system entitiy id changed", "old", s.AsteriskID, "new", info.SystemInfo.EntityID)
+				// We need to exit with non-zero to make sure systemd restarts when service defined with Restart=on-failure
+				os.Exit(1)
+			}
+		}
+	}
+}
+
 // runAnnouncer runs the periodic discovery announcer
 func (s *Server) runAnnouncer(ctx context.Context) {
 	ticker := time.NewTicker(proxy.AnnouncementInterval)