Add BufferChannel for fsm incoming/outgoing channels.

cmd/gobgp/neighbor.go

@@ -533,6 +533,41 @@ func showNeighbor(args []string) error {
 			}
 		}
 	}
+
+	incomingChannelType := "Infinite"
+	if p.Conf.IncomingChannel.Type == api.ChannelType_CHANNEL_TYPE_BUFFER {
+		incomingChannelType = "Buffer"
+	}
+	fmt.Printf("  IncomingChannel: %s\n", incomingChannelType)
+	if p.State != nil && p.State.IncomingChannel != nil {
+		if incomingChannelType == "Buffer" {
+			fmt.Printf("    In: %d\n", p.State.IncomingChannel.In)
+			fmt.Printf("    Notifications: %d\n", p.State.IncomingChannel.Notifications)
+			fmt.Printf("    Collected: %d\n", p.State.IncomingChannel.Collected)
+			fmt.Printf("    Rewritten: %d\n", p.State.IncomingChannel.Rewritten)
+			fmt.Printf("    Retries: %d\n", p.State.IncomingChannel.Retries)
+			fmt.Printf("    Out: %d\n", p.State.IncomingChannel.Out)
+		}
+	}
+	fmt.Printf("    Dropped: %d\n", p.State.IncomingChannelDropped)
+
+	outgoingChannelType := "Infinite"
+	if p.Conf.OutgoingChannel.Type == api.ChannelType_CHANNEL_TYPE_BUFFER {
+		outgoingChannelType = "Buffer"
+	}
+	fmt.Printf("  OutgoingChannel: %s\n", outgoingChannelType)
+	if p.State != nil && p.State.OutgoingChannel != nil {
+		if outgoingChannelType == "Buffer" {
+			fmt.Printf("    In: %d\n", p.State.OutgoingChannel.In)
+			fmt.Printf("    Notifications: %d\n", p.State.OutgoingChannel.Notifications)
+			fmt.Printf("    Collected: %d\n", p.State.OutgoingChannel.Collected)
+			fmt.Printf("    Rewritten: %d\n", p.State.OutgoingChannel.Rewritten)
+			fmt.Printf("    Retries: %d\n", p.State.OutgoingChannel.Retries)
+			fmt.Printf("    Out: %d\n", p.State.OutgoingChannel.Out)
+		}
+	}
+	fmt.Printf("    Dropped: %d\n", p.State.OutgoingChannelDropped)
+
 	return nil
 }
 

internal/pkg/channels/buffer_channel.go

