redis.go

package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os"
	"strings"

	"github.com/redis/go-redis/v9"
)

var (
	ctx = context.Background()
)

func redisHandler(w http.ResponseWriter, r *http.Request) {
	service := r.URL.Query().Get("service")
	redisRoute := strings.ReplaceAll(service, "/", "")
	redisConnectionStr := fmt.Sprintf("%s:6379", redisRoute)
	log.Print(fmt.Sprintf("Using %s as the connstring", redisConnectionStr))
	fmt.Fprintf(w, redisConnector(redisConnectionStr, redisRoute))
}

func cleanRedisOutput(r *redis.StringCmd) string {
	redistoString := r.String()
	cleanString := strings.ReplaceAll(redistoString, "get ", "")
	redisVals := strings.ReplaceAll(cleanString, " ", "")
	return redisVals
}

func redisConnector(connectionString string, version string) string {
	client := redis.NewClient(&redis.Options{
		Addr:     connectionString,
		Password: "",
		DB:       0,
	})

	if err := client.Ping(ctx).Err(); err != nil {
		panic(err)
	}

	for _, e := range os.Environ() {
		pair := strings.SplitN(e, "=", 2)
		err := client.Set(ctx, pair[0], pair[1], 0).Err()
		if err != nil {
			log.Print(err)
		}
	}

	var cursor uint64
	results, _, _ := client.Scan(ctx, cursor, "LAGOON_*", 100).Result()

	var values []string
	for _, res := range results {
		redisKeyVals := client.Get(ctx, res)
		redisVals := cleanRedisOutput(redisKeyVals)
		values = append(values, redisVals)
	}

	keyVals := connectorKeyValues(values)
	host := fmt.Sprintf(`"SERVICE_HOST=%s"`, version)
	redisData := host + "\n" + keyVals
	return redisData
}