package main

import (
	"demo/deliver"
	"fmt"
	"os"
	"os/signal"
	"time"

	"golang.org/x/sys/unix"
)

func oneSenderImpl(s deliver.Deliver) {
	var err error

	buf := make([]byte, dLen)

	for {
		select {
		case <-ctx.Done():
			return
		default:

			if err = s.Send(buf); err != nil {

				fmt.Printf("can't send message on push socket: %s\n", err.Error())
			} else {

				fmt.Printf("send msg length %d\n", len(buf))
			}
		}
	}

}

func oneSender(url string, m deliver.Mode, args ...interface{}) {
	s := deliver.NewServer(m, url, args...)

	go oneSenderImpl(s)

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
	<-c

	cancel()
	s.Close()
}

func nRecvImpl(c deliver.Deliver, index int) {

	var msg []byte
	var err error

	var t int64
	var elapse int64
	count := 0

	for {
		select {
		case <-ctx.Done():
			return
		default:
			msg, err = c.Recv()
			if err != nil {
				fmt.Println("recv error : ", err)
			}
			if t == 0 {
				t = time.Now().UnixNano()
			}
			elapse = time.Now().UnixNano() - t

			count++

			if elapse > 1e9 {
				fmt.Printf("NODE-%d: RECEIVED \"%d\" data len %d, use \"%d\" ns\n",
					index, count, len(msg), elapse)
				elapse = 0
				count = 0
				t = 0
			}
		}

	}
}

func nReciever(url string, m deliver.Mode, count int) {

	var cs []deliver.Deliver

	for i := 0; i < count; i++ {
		c := deliver.NewClient(m, url)
		cs = append(cs, c)

		go nRecvImpl(c, i)
	}

	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, os.Kill, unix.SIGTERM)
	<-c

	cancel()
	for _, v := range cs {
		v.Close()
	}

}