@@ -0,0 +1,235 @@
+package channels
+
+import (
+	"sync/atomic"
+	"time"
+
+	api "github.com/osrg/gobgp/v4/api"
+	"github.com/osrg/gobgp/v4/internal/pkg/table"
+	"github.com/osrg/gobgp/v4/pkg/packet/bgp"
+)
+
+const (
+	// default input channel size.
+	//   specifies a buffer size for input channel.
+	defaultInSize = 64
+)
+
+type BufferMessageInterface interface {
+	PathList() []*table.Path
+	SetPathList(pathList []*table.Path)
+}
+
+type pathKey struct {
+	rf             bgp.Family
+	nlri           string
+	pathIdentifier uint32
+}
+
+type BufferChannel struct {
+	input  chan any
+	output chan any
+
+	in            atomic.Uint64
+	notifications atomic.Uint64
+	collected     atomic.Uint64
+	rewritten     atomic.Uint64
+	retries       atomic.Uint64
+	out           atomic.Uint64
+
+	pathIdxs map[pathKey]int
+	last     BufferMessageInterface
+
+	consumerStop chan struct{}
+}
+
+func NewBufferChannel(inSize int) *BufferChannel {
+	bc := &BufferChannel{
+		output:       make(chan any),
+		pathIdxs:     make(map[pathKey]int),
+		consumerStop: make(chan struct{}),
+	}
+
+	if inSize == 0 {
+		// if inSize not set, use default value
+		inSize = defaultInSize
+	}
+	bc.input = make(chan any, inSize)
+
+	go bc.serve()
+	return bc
+}
+
+func (bc *BufferChannel) Push(m any, timeout time.Duration) bool {
+	var timerCh <-chan time.Time
+	if timeout != 0 {
+		timer := time.NewTimer(timeout)
+		defer timer.Stop()
+
+		timerCh = timer.C
+	}
+
+	select {
+	case bc.input <- m:
+	case <-bc.consumerStop:
+	case <-timerCh:
+		// message was dropped
+		return false
+	}
+
+	return true
+}
+
+func (bc *BufferChannel) Out() <-chan any {
+	return bc.output
+}
+
+func (bc *BufferChannel) Stats() *api.ChannelState {
+	return &api.ChannelState{
+		In:            bc.in.Load(),
+		Notifications: bc.notifications.Load(),
+		Collected:     bc.collected.Load(),
+		Rewritten:     bc.rewritten.Load(),
+		Retries:       bc.retries.Load(),
+		Out:           bc.out.Load(),
+	}
+}
+
+func (bc *BufferChannel) Clean() {
+	bc.Close()
+	// drain all remaining items
+	for range bc.output {
+	}
+}
+
+func (bc *BufferChannel) Close() {
+	close(bc.input)
+}
+
+func (bc *BufferChannel) SetConsumerClosed() {
+	close(bc.consumerStop)
+}
+
+func (bc *BufferChannel) serve() {
+	for {
+		var out chan any
+		if bc.last != nil {
+			out = bc.output
+		}
+
+		select {
+		case elem, open := <-bc.input:
+			if !open {
+				close(bc.output)
+				return
+			}
+
+			bc.onInput(elem)
+		case out <- bc.last:
+			bc.out.Add(1)
+
+			clear(bc.pathIdxs)
+			bc.last = nil
+		}
+	}
+}
+
+func (bc *BufferChannel) onInput(anyElem any) {
+	bc.in.Add(1)
+
+	elem, ok := anyElem.(BufferMessageInterface)
+	if !ok || len(elem.PathList()) == 0 {
+		// pass not BufferChannel's element or notification to output with blocking channel
+		bc.notifications.Add(1)
+
+		if bc.last != nil {
+			bc.output <- bc.last
+			bc.out.Add(1)
+
+			clear(bc.pathIdxs)
+			bc.last = nil
+		}
+
+		bc.output <- anyElem
+		bc.out.Add(1)
+		return
+	}
+
+	if bc.last != nil {
+		bc.collect(elem)
+		return
+	}
+
+	select {
+	case bc.output <- elem:
+		// done
+		bc.out.Add(1)
+	default:
+		// try output later
+		bc.retries.Add(1)
+
+		bc.collect(elem)
+	}
+}
+
+func (bc *BufferChannel) collect(elem BufferMessageInterface) {
+	bc.collected.Add(1)
+
+	pathList := elem.PathList()
+
+	if bc.last == nil {
+		// first
+
+		for idx, path := range pathList {
+			if path == nil || path.IsEOR() {
+				continue
+			}
+
+			key := pathKey{
+				rf:             path.GetFamily(),
+				pathIdentifier: path.GetNlri().PathIdentifier(),
+				nlri:           table.TableKey(path.GetNlri()),
+			}
+
+			bc.pathIdxs[key] = idx
+		}
+	} else {
+		// merge
+
+		nextPathsList := bc.last.PathList()
+
+		for _, path := range pathList {
+			if path == nil {
+				continue
+			}
+
+			if path.IsEOR() {
+				nextPathsList = append(nextPathsList, path)
+				continue
+			}
+
+			key := pathKey{
+				rf:             path.GetFamily(),
+				pathIdentifier: path.GetNlri().PathIdentifier(),
+				nlri:           table.TableKey(path.GetNlri()),
+			}
+
+			idx, ok := bc.pathIdxs[key]
+			if !ok {
+				// new path
+
+				bc.pathIdxs[key] = len(nextPathsList)
+				nextPathsList = append(nextPathsList, path)
+			} else {
+				// rewrite path
+				bc.rewritten.Add(1)
+
+				nextPathsList[idx] = path
+			}
+		}
+
+		elem.SetPathList(nextPathsList)
+	}
+
+	bc.last = elem
+}