mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-03 07:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
60 lines
1.1 KiB
Go
60 lines
1.1 KiB
Go
package state
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log/slog"
|
|
"os"
|
|
"sync"
|
|
)
|
|
|
|
type Store struct {
|
|
path string
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
type State struct {
|
|
LastAuthToken string `json:"last_auth_token"`
|
|
LastProfileFingerprint string `json:"last_profile_fingerprint"`
|
|
LastCPUStatTotal uint64 `json:"last_cpu_stat_total"`
|
|
LastCPUStatIdle uint64 `json:"last_cpu_stat_idle"`
|
|
LastMetricAtUnix int64 `json:"last_metric_at_unix"`
|
|
}
|
|
|
|
func NewStore(path string) *Store {
|
|
return &Store{
|
|
path: path,
|
|
}
|
|
}
|
|
|
|
func (s *Store) Load() (*State, error) {
|
|
s.mu.RLock()
|
|
defer s.mu.RUnlock()
|
|
|
|
data, err := os.ReadFile(s.path)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return &State{}, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
var state State
|
|
if err := json.Unmarshal(data, &state); err != nil {
|
|
return &State{}, nil // Return empty state on corrupted file
|
|
}
|
|
return &state, nil
|
|
}
|
|
|
|
func (s *Store) Save(state *State) error {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
|
|
data, err := json.MarshalIndent(state, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
slog.Debug("saving relay state")
|
|
return os.WriteFile(s.path, data, 0644)
|
|
}
|