49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
// ___ ____ ___ ___
|
|
// \ \ / / | _ | __| \ \ / / || | __ || || _ |
|
|
// \ \/ / |___ | |__ \ \/ / || |___ || ||___|
|
|
// \ / | _ | _ | \ / || __ | || ||\\
|
|
// \/ |___ |___ | \/ || ____| || || \\
|
|
//
|
|
// Copyright (c) 2021 Piotr Biernat. https://pbiernat.dev. MIT License
|
|
// Repo: https://git.pbiernat.dev/golang/vegvisir
|
|
|
|
package cache
|
|
|
|
import (
|
|
"log"
|
|
"vegvisir/pkg/config"
|
|
)
|
|
|
|
const (
|
|
typeRedis = "redis"
|
|
typeMemory = "memory"
|
|
)
|
|
|
|
// Datastore interface
|
|
type Datastore interface {
|
|
SetKey(string, interface{}, int) error
|
|
|
|
GetKey(string) (interface{}, error)
|
|
|
|
IsConnected() bool
|
|
}
|
|
|
|
// GetCacheDatastore function
|
|
func GetCacheDatastore(config config.Cache) *Datastore {
|
|
var datastore Datastore
|
|
|
|
if config.Type == typeRedis {
|
|
datastore = NewRedisDatastore(config.Host, config.Port)
|
|
} else {
|
|
datastore = NewMemoryDatastore()
|
|
}
|
|
|
|
// fail-safe switch to memory datasource
|
|
if !datastore.IsConnected() {
|
|
log.Println("Cache server is not responding, switching to memory cache.")
|
|
datastore = NewMemoryDatastore()
|
|
}
|
|
|
|
return &datastore
|
|
}
|