Add reliabilitylevel_string autogen code

base/node.go

@@ -2,25 +2,32 @@ package base
 
 import (
 	"go.uber.org/zap"
+	"strconv"
 	"time"
 )
 
 type NodeID int
+
+func (n NodeID) String() string {
+	return strconv.Itoa(int(n))
+}
+
 type PacketWithSender struct {
 	Packet *Packet
 	Sender *Node
 }
 
 type Node struct {
-	ID              NodeID
-	Log             *zap.SugaredLogger
-	MessageQueue    chan *PacketWithSender
-	Subscribers     []*Node
-	PacketHistory   map[PacketUUID]*PacketLog
-	Reliability     ReliabilityLevel
-	CpuScore        int
-	packagesSent    int
-	packagesDropped int
+	ID               NodeID
+	Log              *zap.SugaredLogger
+	MessageQueue     chan *PacketWithSender
+	Subscribers      []*Node
+	PacketHistory    map[PacketUUID]*PacketLog
+	Reliability      ReliabilityLevel
+	CpuScore         int
+	PackagesReceived int
+	packagesSent     int
+	packagesDropped  int
 }
 
 type PacketLog struct {
@@ -60,6 +67,7 @@ func (n *Node) RecvPacket(callerNode *Node, p *Packet) {
 	n.MessageQueue <- &PacketWithSender{
 		Sender: callerNode,
 		Packet: p}
+	n.PackagesReceived++
 }
 
 func (n *Node) AlreadyReceived(id PacketUUID) bool {

base/reliability.go

@@ -4,6 +4,7 @@ import (
 	"math/rand"
 )
 
+//go:generate stringer -type=ReliabilityLevel
 type ReliabilityLevel int
 
 const (

cmd/run.go

@@ -4,9 +4,8 @@ Copyright © 2024 NAME HERE <EMAIL ADDRESS>
 package cmd
 
 import (
-	"github.com/vposloncec/go-ssip/orchestration"
-
 	"github.com/spf13/cobra"
+	"github.com/vposloncec/go-ssip/orchestration"
 )
 
 // runCmd represents the run command

export/node.go

@@ -3,7 +3,6 @@ package export
 import (
 	"bytes"
 	"encoding/csv"
-	"fmt"
 	"github.com/vposloncec/go-ssip/base"
 	"strconv"
 )
@@ -28,7 +27,7 @@ func nodeToRow(node *base.Node) []string {
 	row := make([]string, len(nodeHeaders))
 	row[0] = node.ID.String()
 	row[1] = strconv.Itoa(node.PackagesReceived)
-	row[2] = fmt.Sprint(node.Reliability)
+	row[2] = node.Reliability.String()
 
 	return row
 }

orchestration/orchestrator.go

@@ -1,10 +1,13 @@
 package orchestration
 
 import (
+	"fmt"
 	"github.com/spf13/viper"
 	"github.com/vposloncec/go-ssip/base"
+	"github.com/vposloncec/go-ssip/export"
 	"go.uber.org/zap"
 	"go.uber.org/zap/zapcore"
+	"os"
 	"time"
 )
 
@@ -18,9 +21,12 @@ func StartFromInput(nodeNum int, connections []base.ConnectionPair) {
 	graph.Nodes[0].SendPacket(p)
 	time.Sleep(3 * time.Second)
 	graph.CalcPacketReach(p.ID)
+	export.EdgesToCSV(graph.Connections).WriteTo(os.Stdout)
+	fmt.Println("==============================")
+	export.NodesToCSV(graph.Nodes).WriteTo(os.Stdout)
 }
 
-func StartRandom(nodeNum int, connections int) {
+func StartRandom(nodeNum int, connections int) *base.Graph {
 	log := getLogger()
 
 	maxId := nodeNum - 1
@@ -33,6 +39,8 @@ func StartRandom(nodeNum int, connections int) {
 	graph.Nodes[0].SendPacket(p)
 	time.Sleep(3 * time.Second)
 	graph.CalcPacketReach(p.ID)
+
+	return graph
 }
 
 func getLogger() *zap.SugaredLogger {