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