localhots/shezmu
1
0
Fork 0
Code Releases Activity

Kafka backend initializer

Browse Source
This commit is contained in:
Gregory Eremin 2015-10-16 02:28:06 +03:00
parent b6f0d2953f
commit 8f35ee060e
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
satan.go
View File

@ -6,6 +6,7 @@ import (
"time" "time"
"github.com/localhots/satan/backend" "github.com/localhots/satan/backend"
"github.com/localhots/satan/backend/kafka"
) )
// Satan is the master daemon. // Satan is the master daemon.
@ -65,8 +66,6 @@ func (s *Satan) StartDaemons() {
// StopDaemons stops all running daemons. // StopDaemons stops all running daemons.
func (s *Satan) StopDaemons() { func (s *Satan) StopDaemons() {
// First closing backend consumers will begin to close
s.backend.Close()
for _, d := range s.daemons { for _, d := range s.daemons {
close(d.base().shutdown) close(d.base().shutdown)
d.Shutdown() d.Shutdown()
@ -77,6 +76,7 @@ func (s *Satan) StopDaemons() {
close(s.shutdown) close(s.shutdown)
s.wg.Wait() s.wg.Wait()
close(s.queue) close(s.queue)
s.backend.Close()
log.Printf("Task processing latency statistics:\n%s\n", s.latency.snapshot()) log.Printf("Task processing latency statistics:\n%s\n", s.latency.snapshot())
} }
@ -103,3 +103,20 @@ func (s *Satan) processTask(t *task) {
dur := time.Now().UnixNano() - start.UnixNano() dur := time.Now().UnixNano() - start.UnixNano()
t.daemon.base().stats.add(time.Duration(dur)) t.daemon.base().stats.add(time.Duration(dur))
} }
// InitializeKafka initializes Kafka backend.
func (s *Satan) InitializeKafka(id string, brokers []string) error {
k, err := kafka.New(id, brokers)
if err != nil {
return err
}
if err = k.InitializeProducer(); err != nil {
return err
}
if err = k.InitializeConsumer(); err != nil {
return err
}
s.backend = k
return nil
}