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
|
package navitia_api_client
import (
"encoding/json"
"fmt"
"net/http"
"time"
"git.adyxax.org/adyxax/trains/pkg/model"
)
type DeparturesResponse struct {
Disruptions []interface{} `json:"disruptions"`
Notes []interface{} `json:"notes"`
Departures []struct {
DisplayInformations struct {
Direction string `json:"direction"`
Code string `json:"code"`
Network string `json:"network"`
Links []interface{} `json:"links"`
Color string `json:"color"`
Name string `json:"name"`
PhysicalMode string `json:"physical_mode"`
Headsign string `json:"headsign"`
Label string `json:"label"`
Equipments []interface{} `json:"equipments"`
TextColor string `json:"text_color"`
TripShortName string `json:"trip_short_name"`
CommercialMode string `json:"commercial_mode"`
Description string `json:"description"`
} `json:"display_informations"`
StopDateTime struct {
Links []interface{} `json:"links"`
ArrivalDateTime string `json:"arrival_date_time"`
AdditionalInformations []interface{} `json:"additional_informations"`
DepartureDateTime string `json:"departure_date_time"`
BaseArrivalDateTime string `json:"base_arrival_date_time"`
BaseDepartureDateTime string `json:"base_departure_date_time"`
DataFreshness string `json:"data_freshness"`
} `json:"stop_date_time"`
} `json:"departures"`
Context struct {
Timezone string `json:"timezone"`
CurrentDatetime string `json:"current_datetime"`
} `json:"context"`
}
func (c *NavitiaClient) GetDepartures(stop string) (departures []model.Departure, err error) {
request := fmt.Sprintf("%s/coverage/sncf/stop_areas/%s/departures", c.baseURL, stop)
start := time.Now()
c.mutex.Lock()
defer c.mutex.Unlock()
if cachedResult, ok := c.cache[request]; ok {
if start.Sub(cachedResult.ts) < 60*1000*1000*1000 {
return cachedResult.result.([]model.Departure), nil
}
}
req, err := http.NewRequest("GET", request, nil)
if err != nil {
return nil, newHttpClientError("http.NewRequest error", err)
}
resp, err := c.httpClient.Do(req)
if err != nil {
return nil, newHttpClientError("httpClient.Do error", err)
}
defer resp.Body.Close()
if resp.StatusCode == http.StatusOK {
var data DeparturesResponse
if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
return nil, newJsonDecodeError("GetDepartures "+stop, err)
}
// TODO test for no json error
// TODO handle pagination
for i := 0; i < len(data.Departures); i++ {
t, err := time.Parse("20060102T150405", data.Departures[i].StopDateTime.ArrivalDateTime)
if err != nil {
return nil, newDateParsingError(data.Departures[i].StopDateTime.ArrivalDateTime, err)
}
departures = append(departures, model.Departure{data.Departures[i].DisplayInformations.Direction, t.Format("Mon, 02 Jan 2006 15:04:05")})
}
c.cache[request] = cachedResult{
ts: start,
result: departures,
}
} else {
err = newApiError(resp.StatusCode, "GetDepartures "+stop)
}
return
}
|
