136 lines
3.4 KiB
Go
136 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/showwin/speedtest-go/speedtest"
|
|
)
|
|
|
|
var (
|
|
latency = time.Duration(0)
|
|
downspeed = 0.0
|
|
upspeed = 0.0
|
|
|
|
interval int
|
|
port int
|
|
serverID int
|
|
)
|
|
|
|
func init() {
|
|
flag.IntVar(&interval, "interval", 30, "Duration, in minutes, between speedtest runs")
|
|
flag.IntVar(&port, "port", 9091, "The port to listen on")
|
|
flag.IntVar(&serverID, "server", 55137, "The ID of the server to test against")
|
|
flag.Parse()
|
|
}
|
|
|
|
func main() {
|
|
go (func() {
|
|
for {
|
|
checkSpeed()
|
|
time.Sleep(time.Minute * time.Duration(interval))
|
|
}
|
|
})()
|
|
|
|
collector := newSpeedCollector()
|
|
prometheus.MustRegister(collector)
|
|
http.Handle("/metrics", promhttp.Handler())
|
|
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", port), nil))
|
|
}
|
|
|
|
func checkSpeed() {
|
|
log.Println("Performing speedtest")
|
|
user, err := speedtest.FetchUserInfo()
|
|
if err != nil {
|
|
log.Printf("Error fetching user info: %v\n", err)
|
|
return
|
|
}
|
|
serverList, err := speedtest.FetchServers(user)
|
|
if err != nil {
|
|
log.Printf("Error fetching server list: %v\n", err)
|
|
return
|
|
}
|
|
targets, err := serverList.FindServer([]int{serverID})
|
|
if err != nil {
|
|
log.Printf("Error finding server: %v\n", err)
|
|
return
|
|
}
|
|
if len(targets) == 0 {
|
|
log.Printf("No servers found, falling back to defaults...")
|
|
targets, err = serverList.FindServer([]int{serverID})
|
|
if err != nil {
|
|
log.Printf("Error finding server: %v\n", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
target := targets[0]
|
|
log.Printf("Testing against server: %s - %s [%s]\n", target.Name, target.Sponsor, target.ID)
|
|
|
|
target.PingTest()
|
|
target.DownloadTest(false)
|
|
target.UploadTest(false)
|
|
latency = target.Latency
|
|
downspeed = target.DLSpeed
|
|
upspeed = target.ULSpeed
|
|
log.Printf("Finished speedtest. DL=%f UL=%f Ping=%v\n", downspeed, upspeed, latency)
|
|
|
|
if os.Getenv("DEBUG") != "" {
|
|
log.Println("Debug enabled, testing all servers...")
|
|
for _, target := range serverList {
|
|
log.Printf("Testing against server: %s - %s [%s]\n", target.Name, target.Sponsor, target.ID)
|
|
target.PingTest()
|
|
target.DownloadTest(false)
|
|
target.UploadTest(false)
|
|
if target.DLSpeed > 0 && target.ULSpeed > 0 {
|
|
log.Printf("Finished speedtest. DL=%f UL=%f Ping=%v\n", target.DLSpeed, target.ULSpeed, target.Latency)
|
|
} else {
|
|
log.Printf("Finished speedtest. Failed to get valid results\n")
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
type speedCollector struct {
|
|
downMetric *prometheus.Desc
|
|
upMetric *prometheus.Desc
|
|
latencyMetric *prometheus.Desc
|
|
}
|
|
|
|
func newSpeedCollector() *speedCollector {
|
|
return &speedCollector{
|
|
downMetric: prometheus.NewDesc("speedtest_download_speed",
|
|
"Download speed in Mbit/s",
|
|
nil, nil,
|
|
),
|
|
upMetric: prometheus.NewDesc("speedtest_upload_speed",
|
|
"Upload speed in Mbit/s",
|
|
nil, nil,
|
|
),
|
|
latencyMetric: prometheus.NewDesc("speedtest_latency",
|
|
"Latency in ms",
|
|
nil, nil,
|
|
),
|
|
}
|
|
}
|
|
|
|
func (collector *speedCollector) Describe(ch chan<- *prometheus.Desc) {
|
|
ch <- collector.downMetric
|
|
ch <- collector.upMetric
|
|
ch <- collector.latencyMetric
|
|
}
|
|
|
|
func (collector *speedCollector) Collect(ch chan<- prometheus.Metric) {
|
|
ch <- prometheus.MustNewConstMetric(collector.downMetric, prometheus.CounterValue, downspeed)
|
|
ch <- prometheus.MustNewConstMetric(collector.upMetric, prometheus.CounterValue, upspeed)
|
|
ch <- prometheus.MustNewConstMetric(collector.latencyMetric, prometheus.CounterValue, float64(latency.Milliseconds()))
|
|
}
|