50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package storage
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
// FlushToFile flushes the merged servers data to the file
|
|
// specified by path, as indented JSON.
|
|
func (s *Storage) FlushToFile(path string) error {
|
|
s.mergedMutex.RLock()
|
|
defer s.mergedMutex.RUnlock()
|
|
|
|
return s.flushToFile(path)
|
|
}
|
|
|
|
// flushToFile flushes the merged servers data to the file
|
|
// specified by path, as indented JSON. It is not thread-safe.
|
|
func (s *Storage) flushToFile(path string) error {
|
|
const permission = 0o644
|
|
dirPath := filepath.Dir(path)
|
|
if err := os.MkdirAll(dirPath, permission); err != nil {
|
|
return err
|
|
}
|
|
|
|
file, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, permission)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
encoder := json.NewEncoder(file)
|
|
encoder.SetIndent("", " ")
|
|
|
|
for _, obj := range s.mergedServers.ProviderToServers {
|
|
sort.Sort(models.SortableServers(obj.Servers))
|
|
}
|
|
|
|
err = encoder.Encode(&s.mergedServers)
|
|
if err != nil {
|
|
_ = file.Close()
|
|
return err
|
|
}
|
|
|
|
return file.Close()
|
|
}
|