123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130 |
- package xugu
- import (
- "bytes"
- "context"
- "encoding/binary"
- "errors"
- "fmt"
- )
- func xgSockOpenConn(ctx context.Context, pConn *xuguConn) error {
-
-
-
- dsnMessage := generateLoginString(pConn.dsnConfig)
- _, err := pConn.conn.Write([]byte(dsnMessage))
- if err != nil {
- return errors.New("向数据库发起连接失败")
- }
- buffer := make([]byte, 1)
- n, err := pConn.conn.Read(buffer)
- if err != nil {
- return errors.New("接收数据库连接失败:")
- }
- if !bytes.Equal(buffer[:n], []byte("K")) {
- return errors.New("数据库连接失败")
- } else {
- return nil
- }
- }
- func sockSendPutStatement(pConn *xuguConn, sql []byte, values *[]xuguValue, paramCount int) error {
- 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})
-
- var Param_num [4]byte
- binary.BigEndian.PutUint32(Param_num[:], uint32(paramCount))
- pConn.sendBuff.Write(Param_num[:])
- if values != nil {
-
-
-
-
-
-
-
-
-
-
- for _, value := range *values {
-
- if value.paramName == nil {
- var Param_name_len [2]byte
- pConn.sendBuff.Write(Param_name_len[:])
-
-
-
- } else {
- var Param_name_len [2]byte
- binary.BigEndian.PutUint16(Param_name_len[:], uint16(len(value.paramName)))
- pConn.sendBuff.Write(Param_name_len[:])
-
- pConn.sendBuff.Write(value.paramName[:])
- }
-
- Param_INOUT := [2]byte{0x1}
- pConn.sendBuff.Write(reverseBytes(Param_INOUT[:]))
-
- var Param_DType [2]byte
- binary.BigEndian.PutUint16(Param_DType[:], uint16(value.types))
- pConn.sendBuff.Write(Param_DType[:])
-
- Param_Data_Len := make([]byte, 4)
- binary.BigEndian.PutUint32(Param_Data_Len[:], uint32(value.valueLength))
- pConn.sendBuff.Write(Param_Data_Len[:])
-
-
- pConn.sendBuff.Write([]byte(value.value))
- }
- }
- return nil
- }
- func sockSendExecute(pConn *xuguConn) error {
-
- _, err := pConn.conn.Write(pConn.sendBuff.Bytes())
- if err != nil {
- return err
- }
- return nil
- }
- func xuguSockRecvMsg(pConn *xuguConn) (*allResult, error) {
- n, _ := pConn.conn.Read(pConn.readBuff.buf)
- pConn.readBuff.length += n
- rs, err := parseMsg(&pConn.readBuff, pConn)
- if err != nil {
- fmt.Println("这里接收到了err: ", err)
- return nil, err
- }
- pConn.readBuff.reset()
- return rs, nil
- }
|