mirror of
https://github.com/lukaszraczylo/gohoarder.git
synced 2026-06-06 22:59:29 +00:00
lukaszraczylo
6b037a92b4
- [x] Reorder struct fields across codebase for consistency - [x] Add analytics event handlers and tests - [x] Add authentication API key management handlers and tests - [x] Add pre-warming control handlers and tests - [x] Implement S3 storage backend with tests - [x] Implement SMB/CIFS storage backend with tests - [x] Add CDN middleware tests - [x] Integrate analytics tracking into cache manager - [x] Add S3 and SMB storage initialization in app setup - [x] Add CDN caching to proxy handlers - [x] Remove distributed locking (Redis lock manager) - [x] Remove proxy common package and utilities - [x] Remove standalone HTTP server package - [x] Remove logger middleware - [x] Simplify error handling utilities - [x] Update config with S3 and SMB options - [x] Update cache manager signature to include analytics
110 lines
2.3 KiB
Go
110 lines
2.3 KiB
Go
package auth
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// ValidationResult represents a cached credential validation result
|
|
type ValidationResult struct {
|
|
ExpiresAt time.Time
|
|
Reason string
|
|
Allowed bool
|
|
}
|
|
|
|
// ValidationCache caches credential validation results to reduce upstream checks
|
|
type ValidationCache struct {
|
|
cache map[string]*ValidationResult
|
|
mu sync.RWMutex
|
|
ttl time.Duration
|
|
}
|
|
|
|
// NewValidationCache creates a new validation cache
|
|
func NewValidationCache(ttl time.Duration) *ValidationCache {
|
|
vc := &ValidationCache{
|
|
cache: make(map[string]*ValidationResult),
|
|
ttl: ttl,
|
|
}
|
|
|
|
// Start cleanup goroutine
|
|
go vc.cleanupExpired()
|
|
|
|
return vc
|
|
}
|
|
|
|
// Get retrieves a validation result from cache
|
|
// Returns (allowed bool, cached bool, reason string)
|
|
func (vc *ValidationCache) Get(credHash, packageURL string) (bool, bool, string) {
|
|
vc.mu.RLock()
|
|
defer vc.mu.RUnlock()
|
|
|
|
key := credHash + ":" + packageURL
|
|
result, exists := vc.cache[key]
|
|
|
|
if !exists {
|
|
return false, false, ""
|
|
}
|
|
|
|
// Check if expired
|
|
if time.Now().After(result.ExpiresAt) {
|
|
return false, false, ""
|
|
}
|
|
|
|
return result.Allowed, true, result.Reason
|
|
}
|
|
|
|
// Set stores a validation result in cache
|
|
func (vc *ValidationCache) Set(credHash, packageURL string, allowed bool, reason string) {
|
|
vc.mu.Lock()
|
|
defer vc.mu.Unlock()
|
|
|
|
key := credHash + ":" + packageURL
|
|
vc.cache[key] = &ValidationResult{
|
|
Allowed: allowed,
|
|
ExpiresAt: time.Now().Add(vc.ttl),
|
|
Reason: reason,
|
|
}
|
|
}
|
|
|
|
// Invalidate removes a specific entry from cache
|
|
func (vc *ValidationCache) Invalidate(credHash, packageURL string) {
|
|
vc.mu.Lock()
|
|
defer vc.mu.Unlock()
|
|
|
|
key := credHash + ":" + packageURL
|
|
delete(vc.cache, key)
|
|
}
|
|
|
|
// InvalidateAll clears the entire cache
|
|
func (vc *ValidationCache) InvalidateAll() {
|
|
vc.mu.Lock()
|
|
defer vc.mu.Unlock()
|
|
|
|
vc.cache = make(map[string]*ValidationResult)
|
|
}
|
|
|
|
// Size returns the number of cached entries
|
|
func (vc *ValidationCache) Size() int {
|
|
vc.mu.RLock()
|
|
defer vc.mu.RUnlock()
|
|
|
|
return len(vc.cache)
|
|
}
|
|
|
|
// cleanupExpired removes expired entries periodically
|
|
func (vc *ValidationCache) cleanupExpired() {
|
|
ticker := time.NewTicker(1 * time.Minute)
|
|
defer ticker.Stop()
|
|
|
|
for range ticker.C {
|
|
vc.mu.Lock()
|
|
now := time.Now()
|
|
for key, result := range vc.cache {
|
|
if now.After(result.ExpiresAt) {
|
|
delete(vc.cache, key)
|
|
}
|
|
}
|
|
vc.mu.Unlock()
|
|
}
|
|
}
|