Initial QSfera import

Server/pkg/checks/checkgrpc.go

@@ -0,0 +1,27 @@
+package checks
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/qsfera/server/pkg/handlers"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+// NewGRPCCheck checks the reachability of a grpc server.
+func NewGRPCCheck(address string) func(context.Context) error {
+	return func(_ context.Context) error {
+		address, err := handlers.FailSaveAddress(address)
+		if err != nil {
+			return err
+		}
+
+		conn, err := grpc.NewClient(address, grpc.WithTransportCredentials(insecure.NewCredentials()))
+		if err != nil {
+			return fmt.Errorf("could not connect to grpc server: %v", err)
+		}
+		_ = conn.Close()
+		return nil
+	}
+}

Server/pkg/checks/checkhttp.go

@@ -0,0 +1,35 @@
+package checks
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/qsfera/server/pkg/handlers"
+)
+
+// NewHTTPCheck checks the reachability of a http server.
+func NewHTTPCheck(url string) func(context.Context) error {
+	return func(_ context.Context) error {
+		url, err := handlers.FailSaveAddress(url)
+		if err != nil {
+			return err
+		}
+
+		if !strings.HasPrefix(url, "http://") && !strings.HasPrefix(url, "https://") {
+			url = "http://" + url
+		}
+
+		c := http.Client{
+			Timeout: 3 * time.Second,
+		}
+		resp, err := c.Get(url)
+		if err != nil {
+			return fmt.Errorf("could not connect to http server: %v", err)
+		}
+		_ = resp.Body.Close()
+		return nil
+	}
+}

Server/pkg/checks/checknats.go

@@ -0,0 +1,23 @@
+package checks
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/nats-io/nats.go"
+)
+
+// NewNatsCheck checks the reachability of a nats server.
+func NewNatsCheck(natsCluster string, options ...nats.Option) func(context.Context) error {
+	return func(_ context.Context) error {
+		n, err := nats.Connect(natsCluster, options...)
+		if err != nil {
+			return fmt.Errorf("could not connect to nats server: %v", err)
+		}
+		defer n.Close()
+		if n.Status() != nats.CONNECTED {
+			return fmt.Errorf("nats server not connected")
+		}
+		return nil
+	}
+}

Server/pkg/checks/checktcp.go

@@ -0,0 +1,31 @@
+package checks
+
+import (
+	"context"
+	"net"
+	"time"
+
+	"github.com/qsfera/server/pkg/handlers"
+)
+
+// NewTCPCheck returns a check that connects to a given tcp endpoint.
+func NewTCPCheck(address string) func(context.Context) error {
+	return func(_ context.Context) error {
+		address, err := handlers.FailSaveAddress(address)
+		if err != nil {
+			return err
+		}
+
+		conn, err := net.DialTimeout("tcp", address, 3*time.Second)
+		if err != nil {
+			return err
+		}
+
+		err = conn.Close()
+		if err != nil {
+			return err
+		}
+
+		return nil
+	}
+}