Migrate old cache system to Redis

handler/signup/signup.go

@@ -3,13 +3,15 @@ package signupHandler
 import (
 	"bytes"
 	"context"
+	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/fossyy/filekeeper/app"
 	"github.com/fossyy/filekeeper/utils"
 	"github.com/fossyy/filekeeper/view/client/email"
 	signupView "github.com/fossyy/filekeeper/view/client/signup"
+	"github.com/redis/go-redis/v9"
 	"net/http"
-	"sync"
 	"time"
 
 	"github.com/fossyy/filekeeper/types"
@@ -20,41 +22,9 @@ import (
 type UnverifiedUser struct {
 	User       *models.User
 	Code       string
-	mu         sync.Mutex
 	CreateTime time.Time
 }
 
-var VerifyUser map[string]*UnverifiedUser
-var VerifyEmail map[string]string
-
-func init() {
-	VerifyUser = make(map[string]*UnverifiedUser)
-	VerifyEmail = make(map[string]string)
-
-	ticker := time.NewTicker(time.Minute)
-	go func() {
-		for {
-			<-ticker.C
-			currentTime := time.Now()
-			cacheClean := 0
-			cleanID := utils.GenerateRandomString(10)
-			app.Server.Logger.Info(fmt.Sprintf("Cache cleanup [signup] [%s] initiated at %02d:%02d:%02d", cleanID, currentTime.Hour(), currentTime.Minute(), currentTime.Second()))
-
-			for _, data := range VerifyUser {
-				data.mu.Lock()
-				if currentTime.Sub(data.CreateTime) > time.Minute*10 {
-					delete(VerifyUser, data.Code)
-					delete(VerifyEmail, data.User.Email)
-					cacheClean++
-				}
-				data.mu.Unlock()
-			}
-
-			app.Server.Logger.Info(fmt.Sprintf("Cache cleanup [signup] [%s] completed: %d entries removed. Finished at %s", cleanID, cacheClean, time.Since(currentTime)))
-		}
-	}()
-}
-
 func GET(w http.ResponseWriter, r *http.Request) {
 	component := signupView.Main("Filekeeper - Sign up Page", types.Message{
 		Code:    3,
@@ -136,16 +106,18 @@ func verifyEmail(user *models.User) error {
 	var buffer bytes.Buffer
 	var code string
 
-	code = VerifyEmail[user.Email]
+	storedCode, err := app.Server.Cache.GetCache(context.Background(), "VerificationCode:"+user.Email)
-	userData, ok := VerifyUser[code]
+	if err != nil {
-
+		if errors.Is(err, redis.Nil) {
-	if !ok {
+			code = utils.GenerateRandomString(64)
-		code = utils.GenerateRandomString(64)
+		} else {
+			return err
+		}
 	} else {
-		code = userData.Code
+		code = storedCode
 	}
 
-	err := emailView.RegistrationEmail(user.Username, fmt.Sprintf("https://%s/signup/verify/%s", utils.Getenv("DOMAIN"), code)).Render(context.Background(), &buffer)
+	err = emailView.RegistrationEmail(user.Username, fmt.Sprintf("https://%s/signup/verify/%s", utils.Getenv("DOMAIN"), code)).Render(context.Background(), &buffer)
 	if err != nil {
 		return err
 	}
@@ -155,13 +127,25 @@ func verifyEmail(user *models.User) error {
 		Code:       code,
 		CreateTime: time.Now(),
 	}
+	newUnverifiedUser, err := json.Marshal(unverifiedUser)
+	if err != nil {
+		return err
+	}
 
-	VerifyUser[code] = &unverifiedUser
+	err = app.Server.Cache.SetCache(context.Background(), "UnverifiedUser:"+code, newUnverifiedUser, 10*time.Minute)
-	VerifyEmail[user.Email] = code
+	if err != nil {
+		return err
+	}
+
+	err = app.Server.Cache.SetCache(context.Background(), "VerificationCode:"+user.Email, code, 10*time.Minute)
+	if err != nil {
+		return err
+	}
 
 	err = app.Server.Mail.Send(user.Email, "Account Registration Verification", buffer.String())
 	if err != nil {
 		return err
 	}
+
 	return nil
 }

handler/signup/verify/verify.go

@@ -1,28 +1,45 @@
 package signupVerifyHandler
 
 import (
+	"context"
+	"encoding/json"
+	"errors"
 	"github.com/fossyy/filekeeper/app"
+	signupHandler "github.com/fossyy/filekeeper/handler/signup"
 	signupView "github.com/fossyy/filekeeper/view/client/signup"
+	"github.com/redis/go-redis/v9"
 	"net/http"
 
-	signupHandler "github.com/fossyy/filekeeper/handler/signup"
 	"github.com/fossyy/filekeeper/types"
 )
 
 func GET(w http.ResponseWriter, r *http.Request) {
 	code := r.PathValue("code")
-	data, ok := signupHandler.VerifyUser[code]
 
-	if !ok {
+	userDataStr, err := app.Server.Cache.GetCache(context.Background(), "UnverifiedUser:"+code)
-		w.WriteHeader(http.StatusNotFound)
+	if err != nil {
+		if errors.Is(err, redis.Nil) {
+			w.WriteHeader(http.StatusNotFound)
+			return
+		}
+		app.Server.Logger.Error(err.Error())
+		w.WriteHeader(http.StatusInternalServerError)
 		return
 	}
 
-	err := app.Server.Database.CreateUser(data.User)
+	var unverifiedUser signupHandler.UnverifiedUser
+	err = json.Unmarshal([]byte(userDataStr), &unverifiedUser)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		app.Server.Logger.Error(err.Error())
+		return
+	}
+
+	err = app.Server.Database.CreateUser(unverifiedUser.User)
 	if err != nil {
 		component := signupView.Main("Filekeeper - Sign up Page", types.Message{
 			Code:    0,
-			Message: "Email or Username has been registered",
+			Message: "Email or Username has already been registered",
 		})
 		err := component.Render(r.Context(), w)
 		if err != nil {
@@ -33,11 +50,17 @@ func GET(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	delete(signupHandler.VerifyUser, code)
+	err = app.Server.Cache.DeleteCache(context.Background(), "UnverifiedUser:"+code)
-	delete(signupHandler.VerifyEmail, data.User.Email)
+	if err != nil {
+		app.Server.Logger.Error(err.Error())
+	}
+
+	err = app.Server.Cache.DeleteCache(context.Background(), "VerificationCode:"+unverifiedUser.User.Email)
+	if err != nil {
+		app.Server.Logger.Error(err.Error())
+	}
 
 	component := signupView.VerifySuccess("Filekeeper - Verify Page")
-
 	err = component.Render(r.Context(), w)
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)