54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package file
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/naoina/toml"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TOMLDuration a time.Duration inside toml files
|
||
|
|
type TOMLDuration time.Duration
|
||
|
|
|
||
|
|
// UnmarshalText implements encoding.TextUnmarshaler
|
||
|
|
func (d *TOMLDuration) UnmarshalText(data []byte) error {
|
||
|
|
duration, err := time.ParseDuration(string(data))
|
||
|
|
if err == nil {
|
||
|
|
*d = TOMLDuration(duration)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalText implements encoding.TextMarshaler
|
||
|
|
func (d TOMLDuration) MarshalText() ([]byte, error) {
|
||
|
|
return []byte(time.Duration(d).String()), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReadTOML reads a config model from path of a toml file
|
||
|
|
func ReadTOML(file string, data interface{}) error {
|
||
|
|
f, err := os.Open(file)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer f.Close()
|
||
|
|
return toml.NewDecoder(f).Decode(data)
|
||
|
|
}
|
||
|
|
|
||
|
|
// SaveTOML to path
|
||
|
|
func SaveTOML(outputFile string, data interface{}) error {
|
||
|
|
tmpFile := outputFile + ".tmp"
|
||
|
|
|
||
|
|
file, err := os.OpenFile(tmpFile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
err = toml.NewEncoder(file).Encode(data)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
file.Close()
|
||
|
|
return os.Rename(tmpFile, outputFile)
|
||
|
|
}
|