package main

import (
	"fmt"
	"log"
	"net"
	"net/http"

	"github.com/arl/statsviz"
)

func produce(l *net.TCPListener, c chan<- net.Conn) {
	for {
		conn, _ := l.Accept()
		c <- conn
		defer conn.Close()
	}
}

func consume(c <-chan net.Conn) {
	fmt.Println("create goroutine consumer")
	for conn := range c {
		log.Println("someone connected")
		var res = make([]byte, 10)
		nb, err := conn.Read(res)
		if err != nil {
			log.Fatalln(err)
		}
		fmt.Printf("receive %d, contain %s\n", nb, string(res))

		conn.Write([]byte("HTTP/1.1 200 OK\nContent-Type: text/html; encoding=utf8\nContent-Length: 15\nConnection: close\n\n<h1>HELLO</h1>\n"))
		//<-time.After(10 * time.Second)
		fmt.Println("Process ended")
	}
}

func main() {
	c := make(chan net.Conn, 10)
	for i := 0; i < 10; i++ {
		go consume(c)
	}

	addr := net.TCPAddr{
		IP:   net.ParseIP("127.0.0.1"),
		Port: 3000,
	}
	l, err := net.ListenTCP("tcp", &addr)
	if err != nil {
		log.Fatalln(err)
	}

	statsviz.RegisterDefault()
	go func() {
		log.Println("start listening on port 6060 for statsviz")
		log.Println(http.ListenAndServe(":6060", nil))
	}()

	produce(l, c)
}