Finished

lib/redis.go

@@ -0,0 +1,48 @@
+package lib
+
+import (
+	"context"
+	"log"
+	"os"
+
+	"github.com/joho/godotenv"
+	"github.com/redis/go-redis/v9"
+)
+
+var ctx = context.Background()
+
+var RedisClient *redis.Client
+
+func NewClient() *redis.Client {
+	godotenv.Load(".env")
+	redis_host := os.Getenv("REDIS_HOST")
+	redis_password := os.Getenv("REDIS_PASSWORD")
+
+	log.Printf("Connecting to Redis at %s", redis_host)
+	return redis.NewClient(&redis.Options{
+		Addr:     redis_host,
+		Password: redis_password,
+		DB:       0,
+	})
+}
+
+func Publish(client *redis.Client, channel string, message string) error {
+	if client == nil {
+		client = NewClient()
+	}
+
+	return client.Publish(ctx, channel, message).Err()
+}
+
+func Subscribe(client *redis.Client, channel string) (*redis.PubSub, string) {
+	if client == nil {
+		client = NewClient()
+	}
+
+	pubsub := client.Subscribe(ctx, channel)
+	_, err := pubsub.Receive(ctx)
+	if err != nil {
+		log.Fatalf("Error receiving subscription: %v", err)
+	}
+	return pubsub, channel
+}