Beispiel #1
0
func init() {
	flag.IntVar(&port, "port", 10001, "Port/IP to use for the REST interface. Overrides $PORT0 env variable")
	flag.StringVar(&logPath, "logPath", "/var/log/vamp-router/vamp-router.log", "Location of the log file")
	flag.StringVar(&configPath, "configPath", "", "Location of configuration files, defaults to configuration/")
	flag.StringVar(&binaryPath, "binary", helpers.HaproxyLocation(), "Path to the HAproxy binary")
	flag.StringVar(&kafkaHost, "kafkaHost", "", "The hostname or ip address of the Kafka host")
	flag.IntVar(&kafkaPort, "kafkaPort", 9092, "The port of the Kafka host")
	flag.StringVar(&zooConString, "zooConString", "", "A zookeeper ensemble connection string")
	flag.StringVar(&zooConKey, "zooConKey", "magneticio/vamplb", "Zookeeper root key")
	flag.StringVar(&customWorkDir, "customWorkDir", "", "Custom working directory for sockets and pid files, default to data/")
	flag.BoolVar(&headless, "headless", false, "Run without any logging output to the console")
}
Beispiel #2
0
func TestApi_CreateAPI(t *testing.T) {

	log := gologger.MustGetLogger("vamp-router")

	sseChannel := make(chan metrics.Metric)

	sseBroker := &metrics.SSEBroker{
		make(map[chan metrics.Metric]bool),
		make(chan (chan metrics.Metric)),
		make(chan (chan metrics.Metric)),
		sseChannel,
		log,
	}

	haConfig := haproxy.Config{TemplateFile: TEMPLATE_FILE, ConfigFile: CONFIG_FILE, JsonFile: JSON_FILE, PidFile: PID_FILE}
	haRuntime := haproxy.Runtime{Binary: helpers.HaproxyLocation()}

	if _, err := CreateApi(log, &haConfig, &haRuntime, sseBroker, "v.test"); err != nil {
		t.Errorf("Failed to create API")
	}

}
Beispiel #3
0
package haproxy

import (
	"github.com/magneticio/vamp-router/helpers"
	"io/ioutil"
	"os"
	"os/exec"
	"testing"
)

var (
	haRuntime = Runtime{Binary: helpers.HaproxyLocation(), SockFile: "/tmp/haproxy.stat.sock"}
)

func TestRuntime_SetNewPid(t *testing.T) {

	//make sure there is no pidfile present
	os.Remove(PID_FILE)

	if err := haRuntime.SetPid(PID_FILE); err != nil {
		t.Fatalf(err.Error())
	}

}

func TestRuntime_UseExistingPid(t *testing.T) {

	//create a pid file
	emptyPid := []byte("12356")
	ioutil.WriteFile(PID_FILE, emptyPid, 0644)
	defer os.Remove(PID_FILE)