forked from SimpleApplicationsOrg/stock
/
timeSeries.go
98 lines (76 loc) · 2.46 KB
/
timeSeries.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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package alphavantage
import (
"encoding/json"
"fmt"
"github.com/SimpleApplicationsOrg/stock/client"
"github.com/tidwall/gjson"
"log"
)
const (
Intraday = "TIME_SERIES_INTRADAY"
Daily = "TIME_SERIES_DAILY"
DailyAdjusted = "TIME_SERIES_DAILY_ADJUSTED"
Weekly = "TIME_SERIES_WEEKLY"
Monthly = "TIME_SERIES_MONTHLY"
)
// TimeSeriesIntraday returns intraday time series (timestamp, open, high, low, close, volume) of the equity specified, updated realtime.
func (api *AVClient) TimeSeriesIntraday(symbol, interval string) (*TimeSeriesData, error) {
req := client.NewAPIReq().WithPath("query").
AddParam("function", Intraday).
AddParam("symbol", symbol).
AddParam("interval", interval)
res, err := timeSeriesCall(*api.client, req)
metaData := gjson.Get(res, "Meta Data")
timeSeries := gjson.Get(res, fmt.Sprintf("Time Series (%s)", interval))
timeSeriesMetaData, err := buildTimeSeriesData(metaData.String(), timeSeries.String())
if err != nil {
return nil, err
}
return timeSeriesMetaData, nil
}
// TimeSeries gets daily, weekly and monthly series
func (api *AVClient) TimeSeries(function, symbol string) (*TimeSeriesData, error) {
req := client.NewAPIReq().WithPath("query").
AddParam("function", function).
AddParam("symbol", symbol)
res, err := timeSeriesCall(*api.client, req)
metaData := gjson.Get(res, "Meta Data")
timeSeries := gjson.Get(res, timeSeriesType(function))
timeSeriesMetaData, err := buildTimeSeriesData(metaData.String(), timeSeries.String())
if err != nil {
return nil, err
}
return timeSeriesMetaData, nil
}
func timeSeriesType(function string) string {
var timeSeriesType string
if function == Daily || function == DailyAdjusted {
timeSeriesType = "Time Series (Daily)"
} else if function == Weekly {
timeSeriesType = "Weekly Time Series"
} else if function == Monthly {
timeSeriesType = "Monthly Time Series"
}
return timeSeriesType
}
func timeSeriesCall(client client.APIClient, req *client.APIRequest) (string, error) {
res, err := client.Call(req)
if err != nil {
log.Printf("timeSeriesCall: %s", err.Error())
return "", err
}
return res, nil
}
func buildTimeSeriesData(metaData, timeSeries string) (*TimeSeriesData, error) {
var meta MetaData
err := json.Unmarshal([]byte(metaData), &meta)
if err != nil {
return nil, err
}
var series TimeSeries
err = json.Unmarshal([]byte(timeSeries), &series)
if err != nil {
return nil, err
}
return &TimeSeriesData{&meta, &series}, nil
}