123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249 |
- 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))
- if err != nil {
- return 0, err
- }
- } else {
- ret, err = rhSendInt32(pConn, uint32(0))
- if err != nil {
- return 0, err
- }
- }
- 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("?"))
-
- sqlLength := uint32(len(sql))
- var networkBytes [4]byte
- binary.BigEndian.PutUint32(networkBytes[:], sqlLength)
- pConn.sendBuff.Write(networkBytes[:])
-
- pConn.sendBuff.Write(sql)
-
- binary.BigEndian.PutUint32(networkBytes[:], 0)
- pConn.sendBuff.Write([]byte{0})
-
- 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 不为空")
-
- for _, value := range *values {
-
- 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())
-
- if value.paramName == nil {
-
- } else {
- pConn.sendBuff.Write(value.paramName[:])
- fmt.Println("Param_name sendBuff ", pConn.sendBuff.Bytes())
- }
-
- Param_INOUT := [2]byte{0x1}
- pConn.sendBuff.Write(reverseBytes(Param_INOUT[:]))
- fmt.Println("Param_INOUT sendBuff ", pConn.sendBuff.Bytes())
-
- 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())
-
- 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())
-
- 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())
- }
- 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
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- } else {
-
- 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 {
-
-
- 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 {
-
- pConn.sendBuff.Write(buff)
- return length, nil
- }
- }
- }
- func rhSendInt32(pConn *xuguConn, i uint32) (int, error) {
-
- var networkBytes [4]byte
-
- binary.BigEndian.PutUint32(networkBytes[:], i)
-
- 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")
-
-
-
- pConn.conn.Write(pConn.sendBuff.Bytes())
- fmt.Println("\n ---XGC_Execute end")
- return 0
- }
|