f-46: statefile: add a generic atomic state file #46
Persistance d'un état de composant dans un fichier, générique sur le type stocké : temp + chmod 0600 + sync + rename, répertoire en 0700. Chaque écriture remplace le fichier, donc un chmod externe est effacé, et un échec d'encodage ne tronque pas l'état précédent. Load crée le fichier absent et rend le zéro du type ; un fichier vide est un point de départ valide, un fichier corrompu est une erreur. Dans pkg/ plutôt que sous internal/dhcpd : une réutilisation par l'intel est envisagée. Écarte Badger pour cet usage — une instance par subnet coûterait ~74 Mio et 4 goroutines pour ~1 Kio d'état dans un tmpfs, plus un LOCK résiduel après un arrêt brutal. 11 tests, quatre mutations toutes détectées. Signed-off-by: GnomeZworc <nicolas.boufidjeline@g3e.fr>
This commit is contained in:
parent
e44f3aca96
commit
8802de4fb0
2 changed files with 281 additions and 0 deletions
92
pkg/db/statefile/statefile.go
Normal file
92
pkg/db/statefile/statefile.go
Normal file
|
|
@ -0,0 +1,92 @@
|
|||
package statefile
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
const (
|
||||
FileMode = 0o600
|
||||
DirMode = 0o700
|
||||
)
|
||||
|
||||
type File[T any] struct {
|
||||
path string
|
||||
}
|
||||
|
||||
func New[T any](path string) *File[T] {
|
||||
return &File[T]{path: path}
|
||||
}
|
||||
|
||||
func (f *File[T]) Path() string {
|
||||
return f.path
|
||||
}
|
||||
|
||||
func (f *File[T]) Load() (T, error) {
|
||||
var value T
|
||||
|
||||
raw, err := os.ReadFile(f.path)
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
return value, f.Save(value)
|
||||
}
|
||||
if err != nil {
|
||||
return value, fmt.Errorf("read %s: %w", f.path, err)
|
||||
}
|
||||
if len(raw) == 0 {
|
||||
return value, nil
|
||||
}
|
||||
if err := json.Unmarshal(raw, &value); err != nil {
|
||||
var zero T
|
||||
return zero, fmt.Errorf("parse %s: %w", f.path, err)
|
||||
}
|
||||
return value, nil
|
||||
}
|
||||
|
||||
func (f *File[T]) Save(value T) error {
|
||||
raw, err := json.Marshal(value)
|
||||
if err != nil {
|
||||
return fmt.Errorf("encode state for %s: %w", f.path, err)
|
||||
}
|
||||
|
||||
dir := filepath.Dir(f.path)
|
||||
if err := os.MkdirAll(dir, DirMode); err != nil {
|
||||
return fmt.Errorf("create %s: %w", dir, err)
|
||||
}
|
||||
|
||||
tmp, err := os.CreateTemp(dir, filepath.Base(f.path)+".tmp")
|
||||
if err != nil {
|
||||
return fmt.Errorf("create temp state in %s: %w", dir, err)
|
||||
}
|
||||
defer os.Remove(tmp.Name())
|
||||
|
||||
if err := tmp.Chmod(FileMode); err != nil {
|
||||
tmp.Close()
|
||||
return fmt.Errorf("chmod %s: %w", tmp.Name(), err)
|
||||
}
|
||||
if _, err := tmp.Write(raw); err != nil {
|
||||
tmp.Close()
|
||||
return fmt.Errorf("write %s: %w", tmp.Name(), err)
|
||||
}
|
||||
if err := tmp.Sync(); err != nil {
|
||||
tmp.Close()
|
||||
return fmt.Errorf("sync %s: %w", tmp.Name(), err)
|
||||
}
|
||||
if err := tmp.Close(); err != nil {
|
||||
return fmt.Errorf("close %s: %w", tmp.Name(), err)
|
||||
}
|
||||
|
||||
if err := os.Rename(tmp.Name(), f.path); err != nil {
|
||||
return fmt.Errorf("rename %s to %s: %w", tmp.Name(), f.path, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *File[T]) Remove() error {
|
||||
if err := os.Remove(f.path); err != nil && !errors.Is(err, os.ErrNotExist) {
|
||||
return fmt.Errorf("remove %s: %w", f.path, err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue