-
Notifications
You must be signed in to change notification settings - Fork 26
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Support retrieving from a remote RIB.
- Loading branch information
Showing
9 changed files
with
447 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| package reconciler | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
|
|
||
| "github.com/openconfig/gribigo/client" | ||
| "github.com/openconfig/gribigo/rib" | ||
| "google.golang.org/protobuf/encoding/prototext" | ||
|
|
||
| spb "github.com/openconfig/gribi/v1/proto/service" | ||
| ) | ||
|
|
||
| // RemoteRIB implements the RIBTarget interface and wraps a remote gRIBI RIB. | ||
| // The contents are accessed via the gRIBI gRPC API. | ||
| type RemoteRIB struct { | ||
| c *client.Client | ||
|
|
||
| defaultName string | ||
| } | ||
|
|
||
| // NewRemoteRIB returns a new remote gRIBI RIB. The context supplied is used to | ||
| // dial the remote gRIBI server at the address 'addr'. the 'defName' argument | ||
| // is used to identify the name of the default network instance on the server. | ||
| func NewRemoteRIB(ctx context.Context, defName, addr string) (*RemoteRIB, error) { | ||
| gc, err := client.New() | ||
| if err != nil { | ||
| return nil, fmt.Errorf("cannot create gRIBI client, %v", err) | ||
| } | ||
|
|
||
| r := &RemoteRIB{ | ||
| c: gc, | ||
| defaultName: defName, | ||
| } | ||
|
|
||
| if err := r.c.Dial(ctx, addr); err != nil { | ||
| return nil, fmt.Errorf("cannot dial remote server, %v", err) | ||
| } | ||
| return r, nil | ||
| } | ||
|
|
||
| // CleanUp closes the remote connection to the gRIBI server. | ||
| func (r *RemoteRIB) CleanUp() { | ||
| r.c.Close() | ||
| } | ||
|
|
||
| // Get retrieves the contents of the remote gRIBI server's RIB and returns it as a | ||
| // gRIBIgo RIB struct. The context is used for a Get RPC call to the remote server. | ||
| func (r *RemoteRIB) Get(ctx context.Context) (*rib.RIB, error) { | ||
| resp, err := r.c.Get(ctx, &spb.GetRequest{ | ||
| NetworkInstance: &spb.GetRequest_All{ | ||
| All: &spb.Empty{}, | ||
| }, | ||
| Aft: spb.AFTType_ALL, | ||
| }) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("cannot get remote RIB, %v", err) | ||
| } | ||
|
|
||
| fmt.Printf("%s\n", prototext.Format(resp)) | ||
|
|
||
| // We always disable the RIB checking function because we want to see entries that have | ||
| // not got valid references so that we can reconcile them. | ||
| remRIB, err := rib.FromGetResponses(r.defaultName, []*spb.GetResponse{resp}, rib.DisableRIBCheckFn()) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("cannot build remote RIB from responses, %v", err) | ||
| } | ||
|
|
||
| return remRIB, nil | ||
| } |
Oops, something went wrong.