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
|
package navitia_api_client
import (
"encoding/json"
"fmt"
"log"
"net/http"
"git.adyxax.org/adyxax/trains/pkg/model"
)
type TrainStopsResponse struct {
Pagination struct {
StartPage int `json:"start_page"`
ItemsOnPage int `json:"items_on_page"`
ItemsPerPage int `json:"items_per_page"`
TotalResult int `json:"total_result"`
} `json:"pagination"`
StopAreas []struct {
Name string `json:"name"`
ID string `json:"id"`
Codes []interface{} `json:"codes"`
Links []interface{} `json:"links"`
Coord interface{} `json:"coord"`
Label interface{} `json:"label"`
Timezone interface{} `json:"timezone"`
AdministrativeRegion interface{} `json:"administrative_regions"`
} `json:"stop_areas"`
Links []interface{} `json:"links"`
Disruptions []interface{} `json:"disruptions"`
FeedPublishers []interface{} `json:"feed_publishers"`
Context interface{} `json:"context"`
}
func (c *NavitiaClient) GetTrainStops() (trainStops []model.TrainStop, err error) {
return getTrainStopsPage(c, 0)
}
func getTrainStopsPage(c *NavitiaClient, i int) (trainStops []model.TrainStop, err error) {
request := fmt.Sprintf("%s/coverage/sncf/stop_areas?count=1000&start_page=%d", c.baseURL, i)
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 TrainStopsResponse
if err = json.NewDecoder(resp.Body).Decode(&data); err != nil {
return nil, newJsonDecodeError("GetTrainStops ", err)
}
for i := 0; i < len(data.StopAreas); i++ {
trainStops = append(trainStops, model.TrainStop{data.StopAreas[i].ID, data.StopAreas[i].Name})
}
if data.Pagination.ItemsOnPage+data.Pagination.ItemsPerPage*data.Pagination.StartPage < data.Pagination.TotalResult {
log.Printf("pagination %d\n", i)
tss, err := getTrainStopsPage(c, i+1)
if err != nil {
return nil, err
}
trainStops = append(trainStops, tss...)
}
} else {
err = newApiError(resp.StatusCode, "GetTrainStops")
}
return
}
|
