gateway/gateway.go

/*******************************************************************************
 * Copyright (c) 2014-2015 IBM Corp.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Allan Stockdill-Mander
 *    Seth Hoenig
 *******************************************************************************/

package main

import (
	"flag"
	"os"
	"os/signal"

	G "git.eclipse.org/gitroot/paho.incubator/smidge.git/gateway/gate"
)

func main() {
	var gateway G.Gateway
	stopsig := registerSignals()
	gatewayconf := setup()

	G.InitLogger(os.Stdout, os.Stderr) // todo: configurable

	if gatewayconf.IsAggregating() {
		G.INFO.Println("SMIDGE Gateway starting in aggregating mode")
		gateway = initAggregating(gatewayconf, stopsig)
	} else {
		G.INFO.Println("SMIDGE Gateway starting in transparent mode")
		gateway = initTransparent(gatewayconf, stopsig)
	}

	gateway.Start()
}

func setup() *G.GatewayConfig {
	var configFile string
	var port int

	flag.StringVar(&configFile, "c", "", "Configuration File")
	flag.IntVar(&port, "port", 0, "MQTT-G UDP Listening Port")
	flag.Parse()

	if configFile != "" {
		if gc, err := G.ParseConfigFile(configFile); err != nil {
			G.ERROR.Fatal(err)
		} else {
			return gc
		}
	}

	G.ERROR.Fatal("-configuration <file> must be specified")
	return nil
}

func initAggregating(c *G.GatewayConfig, stopsig chan os.Signal) *G.AGateway {
	a := G.NewAGateway(c, stopsig)
	return a
}

func initTransparent(c *G.GatewayConfig, stopsig chan os.Signal) *G.TGateway {
	t := G.NewTGateway(c, stopsig)
	return t
}

func registerSignals() chan os.Signal {
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, os.Kill)
	return c
}