'aclmonitor'

broker/auth.go

@@ -3,6 +3,9 @@ package broker
 import (
 	"hmq/lib/acl"
 	"strings"
+
+	log "github.com/cihub/seelog"
+	"github.com/fsnotify/fsnotify"
 )
 
 const (
@@ -27,3 +30,51 @@ func (c *client) CheckTopicAuth(typ int, topic string) bool {
 	return acl.CheckTopicAuth(aclInfo, typ, ip, username, clientid, topic)
 
 }
+
+var (
+	watchList = []string{"./conf"}
+)
+
+func (b *Broker) handleFsEvent(event fsnotify.Event) error {
+	switch event.Name {
+	case b.config.AclConf:
+		if event.Op&fsnotify.Write == fsnotify.Write ||
+			event.Op&fsnotify.Create == fsnotify.Create {
+			log.Info("text:handling acl config change event:", event)
+			aclconfig, err := acl.AclConfigLoad(event.Name)
+			if err != nil {
+				log.Error("aclconfig change failed, load acl conf error: ", err)
+				return err
+			}
+			b.AclConfig = aclconfig
+		}
+	}
+	return nil
+}
+
+func (b *Broker) StartAclWatcher() {
+	go func() {
+		wch, e := fsnotify.NewWatcher()
+		if e != nil {
+			log.Error("start monitor acl config file error,", e)
+			return
+		}
+		defer wch.Close()
+
+		for _, i := range watchList {
+			if err := wch.Add(i); err != nil {
+				log.Error("start monitor acl config file error,", err)
+				return
+			}
+		}
+		log.Info("watching acl config file change...")
+		for {
+			select {
+			case evt := <-wch.Events:
+				b.handleFsEvent(evt)
+			case err := <-wch.Errors:
+				log.Error("error:", err.Error())
+			}
+		}
+	}()
+}

broker/broker.go

@@ -54,6 +54,7 @@ func NewBroker(config *Config) *Broker {
 			return nil
 		}
 		b.AclConfig = aclconfig
+		b.StartAclWatcher()
 	}
 	return b
 }