Merge pull request #15 from D4-project/tcpreuse

Tcpreuse

d4-goclient.go

@@ -200,7 +200,13 @@ func main() {
 			// copy routine
 			go d4Copy(d4p, c)
 			// Block until the rate limiter allow us to continue
-			<-ratelimiter
+			select {
+			case <-ratelimiter:
+				continue
+			case <-s:
+				logger.Println("Exiting")
+				exit(d4p, 0)
+			}
 		} else if d4.retry > 0 {
 			go func() {
 				logger.Println(fmt.Sprintf("Sleeping for %.f seconds before retry...", d4.retry.Seconds()))
@@ -266,7 +272,7 @@ func d4loadConfig(d4 *d4S) bool {
 	}
 	if (*d4).conf.source == "d4server" {
 		// Parse Input Redis Config
-		tmp := config.ReadConfigFile(*confdir, "redis_d4")
+		tmp := string(readConfFile(d4, "redis_d4"))
 		ss := strings.Split(string(tmp), "/")
 		if len(ss) <= 1 {
 			log.Fatal("Missing Database in Redis input config: should be host:port/database_name")
@@ -414,6 +420,12 @@ func setReaderWriters(d4 *d4S) bool {
 	}
 	isn, dstnet := config.IsNet((*d4).conf.destination)
 	if isn {
+		// First, we test whether a usable connection already exist
+		// (case where the reader run out of data)
+		switch (*d4).dst.w.(type){
+		case net.Conn:
+			// in this case, it's already set up.
+		default:
 			dial := net.Dialer{
 				Timeout:       (*d4).ct,
 				KeepAlive:     (*d4).cka,
@@ -442,6 +454,7 @@ func setReaderWriters(d4 *d4S) bool {
 				}
 				(*d4).dst = newD4Writer(conn, (*d4).conf.key)
 			}
+		}
 	} else {
 		switch (*d4).conf.destination {
 		case "stdout":
@@ -466,7 +479,7 @@ func generateUUIDv4() []byte {
 	if err != nil {
 		log.Fatal(err)
 	}
-	logger.Println(fmt.Sprintf("UUIDv4: %s\n", uuid))
+	logger.Println(fmt.Sprintf("UUIDv4: %s", uuid))
 	return uuid.Bytes()
 }