123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510 |
- package xugu
- import (
- "database/sql/driver"
- "encoding/binary"
- "errors"
- "fmt"
- "strconv"
- "strings"
- "time"
- )
- type xuguValue struct {
- // 布尔值,如果值为 true,表示当前字段的数据类型是大对象数据类型
- islob bool
- //字段名
- paramName []byte
- //paramNameLength [2]byte
- // 字段的实际值
- value []byte
- // 值的长度
- valueLength int
- // 字段的类型
- types fieldType
- }
- // 判断参数个数
- func assertParamCount(query string) int {
- fmt.Println("----assertParamCount func 判断参数个数")
- paramCount := strings.Count(query, "?")
- fmt.Println("Parse.param_count = ", paramCount)
- return paramCount
- }
- func assertParamType(dV driver.Value, values *[]xuguValue) error {
- var dest xuguValue
- switch srcv := dV.(type) {
- case int64:
- S := strconv.FormatInt(srcv, 10)
- dest.value = []byte(S)
- dest.valueLength = strings.Count(S, "") - 1
- dest.islob = false
- dest.types = fieldType_I8
- case float32:
- S := strconv.FormatFloat(float64(srcv), 'f', 6, 64)
- dest.value = []byte(S)
- dest.valueLength = strings.Count(S, "") - 1
- dest.islob = false
- dest.types = fieldType_R4
- case float64:
- S := strconv.FormatFloat(srcv, 'f', 15, 64)
- dest.value = []byte(S)
- dest.valueLength = strings.Count(S, "") - 1
- dest.islob = false
- dest.types = fieldType_R8
- case bool:
- S := strconv.FormatBool(srcv)
- dest.value = []byte(S)
- dest.valueLength = strings.Count(S, "") - 1
- dest.islob = false
- dest.types = fieldType_BOOL
- case string:
- dest.value = []byte(srcv)
- dest.valueLength = strings.Count(srcv, "") - 1
- dest.islob = false
- dest.types = fieldType_CHAR
- case time.Time:
- tm := fmt.Sprintf("%04d-%02d-%02d %02d:%02d:%02d",
- srcv.Year(), int(srcv.Month()), srcv.Day(),
- srcv.Hour(), srcv.Minute(), srcv.Second())
- dest.value = []byte(tm)
- dest.valueLength = strings.Count(tm, "") - 1
- dest.islob = false
- dest.types = fieldType_TIME
- case []byte:
- dest.value = srcv
- dest.valueLength = len(srcv)
- dest.islob = true
- dest.types = fieldType_BLOB
- case nil:
- dest.value = []byte("xugusql")
- dest.valueLength = 0
- dest.islob = false
- dest.types = fieldType_CHAR
- default:
- return errors.New("unknown data type")
- }
- *values = append(*values, dest)
- return nil
- }
- func parseMsg(readBuf *buffer, pConn *xuguConn) (*allResult, error) {
- var err error
- aR := allResult{}
- for {
- char := readBuf.peekChar()
- fmt.Println("parseMsg 内的 peekChar: ", char, "-", string(char))
- switch char {
- case 'K':
- fmt.Println("消息类型为K")
- readBuf.reset()
- return &aR, nil
- case '$':
- readBuf.idx++
- fmt.Println("消息类型为$")
- if aR.f, err = parseFormArgDescri(readBuf); err != nil {
- return nil, err
- }
- aR.rt = formArgDescri
- return &aR, err
- case 'A':
- readBuf.idx++
- fmt.Println("消息类型为A")
- if aR.s, err = parseSelectResult(readBuf); err != nil {
- return nil, err
- }
- aR.rt = selectResult
- return &aR, err
- case 'I':
- readBuf.idx++
- fmt.Println("消息类型为I")
- if aR.i, err = parseInsertResult(readBuf); err != nil {
- return nil, err
- }
- aR.rt = insertResult
- return &aR, err
- case 'U':
- fmt.Println("消息类型为U")
- readBuf.idx++
- if aR.u, err = parseUpdateResult(readBuf); err != nil {
- return nil, err
- }
- aR.rt = updateResult
- return &aR, err
- case 'D':
- fmt.Println("消息类型为D")
- readBuf.idx++
- readBuf.idx++
- if aR.d, err = parseDeleteResult(readBuf); err != nil {
- return nil, err
- }
- aR.rt = deleteResult
- return &aR, err
- case 'E':
- fmt.Println("消息类型为E")
- readBuf.idx++
- if aR.e, err = parseErrInfo(readBuf); err != nil {
- return nil, err
- }
- pConn.errStr = aR.e.ErrStr
- aR.rt = errInfo
- return &aR, err
- case 'W':
- fmt.Println("消息类型为W")
- readBuf.idx++
- if aR.w, err = parseWarnInfo(readBuf); err != nil {
- return nil, err
- }
- aR.rt = warnInfo
- return &aR, err
- case 'M':
- fmt.Println("消息类型为M")
- readBuf.idx++
- if aR.m, err = parseMessage(readBuf); err != nil {
- return nil, err
- }
- aR.rt = message
- return &aR, err
- default:
- fmt.Println("消息类型为其他")
- return nil, errors.New("parseMsg: unknown message type")
- }
- }
- }
- func parseSelectResult(readBuf *buffer) (*SelectResult, error) {
- fmt.Println("调用 parseSelectResult")
- data := &SelectResult{}
- char := readBuf.peekChar()
- fmt.Println("--=char: ", string(char), char)
- //Field_Num
- Field_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- data.Field_Num = Field_Num
- data.rowIdx = 0
- fmt.Println("Field_Num: ", data.Field_Num)
- //获取字段信息
- for i := 0; i < int(Field_Num); i++ {
- field := FieldDescri{}
- //Field_Name_Len
- field.FieldNameLen = int(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
- fmt.Println("field.FieldNameLen: ", field.FieldNameLen)
- //Field_Name:
- field.FieldName = string(readBuf.readNext(field.FieldNameLen, false))
- fmt.Println("field.Field_Name: ", field.FieldName)
- //Field_DType:
- field.FieldType = fieldType(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
- fmt.Println("field.FieldType: ", field.FieldType)
- //Field_Preci_Scale:
- fieldPreciScale := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- if int32(fieldPreciScale) <= 0 {
- field.FieldPreciScale = fieldPreciScaleInfo{
- scale: 0,
- accuracy: 0,
- }
- } else {
- field.FieldPreciScale = fieldPreciScaleInfo{
- scale: uint16(fieldPreciScale >> 16),
- accuracy: uint16(fieldPreciScale & 0xFFFF),
- }
- }
- fmt.Println("field.FieldPreciScale: ", field.FieldPreciScale)
- //Field_Flag:
- field.FieldFlag = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- fmt.Println("field.FieldFlag: ", field.FieldFlag)
- data.Fields = append(data.Fields, field)
- }
- data.Values = make([][]FieldValue, data.Field_Num)
- //获取字段的行值,并判断类型
- // 使用 Peek 方法检查下一个字节是否为'R'或'K'
- fmt.Println("\n\n=========开始获取行数据=================================")
- defer func() {
- fmt.Println("\n\n=========获取行数据结束=================================")
- }()
- char = readBuf.peekChar()
- fmt.Println(" --char: ", string(char))
- readBuf.idx++
- if char == 'K' {
- return data, nil
- } else if char == 'R' {
- colIdx := 0
- //typeIdx := 0
- fmt.Println("开始循环 ")
- for {
- col := FieldValue{}
- //获取数据的大小
- col.Col_len = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- fmt.Println("数据大小为: ", col.Col_len)
- col.Col_Data = readBuf.readNext(int(col.Col_len), false)
- fmt.Println("fieldTypeVarChar data: ", col.Col_Data, string(col.Col_Data))
- data.Values[colIdx] = append(data.Values[colIdx], col)
- colIdx++
- fmt.Println("从查询返回 解析出的值 :", col.Col_Data, string(col.Col_Data))
- char := readBuf.peekChar()
- //既不是R 也不是K 代表该行还有其他字段内容没有读取完成
- if char == 'R' {
- readBuf.idx++
- colIdx = 0
- continue
- } else if char == 'K' {
- return data, nil
- //break
- }
- } //for end
- } else if char == '$' {
- fmt.Println(" 查询返回 $ ")
- fad, err := parseFormArgDescri(readBuf)
- if err != nil {
- return nil, err
- }
- char := readBuf.peekChar()
- //既不是R 也不是K 代表该行还有其他字段内容没有读取完成
- if char == 'K' {
- data.fad = fad
- return data, nil
- //break
- }
- return nil, errors.New("select to $ 解析失败")
- } else {
- return nil, errors.New("解析失败")
- }
- }
- // func parseSelectResult2(readBuf *buffer) (*SelectResult, error) {
- // data := &SelectResult{}
- // char := readBuf.peekChar()
- // if char == 'K' {
- // return nil, errors.New("parseSelectResult error char is K")
- // }
- // fmt.Println("--=char: ", string(char))
- // //Field_Num
- // Field_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- // data.Field_Num = Field_Num
- // //data.rowIdx = 0
- // fmt.Println("Field_Num: ", data.Field_Num)
- // //获取字段信息
- // for i := 0; i < int(Field_Num); i++ {
- // field := FieldDescri{}
- // //Field_Name_Len
- // field.FieldNameLen = int(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
- // fmt.Println("field.FieldNameLen: ", field.FieldNameLen)
- // //Field_Name:
- // field.FieldName = string(readBuf.readNext(field.FieldNameLen, false))
- // fmt.Println("field.Field_Name: ", field.FieldName)
- // //Field_DType:
- // field.FieldType = fieldType(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
- // fmt.Println("field.FieldType: ", field.FieldType)
- // //Field_Preci_Scale:
- // fieldPreciScale := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- // if int32(fieldPreciScale) <= 0 {
- // field.FieldPreciScale = fieldPreciScaleInfo{
- // scale: 0,
- // accuracy: 0,
- // }
- // } else {
- // field.FieldPreciScale = fieldPreciScaleInfo{
- // scale: uint16(fieldPreciScale >> 16),
- // accuracy: uint16(fieldPreciScale & 0xFFFF),
- // }
- // }
- // fmt.Println("field.FieldPreciScale: ", field.FieldPreciScale)
- // //Field_Flag:
- // field.FieldFlag = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- // fmt.Println("field.FieldFlag: ", field.FieldFlag)
- // data.Fields = append(data.Fields, field)
- // }
- // //获取字段的行值,并判断类型
- // // 使用 Peek 方法检查下一个字节是否为'R'或'K'
- // fmt.Println("\n\n=========开始获取行数据=================================")
- // defer func() {
- // fmt.Println("\n\n=========获取行数据结束=================================")
- // }()
- // //data.Values = make([][]FieldValue, 0)
- // data.Values = [][]FieldValue
- // colIdx := 0
- // char = readBuf.peekChar()
- // fmt.Println(" --char: ", string(char))
- // readBuf.idx++
- // if char == 'K' {
- // return data, nil
- // } else if char == 'R' {
- // for {
- // col := FieldValue{}
- // //获取数据的大小
- // col.Col_len = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- // fmt.Println("数据大小为: ", col.Col_len)
- // col.Col_Data = readBuf.readNext(int(col.Col_len), false)
- // fmt.Println("fieldTypeVarChar data: ", col.Col_Data, string(col.Col_Data))
- // data.Values[colIdx] = append(data.Values[colIdx], col)
- // colIdx++
- // fmt.Println("从查询返回 解析出的值 :", col.Col_Data, string(col.Col_Data))
- // char := readBuf.peekChar()
- // //next for
- // if char == 'R' {
- // readBuf.idx++
- // continue
- // //end parse
- // } else if char == 'K' {
- // return data, nil
- // }
- // //swich end
- // fmt.Println("循环结束")
- // fmt.Printf("解析请求 data:%#v\n", data.Fields)
- // return data, nil
- // } //for end
- // } else {
- // return data, errors.New("select parse error")
- // }
- // }
- func parseInsertResult(readBuf *buffer) (*InsertResult, error) {
- //Rowid_Len
- Rowid_Len := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- //Rowid_Data
- encoded := readBuf.readNext(int(Rowid_Len), false)
- //检测是否结束
- char := readBuf.peekChar()
- if char == 'K' {
- return &InsertResult{
- RowidLen: Rowid_Len,
- RowidData: encoded,
- }, nil
- }
- return nil, errors.New("parseInsertResult error")
- }
- func parseUpdateResult(readBuf *buffer) (*UpdateResult, error) {
- updateNum := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- return &UpdateResult{UpdateNum: updateNum}, nil
- }
- func parseDeleteResult(readBuf *buffer) (*DeleteResult, error) {
- deleteNum := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- return &DeleteResult{DeleteNum: deleteNum}, nil
- }
- func parseProcRet(readBuf *buffer) (*ProcRet, error) {
- retDType := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- retDataLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- retData := readBuf.readNext(int(retDataLen), false)
- return &ProcRet{RetDType: retDType, RetDataLen: retDataLen, RetData: retData}, nil
- }
- func parseOutParamRet(readBuf *buffer) (*OutParamRet, error) {
- outParamNo := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- outParamDType := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- outParamLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- outParamData := readBuf.readNext(int(outParamLen), false)
- return &OutParamRet{
- OutParamNo: outParamNo,
- OutParamDType: outParamDType,
- OutParamLen: outParamLen,
- OutParamData: outParamData,
- }, nil
- }
- func parseErrInfo(readBuf *buffer) (*ErrInfo, error) {
- errStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- errStr := readBuf.readNext(int(errStrLen), false)
- return &ErrInfo{ErrStrLen: errStrLen, ErrStr: errStr}, nil
- }
- func parseWarnInfo(readBuf *buffer) (*WarnInfo, error) {
- warnStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- warnStr := readBuf.readNext(int(warnStrLen), false)
- return &WarnInfo{WarnStrLen: warnStrLen, WarnStr: warnStr}, nil
- }
- func parseMessage(readBuf *buffer) (*Message, error) {
- msgStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- msgStr := readBuf.readNext(int(msgStrLen), false)
- return &Message{MsgStrLen: msgStrLen, MsgStr: msgStr}, nil
- }
- func parseFormArgDescri(readBuf *buffer) (*FormArgDescri, error) {
- // FormArgDescri: '$' Arg_Num { Arg_Name_Len Arg_Name Arg_No Arg_DType Arg_Preci_Scale }+
- Arg_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- formArgDescri := &FormArgDescri{ArgNum: Arg_Num}
- fmt.Println("-- parseFormArgDescri Arg_Num:", Arg_Num)
- for i := 0; i < int(Arg_Num); i++ {
- arg := ArgDescri{}
- //Arg_Name_Len
- arg.ArgNameLen = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- //Arg_Name
- arg.ArgName = readBuf.readNext(int(arg.ArgNameLen), false)
- //Arg_No
- arg.ArgNo = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- //Argg_DType
- arg.ArgDType = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- //Arg_Preci_Scale
- arg.ArgPreciScale = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
- formArgDescri.Args = append(formArgDescri.Args, arg)
- }
- fmt.Printf("formArgDescri %#v \n", formArgDescri)
- return formArgDescri, nil
- }
|