/
csv.go
51 lines (47 loc) · 1.17 KB
/
csv.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
package parser
import (
"bytes"
"encoding/csv"
"strconv"
"strings"
"time"
"github.com/patrickmn/go-cache"
)
type Cache struct {
// contains filtered or unexported fields
}
func ParseCsvFileData(byteFile []byte, delimiter, keyName string, localCache *cache.Cache) ([]*map[string]interface{}, error) {
var (
result []*map[string]interface{}
err error
headers []string
data [][]string
delimited int
)
// if already data in cache
if cacheData, found := localCache.Get(keyName); found {
return cacheData.([]*map[string]interface{}), nil
}
// if already delimited
if delimited, err = strconv.Atoi(delimiter); err != nil {
delimited = ','
}
// initialize reader
reader := csv.NewReader(bytes.NewReader(byteFile))
reader.Comma = rune(delimited)
if data, err = reader.ReadAll(); err != nil {
return nil, err
}
// get headers
headers = data[0]
for _, row := range data[1:] {
var tmpMap = make(map[string]interface{})
for j, v := range row {
tmpMap[strings.Join(strings.Split(headers[j], " "), "")] = v
}
result = append(result, &tmpMap)
}
// set data to cache
localCache.Set(keyName, result, 10*time.Minute)
return result, nil
}