70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package tmpfs
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Tmpfs struct {
|
|
record map[string]int64
|
|
Config TmpfsConfig
|
|
wg sync.WaitGroup
|
|
}
|
|
|
|
func (tmpfs *Tmpfs) GetObjFilePath(id int64, configName string) (string) {
|
|
return filepath.Join(tmpfs.Config.Root, strconv.FormatInt(id, 10) + "." + configName + ".ogg")
|
|
}
|
|
|
|
type TmpfsConfig struct {
|
|
FileLifeTime int64 `json:"file_life_time"`
|
|
CleanerInternal int64 `json:"cleaner_internal"`
|
|
Root string `json:"root"`
|
|
}
|
|
|
|
func NewTmpfsConfig() (*TmpfsConfig) {
|
|
config := &TmpfsConfig{}
|
|
return config
|
|
}
|
|
|
|
func NewTmpfs(config TmpfsConfig) *Tmpfs {
|
|
tmpfs := &Tmpfs{
|
|
record: make(map[string]int64),
|
|
Config: config,
|
|
}
|
|
tmpfs.wg.Add(1)
|
|
go tmpfs.Cleaner()
|
|
return tmpfs
|
|
}
|
|
|
|
func (tmpfs *Tmpfs) Record(filename string) {
|
|
tmpfs.record[filename] = time.Now().Unix()
|
|
}
|
|
|
|
func (tmpfs *Tmpfs) Exits(filename string) (bool) {
|
|
_, ok := tmpfs.record[filename]
|
|
return ok
|
|
}
|
|
|
|
func (tmpfs *Tmpfs) Cleaner() {
|
|
var err error
|
|
for {
|
|
now := time.Now().Unix()
|
|
for key, value := range tmpfs.record {
|
|
if now - value > tmpfs.Config.FileLifeTime {
|
|
err = os.Remove(key)
|
|
if err != nil {
|
|
log.Println("[tmpfs] Failed to remove file", err)
|
|
}
|
|
log.Println("[tmpfs] Deleted file", key)
|
|
delete(tmpfs.record, key)
|
|
}
|
|
}
|
|
|
|
time.Sleep(time.Second)
|
|
}
|
|
}
|