Get BGP-LS information from GoBGP

nttcom · Sep 13, 2022 · c0be074 · c0be074
1 parent 4b5d8a7
commit c0be074
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 4 deletions.
diff --git a/cmd/polad/main.go b/cmd/polad/main.go
@@ -9,15 +9,21 @@ import (
 	"flag"
 	"log"
 	"os"
+	"time"
 
 	"go.uber.org/zap"
 
 	"github.com/nttcom/pola/internal/config"
+	"github.com/nttcom/pola/internal/pkg/gobgp"
 	"github.com/nttcom/pola/internal/pkg/table"
 	"github.com/nttcom/pola/pkg/logger"
 	"github.com/nttcom/pola/pkg/server"
 )
 
+const (
+	TED_UPDATE_INTERVAL = 10 // (min)
+)
+
 type Flags struct {
 	ConfigFile string
 }
@@ -49,6 +55,21 @@ func main() {
 
 	tedElemsChan := make(chan []table.TedElem)
 
+	// Get BGP-LS NLRI from GoBGP
+	if c.Global.Gobgp.Port != "" {
+		go func() {
+			for {
+				tedElems, err := gobgp.GetBgplsNlris(c.Global.Gobgp.Address, c.Global.Gobgp.Port)
+				if err != nil {
+					logger.Panic("Failed session with GoBGP", zap.Error(err))
+				}
+				tedElemsChan <- tedElems
+				time.Sleep(TED_UPDATE_INTERVAL * time.Minute)
+			}
+
+		}()
+	}
+
 	o := new(server.PceOptions)
 	o.PcepAddr = c.Global.Pcep.Address
 	o.PcepPort = c.Global.Pcep.Port

diff --git a/configs/polad.yaml.sample b/configs/polad.yaml.sample
@@ -4,7 +4,10 @@ global:
     port: 4189
   grpc:
     address: "127.0.0.1"
-    port: 50051
+    port: 50052
   log:
     path: "/var/log/pola/"
     name: "polad.log"
+  gobgp:
+    address: "127.0.0.1"
+    port: 50051
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -21,10 +21,16 @@ type Log struct {
 	Name string `yaml:"name"`
 }
 
+type Gobgp struct {
+	Address string `yaml:"address"`
+	Port    string `yaml:"port"`
+}
+
 type Global struct {
-	Pcep Pcep `yaml:"pcep"`
-	Grpc Grpc `yaml:"grpc"`
-	Log  Log  `yaml:"log"`
+	Pcep  Pcep  `yaml:"pcep"`
+	Grpc  Grpc  `yaml:"grpc"`
+	Log   Log   `yaml:"log"`
+	Gobgp Gobgp `yaml:"gobgp"`
 }
 
 type Config struct {