James Tombleson
ff4075383a
* added a route to delete subscriptions based on the ID given * added a new route to find a record based on the name and source * added a route to query Discord Web Hooks by Server and Channel names * tested the endpoints and they seem good to test more * updated some routes for subscriptions and formatted files * removed debug file * fixing some panic calls * swag
70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/jtom38/newsbot/collector/domain/model"
|
|
)
|
|
|
|
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 := model.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) (*model.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 &model.CacheItem{}, ErrCacheRecordMissing
|
|
}
|
|
|
|
func (cc *CacheClient) FindByValue(value string) (*model.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 &model.CacheItem{}, ErrCacheRecordMissing
|
|
}
|