104 lines
2.0 KiB
Go
104 lines
2.0 KiB
Go
package gost
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/go-log/log"
|
|
)
|
|
|
|
// Reloader is the interface for objects that support live reloading.
|
|
type Reloader interface {
|
|
Reload(r io.Reader) error
|
|
Period() time.Duration
|
|
}
|
|
|
|
// Stoppable is the interface that indicates a Reloader can be stopped.
|
|
type Stoppable interface {
|
|
Stop()
|
|
Stopped() bool
|
|
}
|
|
|
|
// PeriodReload reloads the config configFile periodically according to the period of the Reloader r.
|
|
func PeriodReload(r Reloader, configFile string) error {
|
|
if r == nil || configFile == "" {
|
|
return nil
|
|
}
|
|
|
|
var lastMod time.Time
|
|
for {
|
|
if r.Period() < 0 {
|
|
log.Log("[reload] stopped:", configFile)
|
|
return nil
|
|
}
|
|
|
|
f, err := os.Open(configFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mt := lastMod
|
|
if finfo, err := f.Stat(); err == nil {
|
|
mt = finfo.ModTime()
|
|
}
|
|
|
|
if !lastMod.IsZero() && !mt.Equal(lastMod) {
|
|
log.Log("[reload]", configFile)
|
|
if err := r.Reload(f); err != nil {
|
|
log.Logf("[reload] %s: %s", configFile, err)
|
|
}
|
|
}
|
|
f.Close()
|
|
lastMod = mt
|
|
|
|
period := r.Period()
|
|
if period == 0 {
|
|
log.Log("[reload] disabled:", configFile)
|
|
return nil
|
|
}
|
|
if period < time.Second {
|
|
period = time.Second
|
|
}
|
|
<-time.After(period)
|
|
}
|
|
}
|
|
|
|
// PeriodReloadRemote reloads the remote configFile periodically according to the period of the Reloader r.
|
|
func PeriodReloadRemote(r Reloader, cfg string) error {
|
|
if r == nil || cfg == "" {
|
|
return nil
|
|
}
|
|
|
|
client := http.Client{}
|
|
for {
|
|
if r.Period() < 0 {
|
|
log.Log("[reload] stopped:", cfg)
|
|
return nil
|
|
}
|
|
resp, err := client.Get(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != 200 {
|
|
return fmt.Errorf("fetch remote resource error, reply status code is not equals to 200")
|
|
}
|
|
if err := r.Reload(resp.Body); err != nil {
|
|
log.Logf("[reload] %s: %s", cfg, err)
|
|
}
|
|
|
|
period := r.Period()
|
|
if period == 0 {
|
|
log.Log("[reload] disabled:", cfg)
|
|
return nil
|
|
}
|
|
if period < time.Second {
|
|
period = time.Second
|
|
}
|
|
<-time.After(period)
|
|
}
|
|
}
|