xugu_rows.go 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. package xugu
  2. import (
  3. "database/sql/driver"
  4. "encoding/binary"
  5. "fmt"
  6. "io"
  7. "math"
  8. "reflect"
  9. "time"
  10. )
  11. type xuguRows struct {
  12. rows_conn *xuguConn
  13. results *SelectResult
  14. colIdx int
  15. prepared bool
  16. }
  17. func (row *xuguRows) Next(dest []driver.Value) error {
  18. gt("Next()")
  19. defer gt("Next() end")
  20. if row.results.rowIdx >= len(row.results.Values[0]) {
  21. //return errors.New("The result set has been released")
  22. return io.EOF
  23. }
  24. // fmt.Println(" -- int(row.results.Field_Num)", int(row.results.Field_Num))
  25. for j := 0; j < int(row.results.Field_Num); j++ {
  26. coluType := row.results.Fields[j].FieldType
  27. //fmt.Println(" --coluType: ", coluType, row.results.Fields[j].typeDatabaseName())
  28. switch coluType {
  29. case fieldType_BINARY,
  30. fieldType_CLOB,
  31. fieldType_BLOB:
  32. dest[j] = row.results.Values[j][row.results.rowIdx].Col_Data
  33. case fieldType_INTERVAL_Y, fieldType_INTERVAL_M, fieldType_INTERVAL_D,
  34. fieldType_INTERVAL_H, fieldType_INTERVAL_S,
  35. fieldType_INTERVAL_MI:
  36. dest[j] = binary.BigEndian.Uint32(row.results.Values[j][row.results.rowIdx].Col_Data)
  37. case fieldType_TIME,
  38. fieldType_TIME_TZ, fieldType_DATE:
  39. tv, _ := time.Parse("15:04:05", string(row.results.Values[j][row.results.rowIdx].Col_Data))
  40. dest[j] = tv
  41. case fieldType_DATETIME,
  42. fieldType_DATETIME_TZ:
  43. if len(row.results.Values[j][row.results.rowIdx].Col_Data) == 0 {
  44. dest[j] = nil
  45. } else {
  46. //timeTmp := binary.LittleEndian.Uint64(reverseBytes(row.results.Values[j][row.results.rowIdx].Col_Data)
  47. timeTmp := binary.BigEndian.Uint64(row.results.Values[j][row.results.rowIdx].Col_Data)
  48. //tv, _ := time.Parse("2006-01-02 15:04:05", string(reverseBytes(row.results.Values[j][row.results.rowIdx].Col_Data)))
  49. tv := time.Unix(0, int64(timeTmp)*1000000)
  50. utcTime := tv.UTC()
  51. dest[j] = utcTime
  52. }
  53. case fieldType_R4:
  54. dest[j] = math.Float32frombits(binary.BigEndian.Uint32(row.results.Values[j][row.results.rowIdx].Col_Data))
  55. fmt.Println("float32 ", dest[j])
  56. case fieldType_R8:
  57. dest[j] = math.Float64frombits(binary.BigEndian.Uint64(row.results.Values[j][row.results.rowIdx].Col_Data))
  58. fmt.Println(" dest[j] r8", dest[j])
  59. case fieldType_NUM:
  60. dest[j] = row.results.Values[j][row.results.rowIdx].Col_Data
  61. case fieldType_I1:
  62. dest[j] = int8(row.results.Values[j][row.results.rowIdx].Col_Data[0])
  63. case fieldType_I2:
  64. dest[j] = int16(binary.BigEndian.Uint16(row.results.Values[j][row.results.rowIdx].Col_Data))
  65. case fieldType_I4:
  66. dest[j] = binary.BigEndian.Uint32(row.results.Values[j][row.results.rowIdx].Col_Data)
  67. case fieldType_I8:
  68. dest[j] = int64(binary.BigEndian.Uint64(row.results.Values[j][row.results.rowIdx].Col_Data))
  69. //dest[j] = row.results.Values[j][row.results.rowIdx].Col_Data
  70. case fieldType_CHAR, fieldType_NCHAR:
  71. dest[j] = string(row.results.Values[j][row.results.rowIdx].Col_Data)
  72. default:
  73. //填入一行的数据
  74. //TODO这里长度改为一行长度
  75. dest[j] = make([]byte, len(row.results.Values))
  76. // Values[字段][0]
  77. dest[j] = row.results.Values[j][row.results.rowIdx].Col_Data
  78. }
  79. }
  80. row.results.rowIdx++
  81. return nil
  82. }
  83. // Columns返回列的名字集,它的个数是从slice的长度中推断出来的。
  84. // 如果不知道特定的列名,应该为该条目返回一个空的字符串
  85. func (row *xuguRows) Columns() []string {
  86. gt("Columns()")
  87. defer gt("Columns() end")
  88. var columns []string
  89. for _, v := range row.results.Fields {
  90. columns = append(columns, v.FieldName)
  91. }
  92. return columns
  93. }
  94. func (row *xuguRows) ColumnTypeScanType(index int) reflect.Type {
  95. fmt.Println(">>>>>ColumnTypeScanType ")
  96. return row.results.Fields[index].scanType()
  97. }
  98. func (self *xuguRows) Close() error {
  99. fmt.Println(">>>>>((self *xuguRows) Close() ")
  100. return nil
  101. }