mirror of https://github.com/miguelmota/cointop
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
2.5 KiB
Go
131 lines
2.5 KiB
Go
package fcache
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
const cachedir = "/tmp"
|
|
|
|
// Set writes item to cache
|
|
func Set(key string, data interface{}, expire time.Duration) error {
|
|
key = regexp.MustCompile("[^a-zA-Z0-9_-]").ReplaceAllLiteralString(key, "")
|
|
file := fmt.Sprintf("fcache.%s.%v", key, strconv.FormatInt(time.Now().Add(expire).Unix(), 10))
|
|
fpath := filepath.Join(cachedir, file)
|
|
|
|
clean(key)
|
|
|
|
serialized, err := serialize(data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var fmutex sync.RWMutex
|
|
fmutex.Lock()
|
|
defer fmutex.Unlock()
|
|
fp, err := os.OpenFile(fpath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer fp.Close()
|
|
if _, err = fp.Write(serialized); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Get reads item from cache
|
|
func Get(key string, dst interface{}) error {
|
|
key = regexp.MustCompile("[^a-zA-Z0-9_-]").ReplaceAllLiteralString(key, "")
|
|
pattern := filepath.Join(cachedir, fmt.Sprintf("fcache.%s.*", key))
|
|
files, err := filepath.Glob(pattern)
|
|
if len(files) < 1 || err != nil {
|
|
return errors.New("fcache: no cache file found")
|
|
}
|
|
|
|
if _, err = os.Stat(files[0]); err != nil {
|
|
return err
|
|
}
|
|
|
|
fp, err := os.OpenFile(files[0], os.O_RDONLY, 0400)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fp.Close()
|
|
|
|
var serialized []byte
|
|
buf := make([]byte, 1024)
|
|
for {
|
|
var n int
|
|
n, err = fp.Read(buf)
|
|
serialized = append(serialized, buf[0:n]...)
|
|
if err != nil || err == io.EOF {
|
|
break
|
|
}
|
|
}
|
|
|
|
if err = deserialize(serialized, dst); err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, file := range files {
|
|
exptime, err := strconv.ParseInt(strings.Split(file, ".")[2], 10, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if exptime < time.Now().Unix() {
|
|
if _, err = os.Stat(file); err == nil {
|
|
os.Remove(file)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// clean removes item from cache
|
|
func clean(key string) error {
|
|
pattern := filepath.Join(cachedir, fmt.Sprintf("fcache.%s.*", key))
|
|
files, _ := filepath.Glob(pattern)
|
|
for _, file := range files {
|
|
if _, err := os.Stat(file); err == nil {
|
|
os.Remove(file)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// serialize encodes a value using binary
|
|
func serialize(src interface{}) ([]byte, error) {
|
|
buf := new(bytes.Buffer)
|
|
if err := gob.NewEncoder(buf).Encode(src); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
// deserialize decodes a value using binary
|
|
func deserialize(src []byte, dst interface{}) error {
|
|
buf := bytes.NewReader(src)
|
|
if err := gob.NewDecoder(buf).Decode(dst); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|