package serf

import (
	"basic.com/valib/logger.git"
	"bufio"
	"bytes"
	"encoding/binary"
	"io"
	"net"
	"strconv"
	"vamicro/config"
	"vamicro/system-service/sys"
)

func rawSendTcpMsg(addr string, sendBuf []byte) error {
	conn, err := net.Dial("tcp", addr)
	if err != nil {
		logger.Debug("net.Dialt err", err)
		return err
	}

	defer conn.Close()

	//发送
	sizeBuf := make([]byte, 4)
	var buf bytes.Buffer
	binary.BigEndian.PutUint32(sizeBuf, uint32(len(sendBuf)))
	buf.Write(sizeBuf)
	buf.Write(sendBuf)
	_, err = conn.Write(buf.Bytes())
	if err != nil {
		logger.Debug("conn.Write err", err)
		return err
	} else {
		logger.Debug("raw send success")
		return nil
	}
}

func RawReceiveTcpMsg() {
	localIp, _, _ := sys.GetLocalIP(config.Server.NetworkAdapter)
	if localIp == "" {
		localIp = "0.0.0.0"
	}
	tcpAddr := localIp + ":" + strconv.Itoa(TcpTransportPort)
	listener, err := net.Listen("tcp", tcpAddr)
	if err != nil {
		logger.Debug("RawReceive server listen err:", err)
		return
	}
	defer listener.Close()

	for {
		conn, err := listener.Accept()
		if err != nil {
			logger.Debug("listener.Accept err:", err)
			continue
		}
		logger.Debug("A transport client connected :" + conn.RemoteAddr().String())
		go readStream(conn)
	}
}

func readStream(conn net.Conn) {
	defer conn.Close()

	var data []byte
	var reader io.Reader = bufio.NewReader(conn)
	sizeBuf := make([]byte, 4)
	if _, err := reader.Read(sizeBuf[:]); err != nil {
		logger.Debug("read tcpStream msg length err:", err)
	} else {
		var msgLen uint32
		binary.Read(bytes.NewBuffer(sizeBuf), binary.BigEndian, &msgLen)
		dataLen := int(msgLen)
		logger.Debug("read tcpStream msg lenth:", dataLen)
		if dataLen > 0 {
			data = make([]byte, dataLen)
			n, err := io.ReadAtLeast(reader, data, dataLen)
			if err == nil {
				logger.Debug("io.ReadAtLeast n:", n)
			} else {
				logger.Debug("io.readAtLeast err:", err)
			}
		}
	}
	QueryTcpResponseChan <- data
}