xugu_driver_go/xugu/xugu_sock_send.go

package xugu

import (
	"encoding/binary"
	"fmt"
)

func SockSendCommandN(pConn *xuguConn, sql string, num int) (int, error) {
	ret := 0
	var err error
	cmd_len := len(sql)
	ret, err = RhSend(pConn, []byte("?"), 1)
	if err != nil {
		return 0, err
	}
	ret, err = rhSendInt32(pConn, uint32(cmd_len))
	if err != nil {
		return 0, err
	}
	ret, err = RhSend(pConn, []byte(sql), cmd_len+1)
	if err != nil {
		return 0, err
	}
	if num != 0 {
		ret, err = rhSendInt32(pConn, uint32(num)) //param num
		if err != nil {
			return 0, err
		}
	} else {
		ret, err = rhSendInt32(pConn, uint32(0))
		if err != nil {
			return 0, err
		}
	}

	return ret, nil
}

// func SockSendCommand0(pConn *xuguConn, sql string) (int, error) {
// 	cmdLen := len(sql)
// 	//发送问号字符 ?
// 	ret, err := RhSend(pConn, []byte("?"), cmdLen)
// 	if err != nil {
// 		return 0, err
// 	}
// 	// Comand_Len
// 	lengthInt32 := uint32(len(sql))
// 	ret, err = rhSendInt32(pConn, lengthInt32)
// 	if err != nil {
// 		return 0, err
// 	}
// 	//  Comand_str
// 	ret, err = RhSend(pConn, []byte(sql), cmdLen+1)
// 	if err != nil {
// 		return 0, err
// 	}

// 	data := make([]byte, 1)
// 	ret, err = RhSend(pConn, data, 1)
// 	if err != nil {
// 		return 0, err
// 	}

// 	// Param_num
// 	ret, err = rhSendInt32(pConn, 0)
// 	if err != nil {
// 		return 0, err
// 	}
// 	fmt.Println("SockSendCommand0 msg", pConn.sendBuff.String())

// 	RhFlush(pConn)

// 	return ret, nil
// }

func sockSendPutStatement(pConn *xuguConn, sql []byte, values *[]xuguValue, paramCount int) {
	if pConn.sendBuff.Len() > 0 {
		//将缓冲区重置为空
		pConn.sendBuff.Reset()
	}
	// ?
	pConn.sendBuff.Write([]byte("?"))
	// Comand_Len
	sqlLength := uint32(len(sql))
	var networkBytes [4]byte
	binary.BigEndian.PutUint32(networkBytes[:], sqlLength)
	pConn.sendBuff.Write(networkBytes[:])

	//  Comand_str
	pConn.sendBuff.Write(sql)
	//'0' 结束符
	binary.BigEndian.PutUint32(networkBytes[:], 0)
	pConn.sendBuff.Write([]byte{0})

	// Param_num
	if paramCount > 0 {
		fmt.Println("paramCount> 0: ", paramCount)
		binary.BigEndian.PutUint32(networkBytes[:], uint32(paramCount))
		pConn.sendBuff.Write(networkBytes[:])
	} else {
		binary.BigEndian.PutUint32(networkBytes[:], 0)
		pConn.sendBuff.Write(networkBytes[:])
	}

	if values != nil {
		fmt.Println("values 不为空")
		//	Param_num   { Param_name_len Param_name Param_INOUT Param_DType Param_Data_Len Param_Data }
		for _, value := range *values {
			//Param_name_len
			var Param_name_len [2]byte
			binary.BigEndian.PutUint16(Param_name_len[:], uint16(len(value.paramName)))
			pConn.sendBuff.Write(Param_name_len[:])
			fmt.Println("Param_name_len sendBuff ", pConn.sendBuff.Bytes())
			//Param_name
			if value.paramName == nil {
				//pConn.sendBuff.Write([]byte{0})
			} else {
				pConn.sendBuff.Write(value.paramName[:])
				fmt.Println("Param_name sendBuff ", pConn.sendBuff.Bytes())
			}

			//Param_INOUT
			Param_INOUT := [2]byte{0x1}
			pConn.sendBuff.Write(reverseBytes(Param_INOUT[:]))
			fmt.Println("Param_INOUT sendBuff ", pConn.sendBuff.Bytes())
			//Param_DType
			var Param_DType [2]byte
			binary.BigEndian.PutUint16(Param_DType[:], uint16(value.types))
			pConn.sendBuff.Write(Param_DType[:])
			fmt.Println("Param_DType sendBuff ", pConn.sendBuff.Bytes())
			//Param_Data_Len
			var Param_Data_Len [4]byte
			binary.BigEndian.PutUint32(Param_Data_Len[:], uint32(value.valueLength))
			pConn.sendBuff.Write(Param_Data_Len[:])
			fmt.Println("Param_Data_Len sendBuff ", pConn.sendBuff.Bytes())
			//Param_Data
			pConn.sendBuff.Write([]byte(value.value))
			fmt.Println("Param_Data sendBuff ", pConn.sendBuff.Bytes())
		}

	}
	fmt.Println("SockSendPutStatement msg", pConn.sendBuff.String())
	fmt.Println("SockSendPutStatement msg", pConn.sendBuff.Bytes())
}

