From 789664e8c32730b0a98874dccfbb167fd9fe20b0 Mon Sep 17 00:00:00 2001 From: Pavel Vorontsov Date: Mon, 17 Feb 2020 19:27:17 +0300 Subject: add support GetTable method for vrfs #2235 --- internal/pkg/table/table.go | 39 +++++++++++++++++++++++++++++++++------ 1 file changed, 33 insertions(+), 6 deletions(-) (limited to 'internal/pkg/table/table.go') diff --git a/internal/pkg/table/table.go b/internal/pkg/table/table.go index bfcf2059..2e0dc94a 100644 --- a/internal/pkg/table/table.go +++ b/internal/pkg/table/table.go @@ -434,14 +434,41 @@ type TableInfo struct { NumAccepted int } -func (t *Table) Info(id string, as uint32) *TableInfo { +type TableInfoOptions struct { + ID string + AS uint32 + VRF *Vrf +} + +func (t *Table) Info(option ...TableInfoOptions) *TableInfo { var numD, numP int + + id := GLOBAL_RIB_NAME + var vrf *Vrf + as := uint32(0) + + for _, o := range option { + if o.ID != "" { + id = o.ID + } + if o.VRF != nil { + vrf = o.VRF + } + as = o.AS + } + for _, d := range t.destinations { - n := 0 - if id == GLOBAL_RIB_NAME { - n = len(d.knownPathList) - } else { - n = len(d.GetKnownPathList(id, as)) + paths := d.GetKnownPathList(id, as) + n := len(paths) + + if vrf != nil { + ps := make([]*Path, 0, len(paths)) + for _, p := range paths { + if CanImportToVrf(vrf, p) { + ps = append(ps, p.ToLocal()) + } + } + n = len(ps) } if n != 0 { numD++ -- cgit v1.2.3