fix: query response data should be [][]string (#39)

Author: everpcpc

con_batch.go

@@ -46,8 +46,8 @@ func (dc *DatabendConn) prepareBatch(ctx context.Context, query string) (ldriver
 	var columnNames, columnTypes []string
 	for i := range r.Data {
 		if len(r.Data[i]) > 1 {
-			columnNames = append(columnNames, fmt.Sprintf("%s", r.Data[i][0]))
-			columnTypes = append(columnTypes, fmt.Sprintf("%s", r.Data[i][1]))
+			columnNames = append(columnNames, r.Data[i][0])
+			columnTypes = append(columnTypes, r.Data[i][1])
 		}
 	}
 

query.go

@@ -25,15 +25,15 @@ type DataField struct {
 }
 
 type QueryResponse struct {
-	Data     [][]interface{} `json:"data"`
-	Error    *QueryError     `json:"error"`
-	FinalURI string          `json:"final_uri"`
-	Id       string          `json:"id"`
-	NextURI  string          `json:"next_uri"`
-	Schema   []DataField     `json:"schema"`
-	State    string          `json:"state"`
-	Stats    QueryStats      `json:"stats"`
-	StatsURI string          `json:"stats_uri"`
+	Data     [][]string  `json:"data"`
+	Error    *QueryError `json:"error"`
+	FinalURI string      `json:"final_uri"`
+	Id       string      `json:"id"`
+	NextURI  string      `json:"next_uri"`
+	Schema   []DataField `json:"schema"`
+	State    string      `json:"state"`
+	Stats    QueryStats  `json:"stats"`
+	StatsURI string      `json:"stats_uri"`
 }
 
 type QueryStats struct {

restful.go

@@ -249,9 +249,10 @@ func (c *APIClient) UploadToStageByPresignURL(stage, fileName string) error {
 	if len(resp.Data) < 1 || len(resp.Data[0]) < 2 {
 		return fmt.Errorf("generate presign url failed")
 	}
-	headers, ok := resp.Data[0][1].(map[string]interface{})
-	if !ok {
-		return fmt.Errorf("no host for presign url")
+	headers := make(map[string]string)
+	err = json.Unmarshal([]byte(resp.Data[0][1]), &headers)
+	if err != nil {
+		return errors.Wrap(err, "failed to unmarshal presign url headers")
 	}
 
 	presignURL := fmt.Sprintf("%v", resp.Data[0][2])

rows.go

@@ -105,7 +105,7 @@ func (r *nextRows) Next(dest []driver.Value) error {
 	r.respData.Data = r.respData.Data[1:]
 
 	for j := range lineData {
-		reader := strings.NewReader(fmt.Sprintf("%v", lineData[j]))
+		reader := strings.NewReader(lineData[j])
 		v, err := r.parsers[j].Parse(reader)
 		if err != nil {
 			r.dc.log("parse error ", err)

rows_test.go

@@ -9,7 +9,7 @@ import (
 
 func TestTextRows(t *testing.T) {
 	rows, err := newNextRows(&DatabendConn{}, &QueryResponse{
-		Data: [][]interface{}{{1, 2, "3"}, {3, 2, "1"}},
+		Data: [][]string{{"1", "2", "3"}, {"3", "2", "1"}},
 		Schema: []DataField{
 			{Name: "age", Type: "Int32"},
 			{Name: "height", Type: "Int64"},