feat(models): add EvpnObject interface

Signed-off-by: Boris Glimcher <[email protected]>

pkg/models/bridge.go

@@ -18,6 +18,9 @@ type Bridge struct {
  VtepIP net.IPNet
 }
 
+// build time check that struct implements interface
+var _ EvpnObject[*pb.LogicalBridge] = (*Bridge)(nil)
+
 // NewBridge creates new SVI object from protobuf message
 func NewBridge(in *pb.LogicalBridge) *Bridge {
  // vtepip := make(net.IP, 4)

pkg/models/interface.go

@@ -0,0 +1,10 @@
+// SPDX-License-Identifier: Apache-2.0
+// Copyright (c) 2022-2023 Dell Inc, or its subsidiaries.
+
+// Package models translates frontend protobuf messages to backend messages
+package models
+
+// EvpnObject is an interface for all domain objects in evpn-gw
+type EvpnObject[T any] interface {
+ ToPb() (T, error)
+}

pkg/models/port.go

@@ -29,6 +29,9 @@ type Port struct {
  LogicalBridgeRefKeys []string
 }
 
+// build time check that struct implements interface
+var _ EvpnObject[*pb.BridgePort] = (*Port)(nil)
+
 // NewPort creates new SVI object from protobuf message
 func NewPort(in *pb.BridgePort) *Port {
  mac := net.HardwareAddr(in.Spec.MacAddress)

pkg/models/svi.go

@@ -21,6 +21,9 @@ type Svi struct {
  RemoteAs uint32
 }
 
+// build time check that struct implements interface
+var _ EvpnObject[*pb.Svi] = (*Svi)(nil)
+
 // NewSvi creates new SVI object from protobuf message
 func NewSvi(in *pb.Svi) *Svi {
  mac := net.HardwareAddr(in.Spec.MacAddress)

pkg/models/vrf.go

@@ -21,6 +21,9 @@ type Vrf struct {
  MacAddress net.HardwareAddr
 }
 
+// build time check that struct implements interface
+var _ EvpnObject[*pb.Vrf] = (*Vrf)(nil)
+
 // NewVrf creates new VRF object from protobuf message
 func NewVrf(in *pb.Vrf) *Vrf {
  mac := net.HardwareAddr(in.Status.Rmac)