Initial QSfera import

Server/pkg/registry/register.go

@@ -0,0 +1,61 @@
+package registry
+
+import (
+	"context"
+	"net/http"
+	"time"
+
+	mRegistry "go-micro.dev/v4/registry"
+
+	"github.com/qsfera/server/pkg/log"
+)
+
+// RegisterService publishes an arbitrary endpoint to the service-registry. This allows querying nodes of
+// non-micro services like reva. No health-checks are done, thus the caller is responsible for canceling.
+func RegisterService(ctx context.Context, logger log.Logger, service *mRegistry.Service, debugAddr string) error {
+	registry := GetRegistry()
+	node := service.Nodes[0]
+
+	logger.Info().Msgf("registering external service %v@%v", node.Id, node.Address)
+
+	rOpts := []mRegistry.RegisterOption{mRegistry.RegisterTTL(GetRegisterTTL())}
+	if err := registry.Register(service, rOpts...); err != nil {
+		logger.Fatal().Err(err).Msgf("Registration error for external service %v", service.Name)
+	}
+
+	t := time.NewTicker(GetRegisterInterval())
+
+	go func() {
+		// check if the service is ready
+		delay := 500 * time.Millisecond
+		for {
+			resp, err := http.DefaultClient.Get("http://" + debugAddr + "/readyz")
+			if err == nil && resp.StatusCode == http.StatusOK {
+				resp.Body.Close()
+				break
+			}
+			time.Sleep(delay)
+			delay *= 2
+		}
+		for {
+			select {
+			case <-t.C:
+				logger.Debug().Interface("service", service).Msg("refreshing external service-registration")
+				err := registry.Register(service, rOpts...)
+				if err != nil {
+					logger.Error().Err(err).Msgf("registration error for external service %v", service.Name)
+				}
+			case <-ctx.Done():
+				logger.Debug().Interface("service", service).Msg("unregistering")
+				t.Stop()
+				err := registry.Deregister(service)
+				if err != nil {
+					logger.Err(err).Msgf("Error unregistering external service %v", service.Name)
+				}
+				return
+			}
+		}
+	}()
+
+	return nil
+}