Add version to zebra-config in gobgp.yang.

2 files changed, 11 insertions, 0 deletions

diff --git a/config/bgp_configs.go b/config/bgp_configs.go
index 9ea05258..17ae937d 100644
--- a/config/bgp_configs.go
+++ b/config/bgp_configs.go
@@ -987,6 +987,9 @@ func (lhs *ZebraState) Equal(rhs *ZebraState) bool {
 			return false
 		}
 	}
+	if lhs.Version != rhs.Version {
+		return false
+	}
 	return true
 }
 
@@ -1021,6 +1024,9 @@ func (lhs *ZebraConfig) Equal(rhs *ZebraConfig) bool {
 			return false
 		}
 	}
+	if lhs.Version != rhs.Version {
+		return false
+	}
 	return true
 }
 
diff --git a/tools/pyang_plugins/gobgp.yang b/tools/pyang_plugins/gobgp.yang
index 44a36299..a4f8e335 100644
--- a/tools/pyang_plugins/gobgp.yang
+++ b/tools/pyang_plugins/gobgp.yang
@@ -909,6 +909,11 @@ module gobgp {
         base ptypes:install-protocol-type;
       }
     }
+    leaf version {
+      type uint8;
+      description
+        "Configure version of zebra protocol.  Default is 2. Supported up to 3.";
+    }
   }
 
   grouping zebra-set {