add redis cache

Author: zmh-program

cache.go

@@ -1 +1,23 @@
 package main
+
+import (
+	"context"
+	"fmt"
+	"github.com/go-redis/redis/v8"
+)
+
+var client *redis.Client
+
+func SetupCache() {
+	client = redis.NewClient(&redis.Options{
+		Addr:     fmt.Sprintf("%s:%d", conf.Redis.Host, conf.Redis.Port),
+		Password: conf.Redis.Password,
+		DB:       conf.Redis.DB,
+	})
+	pong, err := client.Ping(context.Background()).Result()
+	if err != nil {
+		logger.Fatal("Failed to connect to Redis server: ", err)
+	} else {
+		logger.Debug("Connected to Redis server successfully.")
+	}
+}

conf.go

@@ -15,9 +15,11 @@ type Config struct {
 		Port int
 	}
 	Redis struct {
-		Host string
-		Port int
-		DB   int
+		Host       string
+		Port       int
+		Password   string
+		DB         int
+		Expiration int
 	}
 }
 

config.yaml

@@ -4,4 +4,5 @@ server:
 redis:
   host: localhost
   port: 6379
-  db:   0
+  password: ""
+  db: 0

main.go

@@ -15,6 +15,7 @@ func main() {
 	SetupLogger()
 	ReadConfig()
 	ReadToken()
+	SetupCache()
 
 	app.Listen(fmt.Sprintf(":%d", conf.Server.Port))
 }