forked from rook/rook
giovanism
4974be4b20
rook: Remove NetworkSelector type alias rook: Add error return value for multus funcs Signed-off-by: giovanism <giovanism@outlook.co.id>
122 lines
3.3 KiB
Go
122 lines
3.3 KiB
Go
/*
|
||
Copyright 2019 The Rook Authors. All rights reserved.
|
||
|
||
Licensed under the Apache License, Version 2.0 (the "License");
|
||
you may not use this file except in compliance with the License.
|
||
You may obtain a copy of the License at
|
||
|
||
http://www.apache.org/licenses/LICENSE-2.0
|
||
|
||
Unless required by applicable law or agreed to in writing, software
|
||
distributed under the License is distributed on an "AS IS" BASIS,
|
||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
See the License for the specific language governing permissions and
|
||
limitations under the License.
|
||
*/
|
||
|
||
package k8sutil
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"strings"
|
||
|
||
rookalpha "github.com/rook/rook/pkg/apis/rook.io/v1alpha2"
|
||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||
)
|
||
|
||
// parseMultusSelector will parse short and JSON form of individual multus
|
||
// network attachment selection annotation. Valid JSON will be unmarshaled and
|
||
// return as is, while invalid JSON will be tried using
|
||
// <namespace>/<name>@<interface> short syntax.
|
||
func parseMultusSelector(selector string) (map[string]string, error) {
|
||
rawMap := make(map[string]string)
|
||
|
||
err := json.Unmarshal([]byte(selector), &rawMap)
|
||
|
||
if err != nil {
|
||
// it can be in short form
|
||
nsEndIndex := strings.IndexAny(selector, "/")
|
||
if nsEndIndex != -1 {
|
||
rawMap["namespace"] = selector[:nsEndIndex]
|
||
}
|
||
|
||
ifStartIndex := strings.LastIndexAny(selector, "@")
|
||
if ifStartIndex != -1 && len(selector)-ifStartIndex > 1 {
|
||
rawMap["interface"] = selector[ifStartIndex+1:]
|
||
}
|
||
|
||
if nsEndIndex != -1 && ifStartIndex != -1 && ifStartIndex-nsEndIndex > 1 {
|
||
rawMap["name"] = selector[nsEndIndex+1 : ifStartIndex]
|
||
} else if nsEndIndex == -1 && ifStartIndex != -1 {
|
||
rawMap["name"] = selector[:ifStartIndex]
|
||
} else if nsEndIndex != -1 && ifStartIndex == -1 {
|
||
rawMap["name"] = selector[nsEndIndex+1:]
|
||
}
|
||
}
|
||
|
||
if name, ok := rawMap["name"]; !ok || name == "" {
|
||
return nil, fmt.Errorf("parseMultusSelector: missing name")
|
||
}
|
||
|
||
return rawMap, nil
|
||
}
|
||
|
||
// GetMultusIfName return a network interface name that multus will assign when
|
||
// connected to the multus network.
|
||
func GetMultusIfName(selector string) (string, error) {
|
||
multusMap, _ := parseMultusSelector(selector)
|
||
var ifName string
|
||
|
||
if name, ok := multusMap["interfaceRequest"]; ok {
|
||
ifName = name
|
||
}
|
||
if name, ok := multusMap["interface"]; ok {
|
||
ifName = name
|
||
}
|
||
|
||
// fail selector without inteface name
|
||
if ifName == "" {
|
||
return "", fmt.Errorf("GetMultusIfname: missing interface")
|
||
}
|
||
|
||
return ifName, nil
|
||
}
|
||
|
||
// ApplyMultus apply multus selector to Pods
|
||
// Multus supports short and json syntax, use only one kind at a time.
|
||
func ApplyMultus(net rookalpha.NetworkSpec, objectMeta *metav1.ObjectMeta) error {
|
||
v := make([]string, 0, 2)
|
||
shortSyntax := false
|
||
jsonSyntax := false
|
||
|
||
for _, ns := range net.Selectors {
|
||
var multusMap map[string]string
|
||
err := json.Unmarshal([]byte(ns), &multusMap)
|
||
|
||
if err == nil {
|
||
jsonSyntax = true
|
||
} else {
|
||
shortSyntax = true
|
||
}
|
||
|
||
v = append(v, string(ns))
|
||
}
|
||
|
||
if shortSyntax && jsonSyntax {
|
||
return fmt.Errorf("ApplyMultus: Can't mix short and JSON form")
|
||
}
|
||
|
||
networks := strings.Join(v, ", ")
|
||
if jsonSyntax {
|
||
networks = "[" + networks + "]"
|
||
}
|
||
|
||
t := rookalpha.Annotations{
|
||
"k8s.v1.cni.cncf.io/networks": networks,
|
||
}
|
||
t.ApplyToObjectMeta(objectMeta)
|
||
|
||
return nil
|
||
}
|