registry-auth/pkg/server/file.go

package server

import (
	"fmt"
	"go.uber.org/zap"
	"io/ioutil"

	"gopkg.in/fsnotify.v1"
)

func readConfigFile(path string, p *AuthProcessor) error {
	data, err := ioutil.ReadFile(path)
	if err != nil {
		logger.Error(fmt.Sprintf(`read config file from "%s" error: %v`, path, err), zap.String("func", "readConfigFile"))
		return err
	}
	if err := p.LoadFromFile(data); err != nil {
		logger.Error(fmt.Sprintf(`load config file from "%s" error: %v`, path, err), zap.String("func", "readConfigFile"))
	}
	return err
}

func WatchConfigFile(path string, stop <-chan struct{}, p *AuthProcessor) error {
	if err := readConfigFile(path, p); err != nil {
		return err
	}
	w, err := fsnotify.NewWatcher()
	if err != nil {
		return err
	}
	if err := w.Add(path); err != nil {
		return err
	}

	go func() {
		stopped := false
		for !stopped {
			select {
			case <-stop:
				stopped = true
			case ev := <-w.Events:
				if ev.Op == fsnotify.Create || ev.Op == fsnotify.Write {
					readConfigFile(path, p)
				}
			}
		}
	}()
	return nil
}