123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410 |
- package xugu
- import (
- "database/sql/driver"
- "encoding/binary"
- "fmt"
- "strings"
- )
- type ParseParam interface {
- // 转换参数数据类型
- assertParamType(driver.Value, int) error
- // 解析参数数量
- assertParamCount(string) int
- // 解析参数绑定类型(按参数名称绑定类型和按参数位置绑定类型)
- assertBindType(string) int
- // 解析参数名称
- assertParamName(string) error
- }
- // 辅助结构体
- // type xuguValue struct {
- // // 布尔值,如果值为 true,表示当前字段的数据类型是大对象数据类型
- // islob bool
- // // 一个 *C.char 类型的指针,通常指向要绑定的参数值的地址
- // value []byte
- // plob []byte
- // // length 通常指定要绑定的参数数据的实际长度
- // length []int
- // // buff 通常指定要绑定的参数数据的内存缓冲区大小
- // buffSize int
- // // 在参数绑定时,指定表中字段的数据类型
- // types int
- // // 返回代码
- // rcode int
- // }
- type xuguParse struct {
- // bind_type 用于标识参数绑定的类型。
- // 参数绑定类型包括按参数名称绑定和按参数占位符绑定
- bind_type int
- // param_count 用于指定在 SQL 语句中需要绑定的参数数量
- param_count int
- // 当参数绑定类型为按参数名称绑定时,param_names 是参数名称的集合
- param_names []byte
- //Val []xuguValue
- // 当参数绑定类型为按参数占位符绑定时,position 标识参数的位置
- position int
- }
- func (Parse *xuguParse) assertParamCount(query string) int {
- fmt.Println("----assertParamCount func 判断参数个数")
- if Parse.bind_type == 0 {
- Parse.bind_type = Parse.assertBindType(query)
- }
- fmt.Println("**Parse.bind_type is", Parse.bind_type)
- switch Parse.bind_type {
- case BIND_PARAM_BY_POS:
- Parse.param_count = strings.Count(query, "?")
- case BIND_PARAM_BY_NAME:
- fmt.Println(">case BIND_PARAM_BY_NAME")
- Parse.param_count = 0
- pos := 0
- phead := -1
- for {
- pos = strings.IndexByte(query[phead+1:], ':')
- if pos == -1 {
- break
- }
- pos += phead + 1
- tmp := pos
- for tmp > phead {
- tmp--
- if query[tmp] == ' ' {
- continue
- }
- if query[tmp] == ',' || query[tmp] == '(' {
- Parse.param_count++
- }
- break
- }
- phead = pos
- }
- }
- fmt.Println("**Parse.param_count", Parse.param_count)
- return Parse.param_count
- }
- func (Parse *xuguParse) assertBindType(query string) int {
- fmt.Println("-----(param *xuguParse) assertBindType 判断绑定类型")
- Parse.bind_type = strings.IndexByte(query, '?')
- if Parse.bind_type != -1 {
- fmt.Println("** 是大对象语句")
- return BIND_PARAM_BY_POS
- }
- fmt.Println("** 不是大对象语句")
- return BIND_PARAM_BY_NAME
- }
- func (param *xuguParse) assertParamType(dV driver.Value, pos int) error {
- fmt.Println("-----(param *xuguParse) assertParamType 判断参数类型")
- // var dest xuguValue
- // switch dV.(type) {
- // case int64:
- // srcv, ok := dV.(int64)
- // if !ok {
- // news := errorNews("int64")
- // return errors.New(news)
- // }
- // S := strconv.FormatInt(srcv, 10)
- // dest.value = C.CString(S)
- // dest.length = C.int(strings.Count(S, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // case float32:
- // srcv, ok := dV.(float64)
- // if !ok {
- // news := errorNews("float32")
- // return errors.New(news)
- // }
- // S := strconv.FormatFloat(srcv, 'f', 6, 64)
- // dest.value = C.CString(S)
- // dest.length = C.int(strings.Count(S, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // case float64:
- // srcv, ok := dV.(float64)
- // if !ok {
- // news := errorNews("float64")
- // return errors.New(news)
- // }
- // S := strconv.FormatFloat(srcv, 'f', 15, 64)
- // dest.value = C.CString(S)
- // dest.length = C.int(strings.Count(S, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // case bool:
- // srcv, ok := dV.(bool)
- // if !ok {
- // news := errorNews("bool")
- // return errors.New(news)
- // }
- // S := strconv.FormatBool(srcv)
- // dest.value = C.CString(S)
- // dest.length = C.int(strings.Count(S, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // case string:
- // srcv, ok := dV.(string)
- // if !ok {
- // news := errorNews("string")
- // return errors.New(news)
- // }
- // dest.value = C.CString(srcv)
- // dest.length = C.int(strings.Count(srcv, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // if dest.length == 0 {
- // dest.length = 1
- // }
- // case time.Time:
- // srcv, ok := dV.(time.Time)
- // if !ok {
- // news := errorNews("time.Time")
- // return errors.New(news)
- // }
- // tm := fmt.Sprintf("%04d-%02d-%02d %02d:%02d:%02d",
- // srcv.Year(), int(srcv.Month()), srcv.Day(),
- // srcv.Hour(), srcv.Minute(), srcv.Second())
- // dest.value = C.CString(tm)
- // dest.length = C.int(strings.Count(tm, "") - 1)
- // dest.buff = dest.length + 1
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // case []byte:
- // re := cgo_xgc_new_lob(&dest.plob)
- // if re < 0 {
- // return errors.New("Cannot create new large object")
- // }
- // srcv, ok := dV.([]byte)
- // if !ok {
- // news := errorNews("[]byte")
- // return errors.New(news)
- // }
- // cgo_xgc_put_lob_data(
- // &dest.plob,
- // unsafe.Pointer((*C.char)(unsafe.Pointer(&srcv[0]))),
- // len(srcv))
- // cgo_xgc_put_lob_data(&dest.plob, nil, -1)
- // dest.value = nil
- // dest.length = C.int(8)
- // dest.buff = C.int(8)
- // dest.islob = true
- // dest.types = SQL_XG_C_BLOB
- // case nil:
- // dest.value = C.CString("xugusql")
- // dest.length = 0
- // dest.buff = C.int(strings.Count("xugusql", ""))
- // dest.islob = false
- // dest.types = SQL_XG_C_CHAR
- // default:
- // /* OTHER DATA TYPE */
- // return errors.New("Unknown data type")
- // }
- // param.position = pos
- // param.Val = append(param.Val, dest)
- return nil
- }
- func (param *xuguParse) assertParamName(query string) error {
- fmt.Println("----(param *xuguParse) assertParamName 判断参数名")
- if param.param_count <= 0 {
- param.assertParamCount(query)
- }
- pos := 0
- phead := -1
- for true {
- pos = strings.IndexByte(query[phead+1:], ':')
- if pos == -1 {
- break
- }
- pos += phead + 1
- tmp := pos
- for tmp > phead {
- tmp--
- if query[tmp] == ' ' {
- continue
- }
- // Parse parameter positions bound by parameter name
- if query[tmp] == ',' || query[tmp] == '(' {
- parg := pos
- for true {
- parg++
- if query[parg] == ',' || query[parg] == ')' || query[parg] == ' ' {
- param.param_names = append(param.param_names, query[pos+1:parg]...)
- break
- }
- }
- }
- break
- }
- phead = pos
- }
- return nil
- }
- type SelectResult struct {
- Field_Num uint32
- Fields []FieldDescri
- Values [][]ValueData //[字段][字段所有值]
- }
- type FieldDescri struct {
- FieldNameLen int
- FieldName string
- FieldType fieldType
- FieldPreciScale int
- FieldFlag int
- }
- type ValueData struct {
- Col_len uint32
- Col_Data []byte
- }
- func parseSelectResult(readBuf buffer) (*SelectResult, error) {
- data := &SelectResult{}
- char := readBuf.peekChar()
- fmt.Println("--=char: ", string(char))
- switch char {
- case 'A':
- readBuf.idx++
- //Field_Num
- Field_Num := binary.BigEndian.Uint32(readBuf.readNext(4))
- data.Field_Num = Field_Num
- fmt.Println("Field_Num: ", data.Field_Num)
- //获取字段信息
- for i := 0; i < int(Field_Num); i++ {
- field := FieldDescri{}
- //Field_Name_Len
- field.FieldNameLen = int(binary.BigEndian.Uint32(readBuf.readNext(4)))
- fmt.Println("FieldNameLen: ", field.FieldNameLen)
- //Field_Name:
- field.FieldName = string(readBuf.readNext(field.FieldNameLen))
- fmt.Println("FieldName: ", field.FieldName)
- //Field_DType:
- field.FieldType = fieldType(binary.BigEndian.Uint32(readBuf.readNext(4)))
- fmt.Println("FieldType: ", field.FieldType)
- //Field_Preci_Scale:
- field.FieldPreciScale = int(binary.BigEndian.Uint32(readBuf.readNext(4)))
- fmt.Println("FieldPreciScale: ", field.FieldPreciScale)
- //Field_Flag:
- field.FieldFlag = int(binary.BigEndian.Uint32(readBuf.readNext(4)))
- fmt.Println("FieldFlag: ", field.FieldFlag)
- data.Fields = append(data.Fields, field)
- }
- fmt.Println("data的字段总数: ", data.Field_Num)
- for i, v := range data.Fields {
- fmt.Println("data的内容: ", i, v.FieldName)
- }
- data.Values = make([][]ValueData, data.Field_Num)
- //获取字段的行值
- // 使用 Peek 方法检查下一个字节是否为'R'或'K'
- char := readBuf.peekChar()
- fmt.Println("是否为R: ", string(char))
- readBuf.idx++
- if char == 'K' {
- break
- } else if char == 'R' {
- colIdx := 0
- for {
- col := ValueData{}
- col.Col_len = binary.BigEndian.Uint32(readBuf.readNext(4))
- // fmt.Println("Col_len: ", col.Col_len)
- col.Col_Data = readBuf.readNext(int(col.Col_len))
- fmt.Println("Col_Data: ", string(col.Col_Data))
- data.Values[colIdx] = append(data.Values[colIdx], col)
- colIdx++
- char := readBuf.peekChar()
- if char == 'R' {
- fmt.Println("--=char3: ", string(readBuf.peekChar()))
- readBuf.idx++
- colIdx = 0
- continue
- } else if char == 'K' {
- fmt.Println("--=char4: ", string(readBuf.peekChar()))
- break
- }
- }
- } else {
- break
- }
- }
- fmt.Println("data的内容: ", data)
- for i := 0; i < int(data.Field_Num); i++ {
- for j := 0; j < len(data.Values[i]); j++ {
- fmt.Printf("data.Values[%d]: %v\n", i, (data.Values[i][j].Col_Data))
- }
- }
- return data, nil
- }
- func parseInsertResult(readBuf buffer) (*SelectResult, error) {
- }
|