for spark streaming multiplex

main.go

package main

import (
	"fmt"
	"io"
	"log"
	"net"
	"os"

	"github.com/djherbis/bufit"
)

func fromListen(buf *bufit.Buffer) {
	l, err := net.Listen("tcp", fmt.Sprintf(":%v", os.Getenv("CLOJURE_PORT")))
	if err != nil {
		log.Fatal(err)
	}

	defer l.Close()
	for {
		// Wait for a connection.
		conn, err := l.Accept()
		if err != nil {
			log.Fatal(err)
		}
		// Handle the connection in a new goroutine.
		// The loop then returns to accepting, so that
		// multiple connections may be served concurrently.
		go func(c net.Conn) {
			// Echo all incoming data.
			io.Copy(buf, c)
			// Shut down the connection.
			c.Close()
		}(conn)
	}
}

func toListen(buf *bufit.Buffer) {
	l, err := net.Listen("tcp", fmt.Sprintf(":%v", os.Getenv("SPARK_PORT")))
	if err != nil {
		log.Fatal(err)
	}

	defer l.Close()
	for {
		// Wait for a connection.
		conn, err := l.Accept()
		if err != nil {
			log.Fatal(err)
		}
		// Handle the connection in a new goroutine.
		// The loop then returns to accepting, so that
		// multiple connections may be served concurrently.
		go func(c net.Conn) {
			r := buf.NextReader()

			// Echo all incoming data.
			io.Copy(c, r)
			r.Close()
			// Shut down the connection.
			c.Close()
		}(conn)
	}
}

func main() {
	buf := bufit.New()
	go fromListen(buf)
	toListen(buf)
}