Support the vlan tag for client

linkernetworks · Jul 19, 2018 · 42ba7e2 · 42ba7e2
1 parent d7b6858
commit 42ba7e2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/entity/pod.go b/src/entity/pod.go
@@ -19,7 +19,7 @@ type Container struct {
 type PodNetwork struct {
 	Name       string `bson:"name" json:"name"`
 	IfName     string `bson:"ifName" json:"ifName"`
-	VlanTag    int    `bson:"vlanTag" json:"vlanTag"`
+	VlanTag    *int32 `bson:"vlanTag" json:"vlanTag"`
 	IPAddress  string `bson:"ipAddress json:"ipAddress"`
 	Netmask    string `bson:"netmask" json:"netmask"`
 	BridgeName string `bson:"bridgeName" json:"bridgeName"` //its from the entity.Network entity

diff --git a/src/pod/pod.go b/src/pod/pod.go
@@ -3,6 +3,7 @@ package pod
 import (
 	"fmt"
 	"regexp"
+	"strconv"
 
 	"github.com/linkernetworks/mongo"
 	"github.com/linkernetworks/vortex/src/entity"
@@ -106,15 +107,20 @@ func generateNodeLabels(networks []entity.Network) []string {
 	return utils.Intersections(totalNames)
 }
 
-func generateClientCommand(network entity.PodNetwork) []string {
+func generateClientCommand(network entity.PodNetwork) (command []string) {
 	ip := utils.IPToCIDR(network.IPAddress, network.Netmask)
 
-	return []string{
+	command = []string{
 		"-s=unix:///tmp/vortex.sock",
 		"-b=" + network.BridgeName,
 		"-n=" + network.IfName,
 		"-i=" + ip,
 	}
+
+	if network.VlanTag != nil {
+		command = append(command, "-v="+strconv.Itoa(*network.VlanTag))
+	}
+	return
 }
 
 func generateInitContainer(networks []entity.PodNetwork) ([]corev1.Container, error) {