// RhSend Go 版本的 rh_send 函数
// TODO :加密连接还没有实现
func RhSend(pConn *xuguConn, buff []byte, length int) (int, error) {
	pConn.mu.Lock()
	defer pConn.mu.Unlock()
	//没有内容则不发送
	if length == 0 {
		return 0, nil
	}
	//连接里是否有加密
	if pConn.useSSL {
		fmt.Println("加密未实现")
		return length, nil
		// if length > 2*1024 {
		// 	EncryptBuff(pconn, buff)
		// 	if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
		// 		pconn.sendBuff.Reset()
		// 		return 0, err
		// 	}
		// 	pconn.sendBuff.Reset()
		// 	_, err := pconn.conn.Write(bufPrintln()
		// 	return len(buff), err
		// } else if pconn.sendBuff.Len()+length >= BUFF_SIZE {
		// 	if _, err := pconn.conn.Write(pconn.sendBuff.Bytes()); err != nil {
		// 		pconn.sendBuff.Reset()
		// 		return 0, err
		// 	}
		// 	pconn.sendBuff.Reset()
		// 	_, err := pconn.sendBuff.Write(buff)
		// 	if err != nil {
		// 		return 0, err
		// 	}
		// 	EncryptBuff(pconn, pconn.sendBuff.Bytes())
		// 	return len(buff), nil
		// } else {
		// 	pconn.sendBuff.Write(buff)
		// 	EncryptBuff(pconn, pconn.sendBuff.Bytes()[pconn.sendBuff.Len()-length:])
		// 	return length, nil
		// }
	} else {
		/*如果数据长度大于 2048，先发送缓冲区中的数据，然后直接发送新的大数据块。*/
		if length > 2048 {
			if _, err := pConn.conn.Write(pConn.sendBuff.Bytes()); err != nil {
				//将缓冲区重置为空
				pConn.sendBuff.Reset()
				return 0, err
			}
			//将缓冲区重置为空
			pConn.sendBuff.Reset()
			//发送新的大数据块
			_, err := pConn.conn.Write(buff)
			return length, err

		} else if pConn.sendBuff.Len()+length >= 8*1024 {
			//缓冲区空间不足：
			/*- 如果当前缓冲区中的数据加上新数据的长度超过 `8 * 1024` 字节，
			则先发送缓冲区中的数据，然后重置缓冲区，将新数据拷贝到缓冲区中。*/
			if _, err := pConn.conn.Write(pConn.sendBuff.Bytes()); err != nil {
				pConn.sendBuff.Reset()
				return 0, err
			}
			pConn.sendBuff.Reset()
			pConn.sendBuff.Write(buff)
			return length, nil
		} else {
			//TODO: 这里c代码里是添加到缓存buff里面 ，不直接发送
			pConn.sendBuff.Write(buff)
			return length, nil
		}
	}
}

func rhSendInt32(pConn *xuguConn, i uint32) (int, error) {

	// 创建一个 4 字节的缓冲区，用于存储网络字节序的 int32 值
	var networkBytes [4]byte
	// 将 int32 值转换为网络字节序（大端字节序）并写入缓冲区
	binary.BigEndian.PutUint32(networkBytes[:], i)

	//fmt.Println("networkBytes[] ", networkBytes[:])

	return RhSend(pConn, networkBytes[:], 4)
}

func RhFlush(pConn *xuguConn) (int, error) {
	if pConn.sendBuff.Len() > 0 {
		if _, err := pConn.conn.Write(pConn.sendBuff.Bytes()); err != nil {
			pConn.sendBuff.Reset()
			return 0, err
		}
		pConn.sendBuff.Reset()
	}
	return 0, nil
}

func XGC_Execute(pConn *xuguConn) int {
	fmt.Println("\n ---XGC_Execute")
	//发送没有参数的sql
	//SockSendCommand0(pConn, sql)
	//sockSendPutStatement(pConn, []byte(sql), values)
	pConn.conn.Write(pConn.sendBuff.Bytes())
	fmt.Println("\n ---XGC_Execute end")
	return 0
}