將「SELECT *」列讀入[]中的字符串

Question

我想編寫一個Go程序，使用SELECT *將數據庫表中的行轉儲爲csv文件。

轉到提供了優良sql和csv的API，但是csv期望字符串數組，並根據它們的類型在Rows的Scan方法「填充」字段。 因爲我以前不知道表格，所以我不知道有多少列以及它們的類型是什麼。

這是我在Go的第一個程序，所以我掙扎了一下。

如何最好地將Rows實例中的列讀入[]string - 這是「正確」的方式嗎？

謝謝！

UPDATE

我仍與參數掙扎。這是我的代碼，現在我正在使用panic而不是返回error，但我稍後會改變它。在我的測試中，我傳遞了查詢結果和os.Stdout。

func dumpTable(rows *sql.Rows, out io.Writer) error { 
    colNames, err := rows.Columns() 
    if err != nil { 
     panic(err) 
    } 
    if rows.Next() { 
     writer := csv.NewWriter(out) 
     writer.Comma = '\t' 
     cols := make([]string, len(colNames)) 
     processRow := func() { 
      err := rows.Scan(cols...) 
      if err != nil { 
       panic(err) 
      } 
      writer.Write(cols) 
     } 
     processRow() 
     for rows.Next() { 
      processRow() 
     } 
     writer.Flush() 
    } 
    return nil 
} 

對於這一點，我得到cannot use cols (type []string) as type []interface {} in function argument（在writer.Write(cols)線。

然後我測試

readCols := make([]interface{}, len(colNames)) 
    writeCols := make([]string, len(colNames)) 
    processRow := func() { 
     err := rows.Scan(readCols...) 
     if err != nil { 
      panic(err) 
     } 
     // ... CONVERSION? 
     writer.Write(writeCols) 
    } 

這導致panic: sql: Scan error on column index 0: destination not a pointer。

UPDATE 2

我獨立抵達ANisus'解決方案。這是我現在使用的代碼。

func dumpTable(rows *sql.Rows, out io.Writer) error { 
    colNames, err := rows.Columns() 
    if err != nil { 
     panic(err) 
    } 
    writer := csv.NewWriter(out) 
    writer.Comma = '\t' 
    readCols := make([]interface{}, len(colNames)) 
    writeCols := make([]string, len(colNames)) 
    for i, _ := range writeCols { 
     readCols[i] = &writeCols[i] 
    } 
    for rows.Next() { 
     err := rows.Scan(readCols...) 
     if err != nil { 
      panic(err) 
     } 
     writer.Write(writeCols) 
    } 
    if err = rows.Err(); err != nil { 
     panic(err) 
    } 
    writer.Flush() 
    return nil 
} 

Answer 1

爲了直接將Scan的值轉換爲[]string，必須創建一個[]interface{}切片指向您的字符串切片中的每個字符串。

在這裏，你有MySQL的工作示例（只是改變sql.Open -command來匹配您的設置）：

package main 

import (
    "fmt" 
    _ "github.com/go-sql-driver/mysql" 
    "database/sql" 
) 

func main() { 
    db, err := sql.Open("mysql", "user:pass@tcp(localhost:3306)/test?charset=utf8") 
    defer db.Close() 

    if err != nil { 
     fmt.Println("Failed to connect", err) 
     return 
    } 

    rows, err := db.Query(`SELECT 'one' col1, 'two' col2, 3 col3, NULL col4`) 
    if err != nil { 
     fmt.Println("Failed to run query", err) 
     return 
    } 

    cols, err := rows.Columns() 
    if err != nil { 
     fmt.Println("Failed to get columns", err) 
     return 
    } 

    // Result is your slice string. 
    rawResult := make([][]byte, len(cols)) 
    result := make([]string, len(cols)) 

    dest := make([]interface{}, len(cols)) // A temporary interface{} slice 
    for i, _ := range rawResult { 
     dest[i] = &rawResult[i] // Put pointers to each string in the interface slice 
    } 

    for rows.Next() { 
     err = rows.Scan(dest...) 
     if err != nil { 
      fmt.Println("Failed to scan row", err) 
      return 
     } 

     for i, raw := range rawResult { 
      if raw == nil { 
       result[i] = "\\N" 
      } else { 
       result[i] = string(raw) 
      } 
     } 

     fmt.Printf("%#v\n", result) 
    } 
} 

Answer 2

得到列數（也名）只使用列（）函數

http://golang.org/pkg/database/sql/#Rows.Columns

，併爲CSV只能是一個字符串，只需使用[]字節鍵入掃描儀的目標類型。根據實況 ：

If an argument has type *[]byte, Scan saves in that argument a copy of the corresponding data. The copy is owned by the caller and can be modified and held indefinitely.

的數據將不被轉化成它的實數型。 ，然後從這個[]字節中將其轉換爲字符串。

，如果你確信你的表只使用基本類型（字符串，[]字節，零，INT（S），浮點型（S），布爾）你可以直接傳遞字符串作爲DEST

但如果使用其他類型如數組，枚舉等，然後數據不能轉換爲字符串。但這也取決於驅動程序如何處理這些類型。（幾個月前作爲例子，postgres驅動程序無法處理數組，因此他總是返回[]字節，我需要通過我自己來轉換它）

Answer 3

下面的代碼嬌滴滴statisfies您的要求，您可以在https://gist.github.com/hygull/645c3dc39c69b6b69c06f5ea9deee41f得到這個代碼。表格數據也已提供。

/** 
    { 
     "created_on": "26 may 2017", 
     "todos": [ 
      "go get github.com/go-sql-driver/mysql"  
     ], 
     "aim": "Reading fname column into []string(slice of strings)" 
    } 
*/ 


/* 
mysql> select * from users; 
+----+-----------+----------+----------+-------------------------------+--------------+ 
| id | fname  | lname | uname | email       | contact  | 
+----+-----------+----------+----------+-------------------------------+--------------+ 
| 1 | Rishikesh | Agrawani | hygull | rishikesh0014051992@gmail.com | 917353787704 | 
| 2 | Sandeep | E  | sandeep | sandeepeswar8@gmail.com  | 919739040038 | 
| 3 | Darshan | Sidar | darshan | sidardarshan@gmail.com  | 917996917565 | 
| 4 | Surendra | Prajapat | surendra | surendrakgadwal@gmail.com  | 918385894407 | 
| 5 | Mukesh | Jakhar | mukesh | mjakhar.kjahhar@gmail.com  | 919772254140 | 
+----+-----------+----------+----------+-------------------------------+--------------+ 
5 rows in set (0.00 sec) 

mysql> 
*/ 

package main 
import "fmt" 
import "log" 
import (
    _"github.com/go-sql-driver/mysql" 
    "database/sql" 
) 

func main() { 
    // db, err := sql.Open("mysql", "<username>:<password>@tcp(127.0.0.1:<port>)/<dbname>?charset=utf8") 
    db, err := sql.Open("mysql", "hygull:admin@67@tcp(127.0.0.1:3306)/practice_db?charset=utf8") 

    if err != nil { 
     log.Fatal(err) 
    } 

    rows, err := db.Query("select fname from users") 

    if err != nil { 
     log.Fatal(err) 
    } 

    firstnames:=[]string{} 
    for rows.Next() { 
     var fname string 
     rows.Scan(&fname) 
     firstnames = append(firstnames, fname) 
    } 

    fmt.Println(firstnames) 
    db.Close() 
} 

/* 
[Rishikesh Sandeep Darshan Surendra Mukesh] 
*/