update

consul.go

@@ -3,13 +3,12 @@ package main
 import (
 	"fmt"
 	"log"
-	"strconv"
-	"strings"
 	"time"
 
 	"github.com/hashicorp/consul/api"
 )
 
+// Consul client structure
 type Consul struct {
 	RedisTTL  time.Duration
 	Timeout   time.Duration
@@ -20,11 +19,26 @@ type Consul struct {
 	MongoURI  string
 	Database  string
 	RedisURI  string
+	Client    *api.Client
 }
 
-func getClient() {
+// NewClient New consul client
+func NewClient() *Consul {
 	client, err := api.NewClient(api.DefaultConfig())
 	if err != nil {
-		panic(err)
+		log.Fatalf("Consul error: %+v\n", err)
+		if opts.Debug {
+			fmt.Printf("Consul error: %+v\n", err)
+		}
 	}
+
+	consul := &Consul{
+		Client:   client,
+		RedisTTL: opts.RedisTTL,
+		Timeout:  opts.Timeout,
+		MaxError: opts.MaxError,
+		Influxdb: opts.Influxdb,
+	}
+
+	return consul
 }

counter.go

@@ -6,6 +6,7 @@ import (
 	"time"
 )
 
+// Counterchan counter structure
 type Counterchan struct {
 	tipo string
 	val  int
@@ -38,6 +39,7 @@ func NewCounter() *Counter {
 	}
 }
 
+// Run open channel to receive counter
 func (c *Counter) Run() {
 	for {
 		tocount := <-counter

options.go

@@ -28,6 +28,7 @@ type Options struct {
 	Queue       int
 	Retention   float64
 	Consul      string
+	Port        int
 }
 
 var (
@@ -47,6 +48,7 @@ func usage() {
 		-T <running timeout> 
 		-i <influxdb [localname@ip:port]>
 		-C <consul [ip:port]>
+		-P <check port> [port] ## used by consul to check services ##
 		-q <parallel consumer>
 		-R <retention>
 		-v <version>
@@ -72,6 +74,7 @@ func init() {
 	flag.StringVar(&dbs.RedisURI, "r", "", "Redis")
 	flag.StringVar(&opts.LogFile, "l", opts.LogFile, "Logs filename")
 	flag.StringVar(&opts.Consul, "C", opts.Consul, "consul client")
+	flag.IntVar(&opts.Port, "P", 10500, "service check port")
 	flag.DurationVar(&opts.RedisTTL, "t", opts.RedisTTL, "Redis keys TTL")
 	flag.BoolVar(&opts.Version, "v", false, "Version")
 	flag.DurationVar(&opts.Timeout, "T", 0, "Running timeout")