sql: scan row (s) с неизвестным количеством столбцов (выберите * from ...)

Question

У меня есть таблица t, содержащая множество столбцов, и мой sql выглядит так: select * from t. Теперь я хочу только сканировать один или два столбца из набора расширенных строк. Тем не менее, sql.Scan принимает dest ...interface{} в качестве аргументов. Означает ли это, что я должен проверять все и использовать только тот столбец, который мне нужен?

Я знаю, что могу изменить sql от select * до select my_favorite_rows, однако в этом случае у меня нет возможности изменить sql.

Спасибо, Elgs

Answer 1

Вы можете использовать Rows.Columns, например,

package main 

import (
    "database/sql" 
    "fmt" 

    "github.com/lib/pq" 
) 

type Vehicle struct { 
    Id  int 
    Name string 
    Wheels int 
} 

// VehicleCol returns a reference for a column of a Vehicle 
func VehicleCol(colname string, vh *Vehicle) interface{} { 
    switch colname { 
    case "id": 
     return &vh.Id 
    case "name": 
     return &vh.Name 
    case "wheels": 
     return &vh.Wheels 
    default: 
     panic("unknown column " + colname) 
    } 
} 

func panicOnErr(err error) { 
    if err != nil { 
     panic(err.Error()) 
    } 
} 

func main() { 
    conn, err := pq.ParseURL(`postgres://docker:docker@172.17.0.2:5432/pgsqltest?schema=public`) 
    panicOnErr(err) 

    var db *sql.DB 
    db, err = sql.Open("postgres", conn) 
    panicOnErr(err) 

    var rows *sql.Rows 
    rows, err = db.Query("select * from vehicle") 
    panicOnErr(err) 

    // get the column names from the query 
    var columns []string 
    columns, err = rows.Columns() 
    panicOnErr(err) 

    colNum := len(columns) 

    all := []Vehicle{} 

    for rows.Next() { 
     vh := Vehicle{} 

     // make references for the cols with the aid of VehicleCol 
     cols := make([]interface{}, colNum) 
     for i := 0; i < colNum; i++ { 
      cols[i] = VehicleCol(columns[i], &vh) 
     } 

     err = rows.Scan(cols...) 
     panicOnErr(err) 

     all = append(all, vh) 
    } 

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