3 Star 5 Fork 6

三三物联网/ssiot-core

Create your Gitee Account
Explore and code with more than 13.5 million developers,Free private repositories !:)
Sign up
文件
Clone or Download
memory.go 3.43 KB
Copy Edit Raw Blame History
zliu authored 2025-01-07 11:31 +08:00 . MultiHGetAll
package cache
import (
"fmt"
"strconv"
"sync"
"time"
"github.com/spf13/cast"
)
type item struct {
Value string
Expired time.Time
}
// NewMemory memory模式
func NewMemory() *Memory {
return &Memory{
items: new(sync.Map),
}
}
type Memory struct {
items *sync.Map
mutex sync.RWMutex
}
func (*Memory) String() string {
return "memory"
}
func (m *Memory) connect() {
}
func (m *Memory) Get(key string) (string, error) {
item, err := m.getItem(key)
if err != nil || item == nil {
return "", err
}
return item.Value, nil
}
func (m *Memory) getItem(key string) (*item, error) {
var err error
i, ok := m.items.Load(key)
if !ok {
return nil, nil
}
switch i.(type) {
case *item:
item := i.(*item)
if item.Expired.Before(time.Now()) {
//过期
_ = m.del(key)
//过期后删除
return nil, nil
}
return item, nil
default:
err = fmt.Errorf("value of %s type error", key)
return nil, err
}
}
func (m *Memory) Set(key string, val interface{}, expire int) error {
s, err := cast.ToStringE(val)
if err != nil {
return err
}
item := &item{
Value: s,
Expired: time.Now().Add(time.Duration(expire) * time.Second),
}
return m.setItem(key, item)
}
func (m *Memory) HashSet(key string, val ...interface{}) error {
return nil
}
func (m *Memory) HashGetAll(hk string) (map[string]string, error) {
return nil, nil
}
func (m *Memory) setItem(key string, item *item) error {
m.items.Store(key, item)
return nil
}
func (m *Memory) Del(key string) error {
return m.del(key)
}
func (m *Memory) del(key string) error {
m.items.Delete(key)
return nil
}
func (m *Memory) HExists(hk, key string) (bool, error) {
return false, nil
}
func (m *Memory) HashGet(hk, key string) (string, error) {
item, err := m.getItem(hk + key)
if err != nil || item == nil {
return "", err
}
return item.Value, err
}
func (m *Memory) HashDel(hk, key string) error {
return m.del(hk + key)
}
func (m *Memory) Increase(key string) error {
return m.calculate(key, 1)
}
func (m *Memory) Decrease(key string) error {
return m.calculate(key, -1)
}
func (m *Memory) calculate(key string, num int) error {
m.mutex.RLock()
defer m.mutex.RUnlock()
item, err := m.getItem(key)
if err != nil {
return err
}
if item == nil {
err = fmt.Errorf("%s not exist", key)
return err
}
var n int
n, err = cast.ToIntE(item.Value)
if err != nil {
return err
}
n += num
item.Value = strconv.Itoa(n)
return m.setItem(key, item)
}
func (m *Memory) Expire(key string, dur time.Duration) error {
m.mutex.RLock()
defer m.mutex.RUnlock()
item, err := m.getItem(key)
if err != nil {
return err
}
if item == nil {
err = fmt.Errorf("%s not exist", key)
return err
}
item.Expired = time.Now().Add(dur)
return m.setItem(key, item)
}
func (m *Memory) HIncrBy(key, field string, increment int) (int64, error) {
return 0, nil
}
func (m *Memory) ScanKey(key string) []string {
return nil
}
func (m *Memory) LPush(key string, values ...interface{}) (int64, error) {
return 0, nil
}
func (m *Memory) RPush(key string, values ...interface{}) error {
return nil
}
func (m *Memory) BRPop(timeout time.Duration, keys ...string) ([]string, error) {
sa := make([]string, 0)
return sa, nil
}
func (m *Memory) RPop(key string) (string, error) {
return "", nil
}
func (r *Memory) LRange(key string, start, stop int64) ([]string, error) {
return nil, nil
}
func (r *Memory) MultiHGetAll(keys []string) (map[string]map[string]string, error) {
return nil, nil
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/sansaniot/ssiot-core.git
git@gitee.com:sansaniot/ssiot-core.git
sansaniot
ssiot-core
ssiot-core
v1.8.35

Search