diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/gobgp/bmp.go | 17 | ||||
| -rw-r--r-- | cmd/gobgp/common.go | 4 | ||||
| -rw-r--r-- | cmd/gobgpd/main.go | 7 |
3 files changed, 22 insertions, 6 deletions
diff --git a/cmd/gobgp/bmp.go b/cmd/gobgp/bmp.go index 1fbc0ae2..831f6340 100644 --- a/cmd/gobgp/bmp.go +++ b/cmd/gobgp/bmp.go @@ -49,6 +49,13 @@ func modBmpServer(cmdType string, args []string) error { var err error switch cmdType { case cmdAdd: + statisticsTimeout := 0 + if bmpOpts.StatisticsTimeout >= 0 && bmpOpts.StatisticsTimeout <= 65535 { + statisticsTimeout = bmpOpts.StatisticsTimeout + } else { + return fmt.Errorf("invalid statistics-timeout value. it must be in the range 0-65535. default value is 0 and means disabled.") + } + policyType := api.AddBmpRequest_PRE if len(args) > 1 { switch args[1] { @@ -65,9 +72,10 @@ func modBmpServer(cmdType string, args []string) error { } } _, err = client.AddBmp(ctx, &api.AddBmpRequest{ - Address: address, - Port: port, - Type: policyType, + Address: address, + Port: port, + Type: policyType, + StatisticsTimeout: int32(statisticsTimeout), }) case cmdDel: _, err = client.DeleteBmp(ctx, &api.DeleteBmpRequest{ @@ -93,6 +101,9 @@ func newBmpCmd() *cobra.Command { } }, } + if w == cmdAdd { + subcmd.PersistentFlags().IntVarP(&bmpOpts.StatisticsTimeout, "statistics-timeout", "s", 0, "Timeout of statistics report") + } bmpCmd.AddCommand(subcmd) } diff --git a/cmd/gobgp/common.go b/cmd/gobgp/common.go index add3d811..7f582902 100644 --- a/cmd/gobgp/common.go +++ b/cmd/gobgp/common.go @@ -104,6 +104,10 @@ var mrtOpts struct { NextHop net.IP `long:"nexthop" description:"Rewrite nexthop"` } +var bmpOpts struct { + StatisticsTimeout int `short:s long:"statistics-timeout" description:"Interval for Statistics Report"` +} + func formatTimedelta(d int64) string { u := uint64(d) neg := d < 0 diff --git a/cmd/gobgpd/main.go b/cmd/gobgpd/main.go index c880b4b4..5c8f5ac5 100644 --- a/cmd/gobgpd/main.go +++ b/cmd/gobgpd/main.go @@ -216,9 +216,10 @@ func main() { } for _, c := range newConfig.BmpServers { if err := bgpServer.AddBmp(context.Background(), &api.AddBmpRequest{ - Address: c.Config.Address, - Port: c.Config.Port, - Type: api.AddBmpRequest_MonitoringPolicy(c.Config.RouteMonitoringPolicy.ToInt()), + Address: c.Config.Address, + Port: c.Config.Port, + Type: api.AddBmpRequest_MonitoringPolicy(c.Config.RouteMonitoringPolicy.ToInt()), + StatisticsTimeout: int32(c.Config.StatisticsTimeout), }); err != nil { log.Fatalf("failed to set bmp config: %s", err) } |