73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
|
package records
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/csv"
|
||
|
|
"io"
|
||
|
|
"log"
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
type FinancialData struct {
|
||
|
|
Date string
|
||
|
|
Open, High, Low, Close, Adj_Close, Volume float64
|
||
|
|
}
|
||
|
|
|
||
|
|
func readCsvLine(cols []string, line []string) *FinancialData {
|
||
|
|
data := new(FinancialData)
|
||
|
|
for i, col := range line {
|
||
|
|
switch cols[i] {
|
||
|
|
case "Date":
|
||
|
|
data.Date = col
|
||
|
|
case "Open":
|
||
|
|
data.Open = toFloat(col)
|
||
|
|
case "High":
|
||
|
|
data.High = toFloat(col)
|
||
|
|
case "Low":
|
||
|
|
data.Low = toFloat(col)
|
||
|
|
case "Close":
|
||
|
|
data.Close = toFloat(col)
|
||
|
|
case "Adj Close":
|
||
|
|
data.Adj_Close = toFloat(col)
|
||
|
|
case "Volume":
|
||
|
|
data.Volume = toFloat(col)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
|
||
|
|
func toFloat(v string) float64 {
|
||
|
|
f, err := strconv.ParseFloat(v, 64)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalln(err)
|
||
|
|
}
|
||
|
|
return f
|
||
|
|
}
|
||
|
|
|
||
|
|
func readCsvHeader(line []string) []string {
|
||
|
|
cols := []string{}
|
||
|
|
for _, col := range line {
|
||
|
|
cols = append(cols, col)
|
||
|
|
}
|
||
|
|
return cols
|
||
|
|
}
|
||
|
|
|
||
|
|
func ReadCsv(in io.Reader) ([]string, []FinancialData) {
|
||
|
|
cols := []string{}
|
||
|
|
data := []FinancialData{}
|
||
|
|
csvReader := csv.NewReader(in)
|
||
|
|
for {
|
||
|
|
line, err := csvReader.Read()
|
||
|
|
if err == io.EOF {
|
||
|
|
break
|
||
|
|
} else if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
if len(cols) == 0 {
|
||
|
|
cols = readCsvHeader(line)
|
||
|
|
} else {
|
||
|
|
data = append(data, *readCsvLine(cols, line))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return cols, data
|
||
|
|
}
|