This commit is contained in:
Gregory Bednov 2025-07-16 13:57:19 +03:00
commit 9c88f7e744
3 changed files with 14 additions and 14 deletions

View file

@ -25,6 +25,7 @@ type Peer struct {
var PublicPeers string var PublicPeers string
const repoURL = "https://github.com/yggdrasil-network/public-peers" const repoURL = "https://github.com/yggdrasil-network/public-peers"
const localPeersPath = "peers.txt"
func readPeersFile(path string) []url.URL { func readPeersFile(path string) []url.URL {
data, err := os.ReadFile(path) data, err := os.ReadFile(path)
@ -53,13 +54,13 @@ func readPeersFile(path string) []url.URL {
func getPublicPeers() []url.URL { func getPublicPeers() []url.URL {
tempDir, err := os.MkdirTemp("", "public-peers-*") tempDir, err := os.MkdirTemp("", "public-peers-*")
if err != nil { if err != nil {
return readPeersFile("peers.txt") return readPeersFile(localPeersPath)
} }
defer os.RemoveAll(tempDir) defer os.RemoveAll(tempDir)
_, err = git.PlainCloneContext(context.Background(), tempDir, false, &git.CloneOptions{URL: repoURL, Depth: 1}) _, err = git.PlainCloneContext(context.Background(), tempDir, false, &git.CloneOptions{URL: repoURL, Depth: 1})
if err != nil { if err != nil {
return readPeersFile("peers.txt") return readPeersFile(localPeersPath)
} }
var peers []url.URL var peers []url.URL
@ -86,7 +87,7 @@ func getPublicPeers() []url.URL {
}) })
if len(peers) == 0 { if len(peers) == 0 {
return readPeersFile("peers.txt") return readPeersFile(localPeersPath)
} }
return peers return peers
} }

View file

@ -8,7 +8,6 @@ import (
"github.com/gologme/log" "github.com/gologme/log"
"github.com/spf13/viper" "github.com/spf13/viper"
adminapi "github.com/yggdrasil-network/yggdrasil-go/src/admin"
yggConfig "github.com/yggdrasil-network/yggdrasil-go/src/config" yggConfig "github.com/yggdrasil-network/yggdrasil-go/src/config"
"github.com/yggdrasil-network/yggdrasil-go/src/core" "github.com/yggdrasil-network/yggdrasil-go/src/core"
) )
@ -26,7 +25,7 @@ func TestConnectivity(config *viper.Viper) error {
cfg.AdminListen = ygg.GetString("admin_listen") cfg.AdminListen = ygg.GetString("admin_listen")
cfg.Listen = ygg.GetStringSlice("listen") cfg.Listen = ygg.GetStringSlice("listen")
if ygg.GetString("peers") == "auto" { if ygg.GetString("peers") == "auto" {
publicPeers := getPublicPeers() publicPeers := RandomPick(GetClosestPeers(getPublicPeers(), 20), 3)
var urls []string var urls []string
for _, u := range publicPeers { for _, u := range publicPeers {
urls = append(urls, u.String()) urls = append(urls, u.String())
@ -72,15 +71,15 @@ func TestConnectivity(config *viper.Viper) error {
defer n.core.Stop() defer n.core.Stop()
// Start admin socket to query peers. // Start admin socket to query peers.
adminOpts := []adminapi.SetupOption{ //adminOpts := []adminapi.SetupOption{
adminapi.ListenAddress(cfg.AdminListen), // adminapi.ListenAddress(cfg.AdminListen),
} //}
n.admin, err = adminapi.New(n.core, logger, adminOpts...) //n.admin, err = adminapi.New(n.core, logger, adminOpts...)
if err != nil { //if err != nil {
return err // return err
} //}
n.admin.SetupAdminHandlers() //n.admin.SetupAdminHandlers()
defer n.admin.Stop() //defer n.admin.Stop()
// Give the node some time to establish connections. // Give the node some time to establish connections.
time.Sleep(5 * time.Second) time.Sleep(5 * time.Second)