adds speedtest to perf tests

Signed-off-by: Ashraf Fouda <ashraf.m.fouda@gmail.com>

Author: ashraffouda

go.mod

@@ -45,6 +45,7 @@ require (
 	github.com/pkg/errors v0.9.1
 	github.com/rs/zerolog v1.34.0
 	github.com/shirou/gopsutil v3.21.11+incompatible
+	github.com/showwin/speedtest-go v1.7.10
 	github.com/stretchr/testify v1.10.0
 	github.com/threefoldtech/0-fs v1.3.1-0.20240424140157-b488dfedcc56
 	github.com/threefoldtech/tfchain/clients/tfchain-client-go v0.0.0-20241127100051-77e684bcb1b2
@@ -64,7 +65,6 @@ require (
 )
 
 require (
-	github.com/fatih/color v1.18.0 // indirect
 	github.com/frankban/quicktest v1.14.3 // indirect
 	github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad // indirect
 	github.com/hashicorp/go-cleanhttp v0.5.2 // indirect

go.sum

@@ -535,6 +535,8 @@ github.com/safchain/ethtool v0.0.0-20201023143004-874930cb3ce0/go.mod h1:Z0q5wiB
 github.com/seccomp/libseccomp-golang v0.9.2-0.20210429002308-3879420cc921/go.mod h1:JA8cRccbGaA1s33RQf7Y1+q9gHmZX1yB/z9WDN1C6fg=
 github.com/shirou/gopsutil v3.21.11+incompatible h1:+1+c1VGhc88SSonWP6foOcLhvnKlUeu/erjjvaPEYiI=
 github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
+github.com/showwin/speedtest-go v1.7.10 h1:9o5zb7KsuzZKn+IE2//z5btLKJ870JwO6ETayUkqRFw=
+github.com/showwin/speedtest-go v1.7.10/go.mod h1:Ei7OCTmNPdWofMadzcfgq1rUO7mvJy9Jycj//G7vyfA=
 github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
 github.com/sirupsen/logrus v1.0.6/go.mod h1:pMByvHTf9Beacp5x1UXfOR9xyW/9antXMhjMPG0dEzc=
 github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=

pkg/perf/iperf/iperf_task.go

@@ -13,10 +13,11 @@ import (
 	"github.com/cenkalti/backoff"
 	"github.com/pkg/errors"
 	"github.com/rs/zerolog/log"
+	"github.com/showwin/speedtest-go/speedtest"
 	"github.com/threefoldtech/zosbase/pkg/environment"
 	"github.com/threefoldtech/zosbase/pkg/network/iperf"
 	"github.com/threefoldtech/zosbase/pkg/perf"
-	"github.com/threefoldtech/zosbase/pkg/perf/exec_wrapper"
+	execwrapper "github.com/threefoldtech/zosbase/pkg/perf/exec_wrapper"
 	"github.com/threefoldtech/zosbase/pkg/perf/graphql"
 )
 
@@ -66,7 +67,7 @@ func (t *IperfTest) ID() string {
 
 // Cron returns the schedule for the tcp task
 func (t *IperfTest) Cron() string {
-	return "0 0 */6 * * *"
+	return "*/10 * * * * *"
 }
 
 // Description returns the task description
@@ -222,6 +223,31 @@ func (t *IperfTest) runIperfTest(ctx context.Context, clientIP string, tcp bool)
 	if !tcp && len(report.End.Streams) > 0 {
 		iperfResult.DownloadSpeed = report.End.Streams[0].UDP.BitsPerSecond
 	}
+	var speedtestServer *speedtest.Server
+	serverList, _ := speedtest.FetchServers()
+	servers, _ := serverList.FindServer([]int{})
+	if len(servers) > 0 {
+		speedtestServer = servers[0]
+		err = speedtestServer.DownloadTest()
+		if err != nil {
+			log.Error().Err(err).Msg("speedtest download test failed")
+		}
+		err = speedtestServer.UploadTest()
+		if err != nil {
+			log.Error().Err(err).Msg("speedtest upload test failed")
+		}
+		downloadSpeed := float64(speedtestServer.DLSpeed) * 8
+		uploadSpeed := float64(speedtestServer.ULSpeed) * 8
+		log.Debug().Msgf("speedtest: %f/%f  vs iperf results: %f/%f", downloadSpeed, uploadSpeed, iperfResult.DownloadSpeed, iperfResult.UploadSpeed)
+		// take the best result
+		if downloadSpeed > iperfResult.DownloadSpeed {
+			iperfResult.DownloadSpeed = downloadSpeed
+		}
+		if uploadSpeed > iperfResult.UploadSpeed {
+			iperfResult.UploadSpeed = uploadSpeed
+		}
+
+	}
 
 	return iperfResult
 }