MahmoudEmad/herolib
1
0
Fork 0
Code Issues 55 Pull Requests 1 Actions Packages Projects Releases 26 Wiki Activity

...

Browse Source
This commit is contained in:
despiegk
2025-09-07 12:57:32 +04:00
parent 7f52368a2d
commit 36b4d04288
1 changed files with 106 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
lib/osal/ubuntu/mirrors.v
View File

@@ -7,6 +7,15 @@ import os
import time import time
import net.urllib import net.urllib
import net import net
import sync
pub struct PerfResult {
pub mut:
url string
ping_ms int
speed f64
error string
}
// Fetch Ubuntu mirror list // Fetch Ubuntu mirror list
fn fetch_mirrors() ![]string { fn fetch_mirrors() ![]string {
@@ -39,73 +48,124 @@ fn test_download_speed(mirror string) f64 {
} }
// Ping test (rough ICMP substitute using TCP connect on port 80), returns in ms // Ping test (rough ICMP substitute using TCP connect on port 80), returns in ms
fn test_ping(mirror string) int { fn test_ping(mirror string, mut wg sync.WaitGroup, ch chan PerfResult) ! {
u := urllib.parse(mirror) or { return -1 } defer { wg.done() }
u := urllib.parse(mirror) or {
ch <- PerfResult{
url: mirror
ping_ms: -1
speed: 0.0
}
return
}
host := u.host host := u.host
start := time.now() start := time.now()
mut c := net.dial_tcp('${host}:80') or { return -1 } mut c := net.dial_tcp('${host}:80')!
c.set_blocking(false)!
c.set_write_timeout(time.Duration(5000 * time.millisecond))
c.close() or {} c.close() or {}
return int(time.since(start).milliseconds()) ch <- PerfResult{
url: mirror
ping_ms: int(time.since(start).milliseconds())
speed: 0.0
}
} }
struct MirrorResult { // pub fn fix_mirrors() ! {
url string // println('Fetching Ubuntu mirrors...')
ping_ms int // mirrors := fetch_mirrors() or {
speed f64 // print_backtrace()
} // eprintln(err)
// return
// }
// // mut results := []PerfResult{}
// // mut c := 0
// // for m in mirrors {
// // c++
// // ping := test_ping(m)
// // println('Ping: ${ping} ms - ${mirrors.len} - ${c} ${m}')
// // $dbg;
// // }
// // for m in mirrors {
// // println('Speed: ${test_download_speed(m)} KB/s - ${m}')
// // $dbg;
// // speed := test_download_speed(m)
// // if speed > 0 {
// // ping := 0
// // results << PerfResult{
// // url: m
// // ping_ms: ping
// // speed: speed
// // }
// // println('✅ ${m} | ping: ${ping} ms | speed: ${speed:.2f} KB/s')
// // } else {
// // println('❌ ${m} skipped (unreachable or slow)')
// // }
// // $dbg;
// // }
// // println('\n🏆 Best mirrors:')
// // results.sort_with_compare(fn (a &PerfResult, b &PerfResult) int {
// // // Rank primarily by speed, secondarily by ping
// // if a.speed > b.speed {
// // return -1
// // } else if a.speed < b.speed {
// // return 1
// // } else {
// // return a.ping_ms - b.ping_ms
// // }
// // })
// // for r in results[..results.len.min(10)] {
// // println('${r.url} | ${r.ping_ms} ms | ${r.speed:.2f} KB/s')
// // }
// // println(results)
// // $dbg;
// }
pub fn fix_mirrors() ! { pub fn fix_mirrors() ! {
println('Fetching Ubuntu mirrors...') // Create wait group for servers
mut wg := sync.new_waitgroup()
wg.add(500)
ch := chan PerfResult{cap: 1000}
mirrors := fetch_mirrors() or { mirrors := fetch_mirrors() or {
print_backtrace() print_backtrace()
eprintln(err) eprintln(err)
return return
} }
mut results := []MirrorResult{}
mut c := 0 mut c := 0
mut result := []PerfResult{}
for m in mirrors { for m in mirrors {
c++ c++
ping := test_ping(m) println('Start background ping - ${mirrors.len} - ${c} ${m} - Queue len: ${ch.len} / ${ch.cap}')
println('Ping: ${ping} ms - ${mirrors.len} - ${c} ${m}')
$dbg; l := ch.len // number of elements in queue
for l > ch.cap - 2 { // if queue is full, wait
println('Queue full, wait till some are done')
time.sleep(1 * time.second)
}
spawn test_ping(m, mut wg, ch)
} }
for m in mirrors { for {
println('Speed: ${test_download_speed(m)} KB/s - ${m}') value := <-ch or { // receive/pop values from the channel
$dbg; println('Channel closed')
speed := test_download_speed(m) break
if speed > 0 {
ping := 0
results << MirrorResult{
url: m
ping_ms: ping
speed: speed
}
println('✅ ${m} | ping: ${ping} ms | speed: ${speed:.2f} KB/s')
} else {
println('❌ ${m} skipped (unreachable or slow)')
} }
$dbg; println('Received: ${value}')
} }
println('\n🏆 Best mirrors:') println('All pings done 1')
results.sort_with_compare(fn (a &MirrorResult, b &MirrorResult) int {
// Rank primarily by speed, secondarily by ping
if a.speed > b.speed {
return -1
} else if a.speed < b.speed {
return 1
} else {
return a.ping_ms - b.ping_ms
}
})
// for r in results[..results.len.min(10)] { wg.wait()
// println('${r.url} | ${r.ping_ms} ms | ${r.speed:.2f} KB/s')
// }
println(results) println('All pings done')
$dbg;
} }