go/r3-eventbroker_zmq/main.go

   1 // (c) Bernhard Tittelbach, 2013
   2
   3 package main
   4
   5 import (
   6     "fmt"
   7     "os"
   8     "flag"
   9     //~ "time"
  10     "log/syslog"
  11     "log"
  12     pubsub "github.com/tuxychandru/pubsub"
  13     "./r3events"
  14 )
  15
  16 //~ func StringArrayToByteArray(ss []string) [][]byte {
  17     //~ bb := make([][]byte, len(ss))
  18     //~ for index, s := range(ss) {
  19         //~ bb[index] = []byte(s)
  20     //~ }
  21     //~ return bb
  22 //~ }
  23
  24 // ---------- Main Code -------------
  25
  26 var (
  27     doorsub_addr_ string
  28     sensorssub_port_ string
  29     pub_port_ string
  30     keylookup_addr_ string
  31     brain_listen_addr_ string
  32     use_syslog_ bool
  33     Syslog_ *log.Logger
  34 )
  35
  36 func usage() {
  37     fmt.Fprintf(os.Stderr, "Usage: zmq_broker_event_transformer [options]\n")
  38     flag.PrintDefaults()
  39 }
  40
  41 func init() {
  42     flag.StringVar(&doorsub_addr_, "doorsubaddr", "tcp://torwaechter.realraum.at:4242", "zmq door event publish addr")
  43     flag.StringVar(&sensorssub_port_, "sensorsubport", "tcp://*:4243", "zmq public/listen socket addr for incoming sensor data")
  44     flag.StringVar(&pub_port_, "pubport", "tcp://*:4244", "zmq port publishing consodilated events")
  45     flag.StringVar(&keylookup_addr_, "keylookupaddr", "ipc:///run/tuer/door_keyname.ipc", "address to use for key/name lookups")
  46     flag.StringVar(&brain_listen_addr_, "brainlisten", "tcp://*:4245", "address to listen for requests about latest stored event")
  47     flag.BoolVar(&use_syslog_, "syslog", false, "log to syslog local2 facility")
  48     flag.Usage = usage
  49     flag.Parse()
  50 }
  51
  52 func main() {
  53     zmqctx, sub_in_chans, pub_out_socket, keylookup_socket := ZmqsInit(doorsub_addr_, sensorssub_port_, pub_port_, keylookup_addr_)
  54     if sub_in_chans != nil {defer sub_in_chans.Close()}
  55     defer zmqctx.Close()
  56     if pub_out_socket != nil {defer pub_out_socket.Close()}
  57     if keylookup_socket != nil {defer keylookup_socket.Close()}
  58     if sub_in_chans == nil || pub_out_socket == nil || keylookup_socket == nil {
  59         panic("zmq sockets must not be nil !!")
  60     }
  61
  62     if use_syslog_ {
  63         var logerr error
  64         Syslog_, logerr = syslog.NewLogger(syslog.LOG_INFO | (18<<3), 0)
  65         //~ Syslog_, logerr = syslog.NewLogger(syslog.LOG_INFO | syslog.LOG_LOCAL2, 0)
  66         if logerr != nil { panic(logerr) }
  67         Syslog_.Print("started")
  68         defer Syslog_.Print("exiting")
  69     }
  70
  71     ps := pubsub.New(10)
  72     defer ps.Shutdown() // ps.Shutdown should be called before zmq_ctx.Close(), since it will cause goroutines to shutdown and close zqm_sockets which is needed for zmq_ctx.Close() to return
  73     //~ ticker := time.NewTicker(time.Duration(5) * time.Minute)
  74
  75     store_these_events_chan := ps.Sub("door", "doorcmd", "presence", "sensors", "buttons", "movement")
  76     go BrainCenter(zmqctx, brain_listen_addr_, store_these_events_chan)
  77
  78     go MetaEventRoutine_Movement(ps, 10, 20, 10)
  79     go MetaEventRoutine_Presence(ps)
  80
  81     publish_these_events_chan := ps.Sub("door", "doorcmd", "presence", "sensors", "buttons", "movement")
  82     for {
  83         select {
  84             case subin := <- sub_in_chans.In():
  85                 ParseSocketInputLine(subin, ps, keylookup_socket)
  86             //~ case <- ticker.C:
  87                 //~ MakeTimeTick(ps)
  88             case event_interface := <- publish_these_events_chan:
  89                 data, err := r3events.MarshalEvent2ByteByte(event_interface)
  90                 log.Printf("publishing %s",data)
  91                 if err != nil {
  92                     if Syslog_ != nil {Syslog_.Print(err)}
  93                     continue
  94                 }
  95                 if err := pub_out_socket.Send(data); err != nil {
  96                     panic(err)
  97                 }
  98         }
  99     }
 100
 101 }