package syncdb

import (
	"basic.com/valib/logger.git"
	"bufio"
	"bytes"
	"encoding/binary"
	"io"
	"net"
	"strconv"
)

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() {

	tcpAddr := "0.0.0.0:"+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
}