package main

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

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

func nSenderImpl(s deliver.Deliver, index int) {
	var err error

	buf := make([]byte, dLen)

	for {
		select {
		case <-ctx.Done():
			return
		default:
			if err = s.Send(buf); err != nil {

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

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

}

func nSender(url string, m deliver.Mode, count int, args ...interface{}) {

	var cs []deliver.Deliver

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

		go nSenderImpl(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()
	}

}

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

	var msg []byte
	var err error

	var t int64
	var elapse int64
	count := 0

	for {
		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
		}

		// time.Sleep(10 * time.Millisecond)
	}
}

func oneReciever(url string, m deliver.Mode) {

	s := deliver.NewServer(m, url)

	go oneRecvImpl(s, 0)

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

	cancel()
	s.Close()
}