rir/providers.go at master · simcap/rir · GitHub

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
package main

import (
	"bytes"
	"io"
	"io/ioutil"
	"log"
	"net/http"
)

type Provider interface {
	Name() string
	GetData() io.Reader
}

type DefaultProvider struct {
	name string
	url  string
}

func (p *DefaultProvider) Name() string {
	return p.name
}

func (p *DefaultProvider) GetData() io.Reader {
	log.Printf("Fetching %s data", p.Name())
	response, err := http.Get(p.url)
	if err != nil {
		log.Fatal(err)
	}
	defer response.Body.Close()

	if status := response.StatusCode; status != 200 {
		log.Fatalf("HTTP call returned %d", status)
	}

	content, err := ioutil.ReadAll(response.Body)
	if err != nil {
		log.Fatal(err)
	}

	return bytes.NewBuffer(content)
}

var (
	AllProviders = []*CachedProvider{
		NewCachedProvider(
			"afrinic",
			"https://ftp.ripe.net/pub/stats/afrinic/delegated-afrinic-extended-latest",
		),
		NewCachedProvider(
			"apnic",
			"https://ftp.ripe.net/pub/stats/apnic/delegated-apnic-extended-latest",
		),
		NewCachedProvider(
			"arin",
			"https://ftp.ripe.net/pub/stats/arin/delegated-arin-extended-latest",
		),
		NewCachedProvider(
			"lacnic",
			"https://ftp.ripe.net/pub/stats/lacnic/delegated-lacnic-extended-latest",
		),
		NewCachedProvider(
			"ripencc",
			"https://ftp.ripe.net/pub/stats/ripencc/delegated-ripencc-extended-latest",
		),
		//NewCachedProvider(
		//	"iana",
		//	"http://ftp.apnic.net/stats/iana/delegated-iana-latest",
		//),
	}
)