mirror of
https://github.com/go-gitea/gitea
synced 2024-08-20 14:21:21 +02:00
* Allow common redis and leveldb connections Prevents multiple reopening of redis and leveldb connections to the same place by sharing connections. Further allows for more configurable redis connection type using the redisURI and a leveldbURI scheme. Signed-off-by: Andrew Thornton <art27@cantab.net> * add unit-test Signed-off-by: Andrew Thornton <art27@cantab.net> * as per @lunny Signed-off-by: Andrew Thornton <art27@cantab.net> * add test Signed-off-by: Andrew Thornton <art27@cantab.net> * Update modules/cache/cache_redis.go * Update modules/queue/queue_disk.go * Update modules/cache/cache_redis.go * Update modules/cache/cache_redis.go * Update modules/queue/unique_queue_disk.go * Update modules/queue/queue_disk.go * Update modules/queue/unique_queue_disk.go * Update modules/session/redis.go Co-authored-by: techknowlogick <techknowlogick@gitea.io> Co-authored-by: Lauris BH <lauris@nix.lv>
142 lines
3.0 KiB
Go
142 lines
3.0 KiB
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package cache
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
mc "gitea.com/macaron/cache"
|
|
|
|
_ "gitea.com/macaron/cache/memcache" // memcache plugin for cache
|
|
)
|
|
|
|
var (
|
|
conn mc.Cache
|
|
)
|
|
|
|
func newCache(cacheConfig setting.Cache) (mc.Cache, error) {
|
|
return mc.NewCacher(cacheConfig.Adapter, mc.Options{
|
|
Adapter: cacheConfig.Adapter,
|
|
AdapterConfig: cacheConfig.Conn,
|
|
Interval: cacheConfig.Interval,
|
|
})
|
|
}
|
|
|
|
// NewContext start cache service
|
|
func NewContext() error {
|
|
var err error
|
|
|
|
if conn == nil && setting.CacheService.Enabled {
|
|
if conn, err = newCache(setting.CacheService.Cache); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// GetString returns the key value from cache with callback when no key exists in cache
|
|
func GetString(key string, getFunc func() (string, error)) (string, error) {
|
|
if conn == nil || setting.CacheService.TTL == 0 {
|
|
return getFunc()
|
|
}
|
|
if !conn.IsExist(key) {
|
|
var (
|
|
value string
|
|
err error
|
|
)
|
|
if value, err = getFunc(); err != nil {
|
|
return value, err
|
|
}
|
|
err = conn.Put(key, value, int64(setting.CacheService.TTL.Seconds()))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
value := conn.Get(key)
|
|
if v, ok := value.(string); ok {
|
|
return v, nil
|
|
}
|
|
if v, ok := value.(fmt.Stringer); ok {
|
|
return v.String(), nil
|
|
}
|
|
return fmt.Sprintf("%s", conn.Get(key)), nil
|
|
}
|
|
|
|
// GetInt returns key value from cache with callback when no key exists in cache
|
|
func GetInt(key string, getFunc func() (int, error)) (int, error) {
|
|
if conn == nil || setting.CacheService.TTL == 0 {
|
|
return getFunc()
|
|
}
|
|
if !conn.IsExist(key) {
|
|
var (
|
|
value int
|
|
err error
|
|
)
|
|
if value, err = getFunc(); err != nil {
|
|
return value, err
|
|
}
|
|
err = conn.Put(key, value, int64(setting.CacheService.TTL.Seconds()))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
switch value := conn.Get(key).(type) {
|
|
case int:
|
|
return value, nil
|
|
case string:
|
|
v, err := strconv.Atoi(value)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return v, nil
|
|
default:
|
|
return 0, fmt.Errorf("Unsupported cached value type: %v", value)
|
|
}
|
|
}
|
|
|
|
// GetInt64 returns key value from cache with callback when no key exists in cache
|
|
func GetInt64(key string, getFunc func() (int64, error)) (int64, error) {
|
|
if conn == nil || setting.CacheService.TTL == 0 {
|
|
return getFunc()
|
|
}
|
|
if !conn.IsExist(key) {
|
|
var (
|
|
value int64
|
|
err error
|
|
)
|
|
if value, err = getFunc(); err != nil {
|
|
return value, err
|
|
}
|
|
err = conn.Put(key, value, int64(setting.CacheService.TTL.Seconds()))
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
switch value := conn.Get(key).(type) {
|
|
case int64:
|
|
return value, nil
|
|
case string:
|
|
v, err := strconv.ParseInt(value, 10, 64)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return v, nil
|
|
default:
|
|
return 0, fmt.Errorf("Unsupported cached value type: %v", value)
|
|
}
|
|
}
|
|
|
|
// Remove key from cache
|
|
func Remove(key string) {
|
|
if conn == nil {
|
|
return
|
|
}
|
|
_ = conn.Delete(key)
|
|
}
|