James Tombleson
ada453e08a
* updated db, added dto for ListSources, and added delete source * updated from model > models * updated to models * sources now sends back a standard message * updated subscription routes to have beter logid and swagger details * moved the dto objects back to modles given they are not bound to the database * cleaned up how we return the error * cleaned up swag and updated models to take from the base apistatusmodel. less human errors this way * cleaned up swag and updated models * swag updated * updated queue to return a router and also renamed it as it will hold all queue info later on * removed config tag * added subscription details route * article routes have been moved to support dto * updated discordwebhooks to use dto * updated discordwebhookqueue to return details on the items via dto * removed the example routes * updated sources to use dto * subscriptions moved to dto * generated swag
70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/jtom38/newsbot/collector/domain/models"
|
|
)
|
|
|
|
type CacheClient struct {
|
|
group string
|
|
DefaultTimer time.Duration
|
|
}
|
|
|
|
func NewCacheClient(group string) CacheClient {
|
|
return CacheClient{
|
|
group: group,
|
|
DefaultTimer: time.Hour,
|
|
}
|
|
}
|
|
|
|
func (cc *CacheClient) Insert(key string, value string) {
|
|
item := models.CacheItem{
|
|
Key: key,
|
|
Value: value,
|
|
Group: cc.group,
|
|
Expires: time.Now().Add(1 * time.Hour),
|
|
IsTainted: false,
|
|
}
|
|
cacheStorage = append(cacheStorage, &item)
|
|
}
|
|
|
|
func (cc *CacheClient) FindByKey(key string) (*models.CacheItem, error) {
|
|
for _, item := range cacheStorage {
|
|
if item.Group != cc.group {
|
|
continue
|
|
}
|
|
if item.Key != key {
|
|
continue
|
|
}
|
|
|
|
// if it was tainted, renew the timer and remove the taint as this record was still needed
|
|
if item.IsTainted {
|
|
item.IsTainted = false
|
|
item.Expires = time.Now().Add(1 * time.Hour)
|
|
}
|
|
return item, nil
|
|
}
|
|
|
|
return &models.CacheItem{}, ErrCacheRecordMissing
|
|
}
|
|
|
|
func (cc *CacheClient) FindByValue(value string) (*models.CacheItem, error) {
|
|
for _, item := range cacheStorage {
|
|
if item.Group != cc.group {
|
|
continue
|
|
}
|
|
if item.Value != value {
|
|
continue
|
|
}
|
|
|
|
// if it was tainted, renew the timer and remove the taint as this record was still needed
|
|
if item.IsTainted {
|
|
item.IsTainted = false
|
|
item.Expires = time.Now().Add(1 * time.Hour)
|
|
}
|
|
return item, nil
|
|
}
|
|
return &models.CacheItem{}, ErrCacheRecordMissing
|
|
}
|