xugu_parse.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510
  1. package xugu
  2. import (
  3. "database/sql/driver"
  4. "encoding/binary"
  5. "errors"
  6. "fmt"
  7. "strconv"
  8. "strings"
  9. "time"
  10. )
  11. type xuguValue struct {
  12. // 布尔值,如果值为 true,表示当前字段的数据类型是大对象数据类型
  13. islob bool
  14. //字段名
  15. paramName []byte
  16. //paramNameLength [2]byte
  17. // 字段的实际值
  18. value []byte
  19. // 值的长度
  20. valueLength int
  21. // 字段的类型
  22. types fieldType
  23. }
  24. // 判断参数个数
  25. func assertParamCount(query string) int {
  26. fmt.Println("----assertParamCount func 判断参数个数")
  27. paramCount := strings.Count(query, "?")
  28. fmt.Println("Parse.param_count = ", paramCount)
  29. return paramCount
  30. }
  31. func assertParamType(dV driver.Value, values *[]xuguValue) error {
  32. var dest xuguValue
  33. switch srcv := dV.(type) {
  34. case int64:
  35. S := strconv.FormatInt(srcv, 10)
  36. dest.value = []byte(S)
  37. dest.valueLength = strings.Count(S, "") - 1
  38. dest.islob = false
  39. dest.types = fieldType_I8
  40. case float32:
  41. S := strconv.FormatFloat(float64(srcv), 'f', 6, 64)
  42. dest.value = []byte(S)
  43. dest.valueLength = strings.Count(S, "") - 1
  44. dest.islob = false
  45. dest.types = fieldType_R4
  46. case float64:
  47. S := strconv.FormatFloat(srcv, 'f', 15, 64)
  48. dest.value = []byte(S)
  49. dest.valueLength = strings.Count(S, "") - 1
  50. dest.islob = false
  51. dest.types = fieldType_R8
  52. case bool:
  53. S := strconv.FormatBool(srcv)
  54. dest.value = []byte(S)
  55. dest.valueLength = strings.Count(S, "") - 1
  56. dest.islob = false
  57. dest.types = fieldType_BOOL
  58. case string:
  59. dest.value = []byte(srcv)
  60. dest.valueLength = strings.Count(srcv, "") - 1
  61. dest.islob = false
  62. dest.types = fieldType_CHAR
  63. case time.Time:
  64. tm := fmt.Sprintf("%04d-%02d-%02d %02d:%02d:%02d",
  65. srcv.Year(), int(srcv.Month()), srcv.Day(),
  66. srcv.Hour(), srcv.Minute(), srcv.Second())
  67. dest.value = []byte(tm)
  68. dest.valueLength = strings.Count(tm, "") - 1
  69. dest.islob = false
  70. dest.types = fieldType_TIME
  71. case []byte:
  72. dest.value = srcv
  73. dest.valueLength = len(srcv)
  74. dest.islob = true
  75. dest.types = fieldType_BLOB
  76. case nil:
  77. dest.value = []byte("xugusql")
  78. dest.valueLength = 0
  79. dest.islob = false
  80. dest.types = fieldType_CHAR
  81. default:
  82. return errors.New("unknown data type")
  83. }
  84. *values = append(*values, dest)
  85. return nil
  86. }
  87. func parseMsg(readBuf *buffer, pConn *xuguConn) (*allResult, error) {
  88. var err error
  89. aR := allResult{}
  90. for {
  91. char := readBuf.peekChar()
  92. fmt.Println("parseMsg 内的 peekChar: ", char, "-", string(char))
  93. switch char {
  94. case 'K':
  95. fmt.Println("消息类型为K")
  96. readBuf.reset()
  97. return &aR, nil
  98. case '$':
  99. readBuf.idx++
  100. fmt.Println("消息类型为$")
  101. if aR.f, err = parseFormArgDescri(readBuf); err != nil {
  102. return nil, err
  103. }
  104. aR.rt = formArgDescri
  105. return &aR, err
  106. case 'A':
  107. readBuf.idx++
  108. fmt.Println("消息类型为A")
  109. if aR.s, err = parseSelectResult(readBuf); err != nil {
  110. return nil, err
  111. }
  112. aR.rt = selectResult
  113. return &aR, err
  114. case 'I':
  115. readBuf.idx++
  116. fmt.Println("消息类型为I")
  117. if aR.i, err = parseInsertResult(readBuf); err != nil {
  118. return nil, err
  119. }
  120. aR.rt = insertResult
  121. return &aR, err
  122. case 'U':
  123. fmt.Println("消息类型为U")
  124. readBuf.idx++
  125. if aR.u, err = parseUpdateResult(readBuf); err != nil {
  126. return nil, err
  127. }
  128. aR.rt = updateResult
  129. return &aR, err
  130. case 'D':
  131. fmt.Println("消息类型为D")
  132. readBuf.idx++
  133. readBuf.idx++
  134. if aR.d, err = parseDeleteResult(readBuf); err != nil {
  135. return nil, err
  136. }
  137. aR.rt = deleteResult
  138. return &aR, err
  139. case 'E':
  140. fmt.Println("消息类型为E")
  141. readBuf.idx++
  142. if aR.e, err = parseErrInfo(readBuf); err != nil {
  143. return nil, err
  144. }
  145. pConn.errStr = aR.e.ErrStr
  146. aR.rt = errInfo
  147. return &aR, err
  148. case 'W':
  149. fmt.Println("消息类型为W")
  150. readBuf.idx++
  151. if aR.w, err = parseWarnInfo(readBuf); err != nil {
  152. return nil, err
  153. }
  154. aR.rt = warnInfo
  155. return &aR, err
  156. case 'M':
  157. fmt.Println("消息类型为M")
  158. readBuf.idx++
  159. if aR.m, err = parseMessage(readBuf); err != nil {
  160. return nil, err
  161. }
  162. aR.rt = message
  163. return &aR, err
  164. default:
  165. fmt.Println("消息类型为其他")
  166. return nil, errors.New("parseMsg: unknown message type")
  167. }
  168. }
  169. }
  170. func parseSelectResult(readBuf *buffer) (*SelectResult, error) {
  171. fmt.Println("调用 parseSelectResult")
  172. data := &SelectResult{}
  173. char := readBuf.peekChar()
  174. fmt.Println("--=char: ", string(char), char)
  175. //Field_Num
  176. Field_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  177. data.Field_Num = Field_Num
  178. data.rowIdx = 0
  179. fmt.Println("Field_Num: ", data.Field_Num)
  180. //获取字段信息
  181. for i := 0; i < int(Field_Num); i++ {
  182. field := FieldDescri{}
  183. //Field_Name_Len
  184. field.FieldNameLen = int(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
  185. fmt.Println("field.FieldNameLen: ", field.FieldNameLen)
  186. //Field_Name:
  187. field.FieldName = string(readBuf.readNext(field.FieldNameLen, false))
  188. fmt.Println("field.Field_Name: ", field.FieldName)
  189. //Field_DType:
  190. field.FieldType = fieldType(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
  191. fmt.Println("field.FieldType: ", field.FieldType)
  192. //Field_Preci_Scale:
  193. fieldPreciScale := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  194. if int32(fieldPreciScale) <= 0 {
  195. field.FieldPreciScale = fieldPreciScaleInfo{
  196. scale: 0,
  197. accuracy: 0,
  198. }
  199. } else {
  200. field.FieldPreciScale = fieldPreciScaleInfo{
  201. scale: uint16(fieldPreciScale >> 16),
  202. accuracy: uint16(fieldPreciScale & 0xFFFF),
  203. }
  204. }
  205. fmt.Println("field.FieldPreciScale: ", field.FieldPreciScale)
  206. //Field_Flag:
  207. field.FieldFlag = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  208. fmt.Println("field.FieldFlag: ", field.FieldFlag)
  209. data.Fields = append(data.Fields, field)
  210. }
  211. data.Values = make([][]FieldValue, data.Field_Num)
  212. //获取字段的行值,并判断类型
  213. // 使用 Peek 方法检查下一个字节是否为'R'或'K'
  214. fmt.Println("\n\n=========开始获取行数据=================================")
  215. defer func() {
  216. fmt.Println("\n\n=========获取行数据结束=================================")
  217. }()
  218. char = readBuf.peekChar()
  219. fmt.Println(" --char: ", string(char))
  220. readBuf.idx++
  221. if char == 'K' {
  222. return data, nil
  223. } else if char == 'R' {
  224. colIdx := 0
  225. //typeIdx := 0
  226. fmt.Println("开始循环 ")
  227. for {
  228. col := FieldValue{}
  229. //获取数据的大小
  230. col.Col_len = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  231. fmt.Println("数据大小为: ", col.Col_len)
  232. col.Col_Data = readBuf.readNext(int(col.Col_len), false)
  233. fmt.Println("fieldTypeVarChar data: ", col.Col_Data, string(col.Col_Data))
  234. data.Values[colIdx] = append(data.Values[colIdx], col)
  235. colIdx++
  236. fmt.Println("从查询返回 解析出的值 :", col.Col_Data, string(col.Col_Data))
  237. char := readBuf.peekChar()
  238. //既不是R 也不是K 代表该行还有其他字段内容没有读取完成
  239. if char == 'R' {
  240. readBuf.idx++
  241. colIdx = 0
  242. continue
  243. } else if char == 'K' {
  244. return data, nil
  245. //break
  246. }
  247. } //for end
  248. } else if char == '$' {
  249. fmt.Println(" 查询返回 $ ")
  250. fad, err := parseFormArgDescri(readBuf)
  251. if err != nil {
  252. return nil, err
  253. }
  254. char := readBuf.peekChar()
  255. //既不是R 也不是K 代表该行还有其他字段内容没有读取完成
  256. if char == 'K' {
  257. data.fad = fad
  258. return data, nil
  259. //break
  260. }
  261. return nil, errors.New("select to $ 解析失败")
  262. } else {
  263. return nil, errors.New("解析失败")
  264. }
  265. }
  266. // func parseSelectResult2(readBuf *buffer) (*SelectResult, error) {
  267. // data := &SelectResult{}
  268. // char := readBuf.peekChar()
  269. // if char == 'K' {
  270. // return nil, errors.New("parseSelectResult error char is K")
  271. // }
  272. // fmt.Println("--=char: ", string(char))
  273. // //Field_Num
  274. // Field_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  275. // data.Field_Num = Field_Num
  276. // //data.rowIdx = 0
  277. // fmt.Println("Field_Num: ", data.Field_Num)
  278. // //获取字段信息
  279. // for i := 0; i < int(Field_Num); i++ {
  280. // field := FieldDescri{}
  281. // //Field_Name_Len
  282. // field.FieldNameLen = int(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
  283. // fmt.Println("field.FieldNameLen: ", field.FieldNameLen)
  284. // //Field_Name:
  285. // field.FieldName = string(readBuf.readNext(field.FieldNameLen, false))
  286. // fmt.Println("field.Field_Name: ", field.FieldName)
  287. // //Field_DType:
  288. // field.FieldType = fieldType(binary.LittleEndian.Uint32(readBuf.readNext(4, true)))
  289. // fmt.Println("field.FieldType: ", field.FieldType)
  290. // //Field_Preci_Scale:
  291. // fieldPreciScale := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  292. // if int32(fieldPreciScale) <= 0 {
  293. // field.FieldPreciScale = fieldPreciScaleInfo{
  294. // scale: 0,
  295. // accuracy: 0,
  296. // }
  297. // } else {
  298. // field.FieldPreciScale = fieldPreciScaleInfo{
  299. // scale: uint16(fieldPreciScale >> 16),
  300. // accuracy: uint16(fieldPreciScale & 0xFFFF),
  301. // }
  302. // }
  303. // fmt.Println("field.FieldPreciScale: ", field.FieldPreciScale)
  304. // //Field_Flag:
  305. // field.FieldFlag = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  306. // fmt.Println("field.FieldFlag: ", field.FieldFlag)
  307. // data.Fields = append(data.Fields, field)
  308. // }
  309. // //获取字段的行值,并判断类型
  310. // // 使用 Peek 方法检查下一个字节是否为'R'或'K'
  311. // fmt.Println("\n\n=========开始获取行数据=================================")
  312. // defer func() {
  313. // fmt.Println("\n\n=========获取行数据结束=================================")
  314. // }()
  315. // //data.Values = make([][]FieldValue, 0)
  316. // data.Values = [][]FieldValue
  317. // colIdx := 0
  318. // char = readBuf.peekChar()
  319. // fmt.Println(" --char: ", string(char))
  320. // readBuf.idx++
  321. // if char == 'K' {
  322. // return data, nil
  323. // } else if char == 'R' {
  324. // for {
  325. // col := FieldValue{}
  326. // //获取数据的大小
  327. // col.Col_len = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  328. // fmt.Println("数据大小为: ", col.Col_len)
  329. // col.Col_Data = readBuf.readNext(int(col.Col_len), false)
  330. // fmt.Println("fieldTypeVarChar data: ", col.Col_Data, string(col.Col_Data))
  331. // data.Values[colIdx] = append(data.Values[colIdx], col)
  332. // colIdx++
  333. // fmt.Println("从查询返回 解析出的值 :", col.Col_Data, string(col.Col_Data))
  334. // char := readBuf.peekChar()
  335. // //next for
  336. // if char == 'R' {
  337. // readBuf.idx++
  338. // continue
  339. // //end parse
  340. // } else if char == 'K' {
  341. // return data, nil
  342. // }
  343. // //swich end
  344. // fmt.Println("循环结束")
  345. // fmt.Printf("解析请求 data:%#v\n", data.Fields)
  346. // return data, nil
  347. // } //for end
  348. // } else {
  349. // return data, errors.New("select parse error")
  350. // }
  351. // }
  352. func parseInsertResult(readBuf *buffer) (*InsertResult, error) {
  353. //Rowid_Len
  354. Rowid_Len := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  355. //Rowid_Data
  356. encoded := readBuf.readNext(int(Rowid_Len), false)
  357. //检测是否结束
  358. char := readBuf.peekChar()
  359. if char == 'K' {
  360. return &InsertResult{
  361. RowidLen: Rowid_Len,
  362. RowidData: encoded,
  363. }, nil
  364. }
  365. return nil, errors.New("parseInsertResult error")
  366. }
  367. func parseUpdateResult(readBuf *buffer) (*UpdateResult, error) {
  368. updateNum := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  369. return &UpdateResult{UpdateNum: updateNum}, nil
  370. }
  371. func parseDeleteResult(readBuf *buffer) (*DeleteResult, error) {
  372. deleteNum := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  373. return &DeleteResult{DeleteNum: deleteNum}, nil
  374. }
  375. func parseProcRet(readBuf *buffer) (*ProcRet, error) {
  376. retDType := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  377. retDataLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  378. retData := readBuf.readNext(int(retDataLen), false)
  379. return &ProcRet{RetDType: retDType, RetDataLen: retDataLen, RetData: retData}, nil
  380. }
  381. func parseOutParamRet(readBuf *buffer) (*OutParamRet, error) {
  382. outParamNo := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  383. outParamDType := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  384. outParamLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  385. outParamData := readBuf.readNext(int(outParamLen), false)
  386. return &OutParamRet{
  387. OutParamNo: outParamNo,
  388. OutParamDType: outParamDType,
  389. OutParamLen: outParamLen,
  390. OutParamData: outParamData,
  391. }, nil
  392. }
  393. func parseErrInfo(readBuf *buffer) (*ErrInfo, error) {
  394. errStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  395. errStr := readBuf.readNext(int(errStrLen), false)
  396. return &ErrInfo{ErrStrLen: errStrLen, ErrStr: errStr}, nil
  397. }
  398. func parseWarnInfo(readBuf *buffer) (*WarnInfo, error) {
  399. warnStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  400. warnStr := readBuf.readNext(int(warnStrLen), false)
  401. return &WarnInfo{WarnStrLen: warnStrLen, WarnStr: warnStr}, nil
  402. }
  403. func parseMessage(readBuf *buffer) (*Message, error) {
  404. msgStrLen := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  405. msgStr := readBuf.readNext(int(msgStrLen), false)
  406. return &Message{MsgStrLen: msgStrLen, MsgStr: msgStr}, nil
  407. }
  408. func parseFormArgDescri(readBuf *buffer) (*FormArgDescri, error) {
  409. // FormArgDescri: '$' Arg_Num { Arg_Name_Len Arg_Name Arg_No Arg_DType Arg_Preci_Scale }+
  410. Arg_Num := binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  411. formArgDescri := &FormArgDescri{ArgNum: Arg_Num}
  412. fmt.Println("-- parseFormArgDescri Arg_Num:", Arg_Num)
  413. for i := 0; i < int(Arg_Num); i++ {
  414. arg := ArgDescri{}
  415. //Arg_Name_Len
  416. arg.ArgNameLen = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  417. //Arg_Name
  418. arg.ArgName = readBuf.readNext(int(arg.ArgNameLen), false)
  419. //Arg_No
  420. arg.ArgNo = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  421. //Argg_DType
  422. arg.ArgDType = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  423. //Arg_Preci_Scale
  424. arg.ArgPreciScale = binary.LittleEndian.Uint32(readBuf.readNext(4, true))
  425. formArgDescri.Args = append(formArgDescri.Args, arg)
  426. }
  427. fmt.Printf("formArgDescri %#v \n", formArgDescri)
  428. return formArgDescri, nil
  429. }