Created
August 20, 2021 09:54
-
-
Save linw1995/7e61ff81625848c277920f32941d2e88 to your computer and use it in GitHub Desktop.
Golang read csv data row into struct
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"bytes" | |
"encoding/csv" | |
"fmt" | |
"io" | |
"reflect" | |
) | |
type Reader struct { | |
*csv.Reader | |
Fields []string | |
} | |
func NewReader(r io.Reader) *Reader { | |
rd := csv.NewReader(r) | |
rd.FieldsPerRecord = -1 | |
fields, err := rd.Read() | |
if err != nil { | |
panic(err) | |
} | |
return &Reader{ | |
Reader: rd, | |
Fields: fields, | |
} | |
} | |
func (r *Reader) ReadInto(v interface{}) (err error) { | |
rv := reflect.ValueOf(v) | |
for rv.Kind() != reflect.Ptr || rv.IsNil() { | |
return fmt.Errorf("Invalid type of output %s", reflect.TypeOf(v)) | |
} | |
row, err := r.Read() | |
if err != nil { | |
return | |
} | |
mapping := make(map[string]string) | |
lenrow := len(row) | |
for idx, fieldname := range r.Fields { | |
if idx < lenrow { | |
mapping[fieldname] = row[idx] | |
} else { | |
mapping[fieldname] = "" | |
} | |
} | |
typ := reflect.TypeOf(v).Elem() | |
obj := rv.Elem() | |
for i := 0; i < typ.NumField(); i++ { | |
nf := typ.Field(i) | |
name := nf.Tag.Get("csv") | |
obj.FieldByName(nf.Name).SetString(mapping[name]) | |
} | |
return nil | |
} | |
type Row struct { | |
A string `csv:"a"` | |
B string `csv:"b"` | |
} | |
func main() { | |
src := bytes.NewReader([]byte(`a,b,c | |
1,2,3 | |
2,3,4`)) | |
r := NewReader(src) | |
for { | |
var row Row | |
err := r.ReadInto(&row) | |
if err != nil { | |
fmt.Println(err) | |
return | |
} | |
fmt.Printf("%#v\n", row) | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
https://play.golang.org/p/sakfw-IUyNm