116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-2021 Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public Licensee as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public Licensee for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public Licensee
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package redis
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/gob"
|
|
"errors"
|
|
|
|
"code.vikunja.io/api/pkg/red"
|
|
"github.com/go-redis/redis/v8"
|
|
)
|
|
|
|
// Storage is a redis implementation of a keyvalue storage
|
|
type Storage struct {
|
|
client *redis.Client
|
|
}
|
|
|
|
// NewStorage creates a new redis key value storage
|
|
func NewStorage() *Storage {
|
|
red.InitRedis()
|
|
|
|
return &Storage{
|
|
client: red.GetRedis(),
|
|
}
|
|
}
|
|
|
|
// Put puts a value into redis
|
|
func (s *Storage) Put(key string, value interface{}) (err error) {
|
|
|
|
var v interface{}
|
|
|
|
switch value.(type) {
|
|
case int:
|
|
v = value
|
|
case int8:
|
|
v = value
|
|
case int16:
|
|
v = value
|
|
case int32:
|
|
v = value
|
|
case int64:
|
|
v = value
|
|
default:
|
|
var buf bytes.Buffer
|
|
enc := gob.NewEncoder(&buf)
|
|
err = enc.Encode(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.client.Set(context.Background(), key, buf.Bytes(), 0).Err()
|
|
}
|
|
|
|
return s.client.Set(context.Background(), key, v, 0).Err()
|
|
}
|
|
|
|
// Get retrieves a saved value from redis
|
|
func (s *Storage) Get(key string) (value interface{}, exists bool, err error) {
|
|
value, err = s.client.Get(context.Background(), key).Result()
|
|
if err != nil && errors.Is(err, redis.Nil) {
|
|
return nil, false, nil
|
|
}
|
|
return value, true, err
|
|
}
|
|
|
|
func (s *Storage) GetWithValue(key string, value interface{}) (exists bool, err error) {
|
|
b, err := s.client.Get(context.Background(), key).Bytes()
|
|
if err != nil {
|
|
if errors.Is(err, redis.Nil) {
|
|
return false, nil
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
_, err = buf.Write(b)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
dec := gob.NewDecoder(&buf)
|
|
err = dec.Decode(value)
|
|
return true, err
|
|
}
|
|
|
|
// Del removed a value from redis
|
|
func (s *Storage) Del(key string) (err error) {
|
|
return s.client.Del(context.Background(), key).Err()
|
|
}
|
|
|
|
// IncrBy increases the value saved at key by the amount provided through update
|
|
func (s *Storage) IncrBy(key string, update int64) (err error) {
|
|
return s.client.IncrBy(context.Background(), key, update).Err()
|
|
}
|
|
|
|
// DecrBy decreases the value saved at key by the amount provided through update
|
|
func (s *Storage) DecrBy(key string, update int64) (err error) {
|
|
return s.client.DecrBy(context.Background(), key, update).Err()
|
|
}
|