update colors

Author: zmh-program

cache.go

@@ -4,20 +4,29 @@ import (
 	"context"
 	"fmt"
 	"github.com/go-redis/redis/v8"
+	"github.com/kataras/iris/v12"
 )
 
-var client *redis.Client
+var cache *redis.Client
 
 func SetupCache() {
-	client = redis.NewClient(&redis.Options{
+	cache = redis.NewClient(&redis.Options{
 		Addr:     fmt.Sprintf("%s:%d", conf.Redis.Host, conf.Redis.Port),
 		Password: conf.Redis.Password,
 		DB:       conf.Redis.DB,
 	})
-	_, err := client.Ping(context.Background()).Result()
+	_, err := cache.Ping(context.Background()).Result()
 	if err != nil {
 		logger.Fatal("Failed to connect to Redis server: ", err)
 	} else {
 		logger.Debug("Connected to Redis server successfully.")
 	}
 }
+
+func SetCache(ctx iris.Context, key string, value interface{}) error {
+	return cache.Set(ctx.Request().Context(), key, value, conf.Redis.Expiration).Err()
+}
+
+func GetCache(ctx iris.Context, key string) (string, error) {
+	return cache.Get(ctx.Request().Context(), key).Result()
+}

conf.go

@@ -19,7 +19,7 @@ type Config struct {
 		Port       int
 		Password   string
 		DB         int
-		Expiration int
+		Expiration time.Duration
 	}
 }
 

logger.go

@@ -13,11 +13,25 @@ type Formatter struct {
 }
 
 func (f *Formatter) Format(entry *logrus.Entry) ([]byte, error) {
-	timestamp := time.Now().Format("15:04")
+	timestamp := time.Now().Format("2006/01/02 15:04")
 	level := strings.ToUpper(entry.Level.String())
 	message := entry.Message
 
-	return []byte(fmt.Sprintf("[%s] %s - %s\n", level, timestamp, message)), nil
+	var color string
+	switch entry.Level {
+	case logrus.DebugLevel:
+		color = "\u001B[33m"
+	case logrus.InfoLevel:
+		color = "\033[32m"
+	case logrus.WarnLevel:
+		color = "\033[33m"
+	case logrus.ErrorLevel, logrus.FatalLevel, logrus.PanicLevel:
+		color = "\033[31m"
+	default:
+		color = "\033[0m"
+	}
+
+	return []byte(fmt.Sprintf("%s[%s]\u001B[0m %s - %s\n", color, level, timestamp, message)), nil
 }
 
 func SetupLogger() {