x

core/node_temp.go

@@ -0,0 +1,133 @@
+package core
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"sync"
+	"time"
+
+	"github.com/cdle/sillyplus/utils"
+)
+
+var temp *PersistentKeyValueStore
+var tempPath = utils.GetDataHome() + "temp.json"
+
+type PersistentKeyValueStore struct {
+	sync.RWMutex
+	data map[string]Value
+}
+
+type Value struct {
+	Data      interface{} `json:"data"`
+	ExpiredAt time.Time   `json:"expired_at"`
+}
+
+func NewPersistentKeyValueStore() *PersistentKeyValueStore {
+	return &PersistentKeyValueStore{
+		data: make(map[string]Value),
+	}
+}
+
+func (s *PersistentKeyValueStore) Set(key string, value interface{}, dur int) error {
+	s.Lock()
+	defer s.Unlock()
+	if dur == 0 {
+		dur = 86400
+	}
+	expiredAt := time.Now().Add(time.Duration(dur) * time.Second)
+	s.data[key] = Value{
+		Data:      value,
+		ExpiredAt: expiredAt,
+	}
+	go func() {
+		jsonBytes, err := json.Marshal(s.data)
+		if err == nil {
+			ioutil.WriteFile(tempPath, jsonBytes, 0644)
+		}
+	}()
+	return nil
+}
+
+func (s *PersistentKeyValueStore) Get(key string) interface{} {
+	s.RLock()
+	defer s.RUnlock()
+	value, ok := s.data[key]
+	if !ok || time.Now().After(value.ExpiredAt) {
+		return nil
+	}
+	return value.Data
+}
+
+func (s *PersistentKeyValueStore) Delete(key string) error {
+	s.Lock()
+	defer s.Unlock()
+	delete(s.data, key)
+	// Serialize data to JSON and write to file
+	jsonBytes, err := json.Marshal(s.data)
+	if err != nil {
+		return err
+	}
+	err = ioutil.WriteFile(tempPath, jsonBytes, 0644)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (s *PersistentKeyValueStore) LoadFromFile() error {
+	s.Lock()
+	defer s.Unlock()
+
+	// Read data from file
+	jsonBytes, err := ioutil.ReadFile(tempPath)
+	if err != nil {
+		return err
+	}
+	// Deserialize JSON data
+	err = json.Unmarshal(jsonBytes, &s.data)
+	if err != nil {
+		return err
+	}
+	// Delete expired data
+	now := time.Now()
+	for key, value := range s.data {
+		if now.After(value.ExpiredAt) {
+			delete(s.data, key)
+		}
+	}
+	// Serialize data to JSON and write to file
+	jsonBytes, err = json.Marshal(s.data)
+	if err != nil {
+		return err
+	}
+	err = ioutil.WriteFile(tempPath, jsonBytes, 0644)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func init() {
+	temp = NewPersistentKeyValueStore()
+	// Load data from file
+	temp.LoadFromFile()
+	// if err != nil {
+	// 	fmt.Println("Failed to load data from file:", err)
+	// }
+	// // Set a key-value pair
+	// err = store.Set("foo", map[string]interface{}{
+	// 	"bar": "baz",
+	// }, 1000)
+	// if err != nil {
+	// 	fmt.Println("Failed to set key-value pair:", err)
+	// }
+	// // Get a value
+	// value := store.Get("foo")
+	// fmt.Println(value)
+
+	// // Delete a key-value pair
+	// err = store.Delete("foo")
+	// if err != nil {
+	// 	fmt.Println("Failed to delete key-value pair:", err)
+	// }
+}