56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TextSearchResult represents a single search result item.
|
||
|
|
type TextSearchResult struct {
|
||
|
|
URL string
|
||
|
|
Header string
|
||
|
|
Description string
|
||
|
|
Source string
|
||
|
|
}
|
||
|
|
|
||
|
|
// CacheKey represents the key used to store search results in the cache.
|
||
|
|
type CacheKey struct {
|
||
|
|
Query string
|
||
|
|
Page int
|
||
|
|
Safe string
|
||
|
|
Lang string
|
||
|
|
}
|
||
|
|
|
||
|
|
// ResultsCache is a thread-safe map for caching search results by composite keys.
|
||
|
|
type ResultsCache struct {
|
||
|
|
mu sync.Mutex
|
||
|
|
results map[string][]TextSearchResult
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewResultsCache creates a new ResultsCache.
|
||
|
|
func NewResultsCache() *ResultsCache {
|
||
|
|
return &ResultsCache{
|
||
|
|
results: make(map[string][]TextSearchResult),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get retrieves the results for a given key from the cache.
|
||
|
|
func (rc *ResultsCache) Get(key CacheKey) ([]TextSearchResult, bool) {
|
||
|
|
rc.mu.Lock()
|
||
|
|
defer rc.mu.Unlock()
|
||
|
|
results, exists := rc.results[rc.keyToString(key)]
|
||
|
|
return results, exists
|
||
|
|
}
|
||
|
|
|
||
|
|
// Set stores the results for a given key in the cache.
|
||
|
|
func (rc *ResultsCache) Set(key CacheKey, results []TextSearchResult) {
|
||
|
|
rc.mu.Lock()
|
||
|
|
defer rc.mu.Unlock()
|
||
|
|
rc.results[rc.keyToString(key)] = results
|
||
|
|
}
|
||
|
|
|
||
|
|
// keyToString converts a CacheKey to a string representation.
|
||
|
|
func (rc *ResultsCache) keyToString(key CacheKey) string {
|
||
|
|
return fmt.Sprintf("%s|%d|%s|%s", key.Query, key.Page, key.Safe, key.Lang)
|
||
|
|
}
|