jrouter/route.go

package main

import (
	"cmp"
	"fmt"
	"slices"
	"sync"
	"time"

	"github.com/sfiera/multitalk/pkg/ddp"
)

type route struct {
	extended bool
	netStart ddp.Network
	netEnd   ddp.Network
	peer     *peer
	metric   uint8
	last     time.Time
}

var (
	routingTableMu sync.Mutex
	routingTable   = make(map[ddp.Network][]*route)

	allRoutesMu sync.Mutex
	allRoutes   = make(map[*route]struct{})
)

func lookupRoute(network ddp.Network) *route {
	routingTableMu.Lock()
	defer routingTableMu.Unlock()

	rs := routingTable[network]
	if len(rs) == 0 {
		return nil
	}
	return rs[0]
}

func upsertRoutes(extended bool, netStart, netEnd ddp.Network, peer *peer, metric uint8) error {
	if netStart > netEnd {
		return fmt.Errorf("invalid network range [%d, %d]", netStart, netEnd)
	}
	r := &route{
		extended: extended,
		netStart: netStart,
		netEnd:   netEnd,
		peer:     peer,
		metric:   metric,
		last:     time.Now(),
	}

	allRoutesMu.Lock()
	allRoutes[r] = struct{}{}
	allRoutesMu.Unlock()

	routingTableMu.Lock()
	defer routingTableMu.Unlock()
	for n := netStart; n <= netEnd; n++ {
		routingTable[n] = append(routingTable[n], r)
		slices.SortFunc(routingTable[n], func(r, s *route) int {
			return cmp.Compare(r.metric, s.metric)
		})
	}
	return nil
}