daemon.go | 17 +++++++++-------- events.go | 16 +++++++++++----- diff --git a/daemon.go b/daemon.go index a94cb709bbdc06806a966691d222bc158e4b8e8572c58bb0351d3e23420bb442..cae01970d40f173f6ca8ece99bb7e3a1c78f41db274e84532e81a9758ca6ce04 100644 --- a/daemon.go +++ b/daemon.go @@ -48,7 +48,7 @@ version string hostname *string motd *string passwords *string - clients map[*Client]bool + clients map[*Client]struct{} clientAliveness map[*Client]*ClientAlivenessState rooms map[string]*Room roomSinks map[*Room]chan ClientEvent @@ -59,12 +59,12 @@ } func NewDaemon(version string, hostname, motd, passwords *string, logSink chan<- LogEvent, stateSink chan<- StateEvent) *Daemon { daemon := Daemon{ - version: version, - hostname: hostname, - motd: motd, + version: version, + hostname: hostname, + motd: motd, passwords: passwords, } - daemon.clients = make(map[*Client]bool) + daemon.clients = make(map[*Client]struct{}) daemon.clientAliveness = make(map[*Client]*ClientAlivenessState) daemon.rooms = make(map[string]*Room) daemon.roomSinks = make(map[*Room]chan ClientEvent) @@ -303,8 +303,9 @@ } } func (daemon *Daemon) Processor(events <-chan ClientEvent) { + var now time.Time for event := range events { - now := time.Now() + now = time.Now() client := event.client // Check for clients aliveness @@ -334,9 +335,9 @@ } switch event.eventType { case EventNew: - daemon.clients[client] = true + daemon.clients[client] = struct{}{} daemon.clientAliveness[client] = &ClientAlivenessState{ - pingSent: false, + pingSent: false, timestamp: now, } case EventDel: diff --git a/events.go b/events.go index 3f597a1671c80f4576fdc99701994ee9107b07d2c1e64a66971a57c815996d58..da2452382dfc4363968f0ccfa3dbc3bd2147c9b4e9e7d87be418ed0b8cafd782 100644 --- a/events.go +++ b/events.go @@ -66,9 +66,12 @@ func Logger(logdir string, events <-chan LogEvent) { mode := os.O_CREATE | os.O_WRONLY | os.O_APPEND perm := os.FileMode(0660) var format string + var logfile string + var fd *os.File + var err error for event := range events { - logfile := path.Join(logdir, event.where) - fd, err := os.OpenFile(logfile, mode, perm) + logfile = path.Join(logdir, event.where) + fd, err = os.OpenFile(logfile, mode, perm) if err != nil { log.Println("Can not open logfile", logfile, err) continue @@ -96,10 +99,13 @@ // Room state events saver // Room states shows that either topic or key has been changed // Each room's state is written to separate file in statedir func StateKeeper(statedir string, events <-chan StateEvent) { + var fn string + var data string + var err error for event := range events { - fn := path.Join(statedir, event.where) - data := event.topic + "\n" + event.key + "\n" - err := ioutil.WriteFile(fn, []byte(data), os.FileMode(0660)) + fn = path.Join(statedir, event.where) + data = event.topic + "\n" + event.key + "\n" + err = ioutil.WriteFile(fn, []byte(data), os.FileMode(0660)) if err != nil { log.Printf("Can not write statefile %s: %v", fn, err) }