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     door_cmd_addr_ string
  31     use_syslog_ bool
  32     enable_debuglog_ bool
  33 )
  34
  35 func usage() {
  36     fmt.Fprintf(os.Stderr, "Usage: zmq_broker_event_transformer [options]\n")
  37     flag.PrintDefaults()
  38 }
  39
  40 func init() {
  41     flag.StringVar(&door_cmd_addr_, "doorcmdaddr", "ipc:///run/tuer/door_cmd.ipc", "zmq door event publish addr")
  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.BoolVar(&enable_debuglog_, "debug", false, "enable debug logging")
  49     flag.Usage = usage
  50     flag.Parse()
  51 }
  52
  53 func main() {
  54     if enable_debuglog_ { LogEnableDebuglog() }
  55     if use_syslog_ {
  56         LogEnableSyslog()
  57         Syslog_.Print("started")
  58         defer Syslog_.Print("exiting")
  59     }
  60
  61     zmqctx, sub_in_chans, pub_out_socket, keylookup_socket := ZmqsInit(doorsub_addr_, sensorssub_port_, pub_port_, keylookup_addr_)
  62     if sub_in_chans != nil {defer sub_in_chans.Close()}
  63     defer zmqctx.Close()
  64     if pub_out_socket != nil {defer pub_out_socket.Close()}
  65     if keylookup_socket != nil {defer keylookup_socket.Close()}
  66     if sub_in_chans == nil || pub_out_socket == nil || keylookup_socket == nil {
  67         panic("zmq sockets must not be nil !!")
  68     }
  69
  70     ps := pubsub.New(10)
  71     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
  72     //~ ticker := time.NewTicker(time.Duration(5) * time.Minute)
  73
  74     store_these_events_chan := ps.Sub("door", "doorcmd", "presence", "sensors", "buttons", "movement")
  75     go BrainCenter(zmqctx, brain_listen_addr_, store_these_events_chan)
  76
  77     go MetaEventRoutine_Movement(ps, 10, 20, 10)
  78     go MetaEventRoutine_Presence(ps)
  79
  80     // --- get update on most recent status ---
  81     answ := ZmqsAskQuestionsAndClose(zmqctx, door_cmd_addr_, [][][]byte{[][]byte{[]byte("status")}})
  82     for _, a := range(answ) {
  83         ParseSocketInputLine(a, ps, keylookup_socket)
  84     }
  85
  86     publish_these_events_chan := ps.Sub("door", "doorcmd", "presence", "sensors", "buttons", "movement")
  87     for {
  88         select {
  89             case subin := <- sub_in_chans.In():
  90                 ParseSocketInputLine(subin, ps, keylookup_socket)
  91             //~ case <- ticker.C:
  92                 //~ MakeTimeTick(ps)
  93             case event_interface := <- publish_these_events_chan:
  94                 data, err := r3events.MarshalEvent2ByteByte(event_interface)
  95                 Debug_.Printf("publishing %s",data)
  96                 if err != nil {
  97                     Syslog_.Print(err)
  98                     continue
  99                 }
 100                 if err := pub_out_socket.Send(data); err != nil {
 101                     panic(err)
 102                 }
 103         }
 104     }
 105
 106 }