From dfe8e3b47381167a73a5122c794aa711188c2bea Mon Sep 17 00:00:00 2001 From: Lee Jaeyong Date: Fri, 21 Sep 2018 04:43:39 +0900 Subject: [PATCH] Fix config file types of csv_skip_rows and csv_skip_columns (#4726) --- internal/config/config.go | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/internal/config/config.go b/internal/config/config.go index 8010ab6fe..cd4d9825c 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -1541,24 +1541,24 @@ func getParserConfig(name string, tbl *ast.Table) (*parsers.Config, error) { if node, ok := tbl.Fields["csv_skip_rows"]; ok { if kv, ok := node.(*ast.KeyValue); ok { - if str, ok := kv.Value.(*ast.String); ok { - iVal, err := strconv.Atoi(str.Value) - c.CSVSkipRows = iVal + if integer, ok := kv.Value.(*ast.Integer); ok { + v, err := integer.Int() if err != nil { - return nil, fmt.Errorf("E! parsing to int: %v", err) + return nil, err } + c.CSVHeaderRowCount = int(v) } } } if node, ok := tbl.Fields["csv_skip_columns"]; ok { if kv, ok := node.(*ast.KeyValue); ok { - if str, ok := kv.Value.(*ast.String); ok { - iVal, err := strconv.Atoi(str.Value) - c.CSVSkipColumns = iVal + if integer, ok := kv.Value.(*ast.Integer); ok { + v, err := integer.Int() if err != nil { - return nil, fmt.Errorf("E! parsing to int: %v", err) + return nil, err } + c.CSVHeaderRowCount = int(v) } } }