package main

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

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

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

	buf := make([]byte, dLen)

	copy(buf, []byte("hello, give you this"))
	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))
			}
		}

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

}

func shmReciever2(url string, count int, args ...interface{}) {
	s := deliver.NewServer(deliver.Shm, url, args...)

	go shmRecvImpl2(s, 0)

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

	cancel()
	s.Close()
}

func shmRecvImpl2(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 %s len %d, use \"%d\" ns\n",
					index, count, string(msg), len(msg), elapse)
				elapse = 0
				count = 0
				t = 0
			}
		}

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

}

func shmSender2(url string, count int) {

	var cs []deliver.Deliver
	for i := 0; i < count; i++ {
		c := deliver.NewClient(deliver.Shm, url)
		cs = append(cs, c)
		go shmSenderImpl2(c)
	}

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

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