still testing yggdrasil

cli/testyggdrasil.go

@@ -1,9 +1,19 @@
 package cli
 
 import (
+	"context"
 	"fmt"
+	"io"
 	"lbc/cfg"
 	"lbc/yggdrasil"
+	"log"
+	"net"
+	"net/url"
+	"os"
+	"os/signal"
+	"strings"
+	"syscall"
+	"time"
 
 	"github.com/spf13/cobra"
 )
@@ -14,16 +24,106 @@ var testYggdrasilCmd = &cobra.Command{
 	RunE: func(cmd *cobra.Command, args []string) error {
 		v, err := cfg.LoadViperConfig(defaultConfigPath)
 		if err != nil {
-			return fmt.Errorf("не удалось прочитать конфигурацию: %w", err)
+			fmt.Fprintf(os.Stderr, "не удалось прочитать конфигурацию viper: %v", err)
+			os.Exit(1)
 		}
-		if err := yggdrasil.TestConnectivity(v); err != nil {
-			return fmt.Errorf("тест не пройден: %w", err)
+		config, err := cfg.ReadConfig(defaultConfigPath)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "конфигурация не прочитана: %v", err)
 		}
-		fmt.Println("Yggdrasil connectivity test successful")
+		ctx, cancel := context.WithCancel(context.Background())
+		laddrReturner := make(chan string, 2)
+		go yggdrasil.Yggdrasil(v, laddrReturner)
+		go notblockchain(ctx, dbPath, config, laddrReturner)
+
+		sigCh := make(chan os.Signal, 1)
+		signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+		<-sigCh  // ждём SIGINT/SIGTERM
+		cancel() // говорим горутинам завершаться
 		return nil
 	},
 }
 
+func notblockchain(ctx context.Context, dbPath string, config *cfg.Config, laddrReturner chan string) error {
+	isGenesis := false
+	laddr := "tcp://" + <-laddrReturner
+	p2peers := <-laddrReturner
+
+	u, err := url.Parse(laddr)
+	if err != nil {
+		log.Fatalf("bad laddr %q: %v", laddr, err)
+	}
+	proto := u.Scheme    // "tcp"
+	listenAddr := u.Host // "127.0.0.1:8000"
+	ln, err := net.Listen(proto, listenAddr)
+	if err != nil {
+		log.Fatalf("listen %s failed: %v", listenAddr, err)
+	}
+	log.Printf("listening on %s://%s", proto, listenAddr)
+
+	// --- парсим пиров ---
+	var peerAddrs []string
+	for _, entry := range strings.Split(p2peers, ",") {
+		parts := strings.SplitN(entry, "@", 2)
+		if len(parts) != 2 {
+			log.Printf("skip invalid peer entry: %q", entry)
+			continue
+		}
+		peerAddrs = append(peerAddrs, parts[1])
+	}
+	if len(peerAddrs) == 0 {
+		log.Fatalf("нет ни одного валидного пира в %q", p2peers)
+	}
+
+	// --- стартовое сообщение от не-genesis ---
+	if !isGenesis {
+		initial := []byte("HELLO_FROM_JOINER\n")
+		for _, pa := range peerAddrs {
+			go sendToPeer(proto, pa, initial)
+		}
+	}
+
+	// --- главный цикл приёма+форвард---
+	for {
+		conn, err := ln.Accept()
+		if err != nil {
+			log.Printf("accept error: %v", err)
+			continue
+		}
+		go func(c net.Conn) {
+			defer c.Close()
+			data, err := io.ReadAll(c)
+			if err != nil {
+				log.Printf("read error: %v", err)
+				return
+			}
+
+			log.Printf("received %d bytes, waiting 10s before forwarding", len(data))
+			time.Sleep(10 * time.Second)
+
+			for _, pa := range peerAddrs {
+				go sendToPeer(proto, pa, data)
+			}
+		}(conn)
+	}
+}
+
+// sendToPeer коннектится к одному пиру и шлёт данные.
+func sendToPeer(proto, addr string, data []byte) {
+	conn, err := net.Dial(proto, addr)
+	if err != nil {
+		log.Printf("dial %s://%s error: %v", proto, addr, err)
+		return
+	}
+	defer conn.Close()
+
+	if _, err := conn.Write(data); err != nil {
+		log.Printf("write to %s error: %v", addr, err)
+	} else {
+		log.Printf("forwarded %d bytes to %s", len(data), addr)
+	}
+}
+
 func init() {
 	rootCmd.AddCommand(testYggdrasilCmd)
 }