feat: cache + paths #1
3 changed files with 148 additions and 0 deletions
77
cache/file.go
vendored
Normal file
77
cache/file.go
vendored
Normal file
|
@ -0,0 +1,77 @@
|
|||
package cache
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"git.nakama.town/fmartingr/gotoolkit/model"
|
||||
)
|
||||
|
||||
var _ model.Cache = (*FileCache)(nil)
|
||||
|
||||
type FileCache struct {
|
||||
path string
|
||||
}
|
||||
|
||||
func (c *FileCache) Get(key string) (result any, err error) {
|
||||
path := filepath.Join(c.path, key)
|
||||
|
||||
contents, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
if errors.Is(err, os.ErrNotExist) || errors.Is(err, os.ErrPermission) {
|
||||
return result, model.ErrCacheKeyDontExist
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
return contents, nil
|
||||
}
|
||||
|
||||
func (c *FileCache) GetWithExpiry(key string, expiration time.Duration) (result any, err error) {
|
||||
path := filepath.Join(c.path, key)
|
||||
info, err := os.Stat(path)
|
||||
if err != nil {
|
||||
return result, model.ErrCacheKeyDontExist
|
||||
}
|
||||
|
||||
if info.ModTime().Add(expiration).Before(time.Now()) {
|
||||
c.Delete(key)
|
||||
return result, model.ErrCacheKeyDontExist
|
||||
}
|
||||
|
||||
return c.Get(key)
|
||||
}
|
||||
|
||||
func (c *FileCache) Set(key string, value any) error {
|
||||
path := filepath.Join(c.path, key)
|
||||
|
||||
if err := os.WriteFile(path, value.([]byte), 0766); err != nil {
|
||||
return fmt.Errorf("error writting cache file: %s", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *FileCache) Delete(key string) error {
|
||||
path := filepath.Join(c.path, key)
|
||||
return os.Remove(path)
|
||||
}
|
||||
|
||||
func NewFileCache(folderName string) (*FileCache, error) {
|
||||
userCacheDir, err := os.UserCacheDir()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not retrieve user cache directory: %w", err)
|
||||
}
|
||||
path := filepath.Join(userCacheDir, folderName)
|
||||
|
||||
if err := os.MkdirAll(path, 0755); err != nil {
|
||||
return nil, fmt.Errorf("couldn't create cache directory: %w", err)
|
||||
}
|
||||
|
||||
return &FileCache{
|
||||
path: path,
|
||||
}, nil
|
||||
}
|
53
cache/memory.go
vendored
Normal file
53
cache/memory.go
vendored
Normal file
|
@ -0,0 +1,53 @@
|
|||
package cache
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"git.nakama.town/fmartingr/gotoolkit/model"
|
||||
)
|
||||
|
||||
var _ model.Cache = (*MemoryCache)(nil)
|
||||
|
||||
type MemoryCache struct {
|
||||
data map[string]any
|
||||
dataMu sync.RWMutex
|
||||
}
|
||||
|
||||
func (c *MemoryCache) Get(key string) (result any, err error) {
|
||||
c.dataMu.RLock()
|
||||
defer c.dataMu.RUnlock()
|
||||
|
||||
result, exists := c.data[key]
|
||||
if !exists {
|
||||
return result, model.ErrCacheKeyDontExist
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (c *MemoryCache) GetWithExpiry(key string, expiration time.Duration) (any, error) {
|
||||
// TODO: Implement expiration in memory cache.
|
||||
return c.Get(key)
|
||||
}
|
||||
|
||||
func (c *MemoryCache) Set(key string, value any) error {
|
||||
c.dataMu.Lock()
|
||||
c.data[key] = value
|
||||
c.dataMu.Unlock()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *MemoryCache) Delete(key string) error {
|
||||
c.dataMu.Lock()
|
||||
delete(c.data, key)
|
||||
c.dataMu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewMemoryCache() *MemoryCache {
|
||||
return &MemoryCache{
|
||||
data: make(map[string]any),
|
||||
}
|
||||
}
|
18
model/cache.go
Normal file
18
model/cache.go
Normal file
|
@ -0,0 +1,18 @@
|
|||
package model
|
||||
|
||||
import (
|
||||
"errors"
|
||||
)
|
||||
|
||||
var ErrCacheKeyDontExist = errors.New("cache key don't exist")
|
||||
|
||||
type Cache interface {
|
||||
Get(key string) (any, error)
|
||||
GetWithOptions(key string, opts ...CacheGetOption)
|
||||
Set(key string, value any) error
|
||||
SetWithOptions(key string, value any, opts ...CacheSetOption)
|
||||
Delete(key string) error
|
||||
}
|
||||
|
||||
type CacheGetOption func()
|
||||
type CacheSetOption func()
|
Loading…
Add table
Add a link
Reference in a new issue