bagas/filekeeper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Migrate old cache system to Redis

Browse Source
This commit is contained in:
Bagas Aulia Rezki
2024-09-08 20:04:35 +07:00
parent 6438e1812d
commit e57b5a0060
2 changed files with 58 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
handler/signup/signup.go
View File

@ -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]
userData, ok := VerifyUser[code]
if !ok {
code = utils.GenerateRandomString(64)
storedCode, err := app.Server.Cache.GetCache(context.Background(), "VerificationCode:"+user.Email)
if err != nil {
if errors.Is(err, redis.Nil) {
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
VerifyEmail[user.Email] = code
err = app.Server.Cache.SetCache(context.Background(), "UnverifiedUser:"+code, newUnverifiedUser, 10*time.Minute)
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
}

41
handler/signup/verify/verify.go
View File

@ -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 {
w.WriteHeader(http.StatusNotFound)
userDataStr, err := app.Server.Cache.GetCache(context.Background(), "UnverifiedUser:"+code)
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)
delete(signupHandler.VerifyEmail, data.User.Email)
err = app.Server.Cache.DeleteCache(context.Background(), "UnverifiedUser:"+code)
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)