github.com/isyscore/isc-gobase@v1.5.3-0.20231218061332-cbc7451899e9/cache/operation_list.go

package cache

import (
	"errors"
	"reflect"
)

func (c *Cache) AddItem(key string, value ...any) error {
	c.mu.Lock()
	defer c.mu.Unlock()

	if item, found := c.items[key]; found {
		data := item.Data
		if reflect.TypeOf(data).Kind() != reflect.Slice {
			return errors.New("key 对应的数据类型不是 slice")
		}
		item.Data = append(data.([]any), value...)
		c.items[key] = item
	} else {
		e := c.getUnixNano()
		data := Item{
			Data: value,
			Ttl:  e,
		}
		c.items[key] = data
	}
	return nil
}

//SetItem set or replace a value of items by index
func (c *Cache) SetItem(key string, idx int, value any) error {
	c.mu.Lock()
	defer c.mu.Unlock()
	if item, found := c.items[key]; found {
		data := item.Data
		if reflect.TypeOf(data).Kind() != reflect.Slice {
			return errors.New("key 对应的数据类型不是 slice")
		}
		items := data.([]any)
		if len(items) <= idx {
			return errors.New("数组下标越界")
		}

		items[idx] = value
		item.Data = items
		return nil
	}
	return errors.New("key不存在")
}

//GetItem return an array of points or nil
func (c *Cache) GetItem(key string) []any {
	c.mu.Lock()
	defer c.mu.Unlock()
	if item, found := c.items[key]; !found {
		return nil
	} else {
		data := item.Data
		if reflect.TypeOf(data).Kind() != reflect.Slice {
			return nil
		}
		return data.([]any)
	}
}

//GetItemByIndex return a value of Type is T or nil
func (c *Cache) GetItemByIndex(key string, idx int) any {
	if idx < 0 {
		return nil
	}
	if items := c.GetItem(key); items != nil {
		if len(items) <= idx {
			return nil
		}
		return items[idx]
	}
	return nil
}

//RemoveItem an item from the cache. Does nothing if the key is not in the cache.
func (c *Cache) RemoveItem(key string, idx int) error {
	if idx < 0 {
		c.Remove(key)
		return nil
	}
	item := c.GetItem(key)
	if item == nil {
		return errors.New("key不存在")
	}
	if len(item) <= idx {
		return nil
	}
	newItem := item[:idx]
	newItem = append(newItem, item[idx+1:]...)
	_ = c.Set(key, newItem)
	return nil
}