91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package user
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/fossyy/filekeeper/utils"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/fossyy/filekeeper/db"
|
|
"github.com/fossyy/filekeeper/logger"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type Cache struct {
|
|
users map[string]*UserWithExpired
|
|
mu sync.Mutex
|
|
}
|
|
|
|
type UserWithExpired struct {
|
|
UserID uuid.UUID
|
|
Username string
|
|
Email string
|
|
Password string
|
|
AccessAt time.Time
|
|
}
|
|
|
|
var log *logger.AggregatedLogger
|
|
var UserCache *Cache
|
|
|
|
// TESTTING VAR
|
|
var database db.Database
|
|
|
|
func init() {
|
|
log = logger.Logger()
|
|
database = db.NewPostgresDB(utils.Getenv("DB_USERNAME"), utils.Getenv("DB_PASSWORD"), utils.Getenv("DB_HOST"), utils.Getenv("DB_PORT"), utils.Getenv("DB_NAME"))
|
|
|
|
UserCache = &Cache{users: make(map[string]*UserWithExpired)}
|
|
ticker := time.NewTicker(time.Hour * 8)
|
|
|
|
go func() {
|
|
for {
|
|
<-ticker.C
|
|
currentTime := time.Now()
|
|
cacheClean := 0
|
|
log.Info(fmt.Sprintf("Cache cleanup initiated at %02d:%02d:%02d", currentTime.Hour(), currentTime.Minute(), currentTime.Second()))
|
|
|
|
UserCache.mu.Lock()
|
|
for _, user := range UserCache.users {
|
|
if currentTime.Sub(user.AccessAt) > time.Hour*8 {
|
|
delete(UserCache.users, user.Email)
|
|
cacheClean++
|
|
}
|
|
}
|
|
UserCache.mu.Unlock()
|
|
|
|
log.Info(fmt.Sprintf("Cache cleanup completed: %d entries removed. Finished at %s", cacheClean, time.Since(currentTime)))
|
|
}
|
|
}()
|
|
}
|
|
|
|
func Get(email string) (*UserWithExpired, error) {
|
|
UserCache.mu.Lock()
|
|
defer UserCache.mu.Unlock()
|
|
|
|
if user, ok := UserCache.users[email]; ok {
|
|
return user, nil
|
|
}
|
|
|
|
userData, err := database.GetUser(email)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
UserCache.users[email] = &UserWithExpired{
|
|
UserID: userData.UserID,
|
|
Username: userData.Username,
|
|
Email: userData.Email,
|
|
Password: userData.Password,
|
|
AccessAt: time.Now(),
|
|
}
|
|
|
|
return UserCache.users[email], nil
|
|
}
|
|
|
|
func DeleteCache(email string) {
|
|
UserCache.mu.Lock()
|
|
defer UserCache.mu.Unlock()
|
|
|
|
delete(UserCache.users, email)
|
|
}
|