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