newsbot-api/services/cache/cache.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
}