mirror of https://github.com/k3s-io/k3s
241 lines
7.3 KiB
Go
241 lines
7.3 KiB
Go
/*
|
|
Copyright 2014 The Kubernetes Authors.
|
|
|
|
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 versioned
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/util/intstr"
|
|
"k8s.io/kubernetes/pkg/kubectl/generate"
|
|
)
|
|
|
|
// The only difference between ServiceGeneratorV1 and V2 is that the service port is named "default" in V1, while it is left unnamed in V2.
|
|
type ServiceGeneratorV1 struct{}
|
|
|
|
func (ServiceGeneratorV1) ParamNames() []generate.GeneratorParam {
|
|
return paramNames()
|
|
}
|
|
|
|
func (ServiceGeneratorV1) Generate(params map[string]interface{}) (runtime.Object, error) {
|
|
params["port-name"] = "default"
|
|
return generateService(params)
|
|
}
|
|
|
|
type ServiceGeneratorV2 struct{}
|
|
|
|
func (ServiceGeneratorV2) ParamNames() []generate.GeneratorParam {
|
|
return paramNames()
|
|
}
|
|
|
|
func (ServiceGeneratorV2) Generate(params map[string]interface{}) (runtime.Object, error) {
|
|
return generateService(params)
|
|
}
|
|
|
|
func paramNames() []generate.GeneratorParam {
|
|
return []generate.GeneratorParam{
|
|
{Name: "default-name", Required: true},
|
|
{Name: "name", Required: false},
|
|
{Name: "selector", Required: true},
|
|
// port will be used if a user specifies --port OR the exposed object
|
|
// has one port
|
|
{Name: "port", Required: false},
|
|
// ports will be used iff a user doesn't specify --port AND the
|
|
// exposed object has multiple ports
|
|
{Name: "ports", Required: false},
|
|
{Name: "labels", Required: false},
|
|
{Name: "external-ip", Required: false},
|
|
{Name: "load-balancer-ip", Required: false},
|
|
{Name: "type", Required: false},
|
|
{Name: "protocol", Required: false},
|
|
// protocols will be used to keep port-protocol mapping derived from
|
|
// exposed object
|
|
{Name: "protocols", Required: false},
|
|
{Name: "container-port", Required: false}, // alias of target-port
|
|
{Name: "target-port", Required: false},
|
|
{Name: "port-name", Required: false},
|
|
{Name: "session-affinity", Required: false},
|
|
{Name: "cluster-ip", Required: false},
|
|
}
|
|
}
|
|
|
|
func generateService(genericParams map[string]interface{}) (runtime.Object, error) {
|
|
params := map[string]string{}
|
|
for key, value := range genericParams {
|
|
strVal, isString := value.(string)
|
|
if !isString {
|
|
return nil, fmt.Errorf("expected string, saw %v for '%s'", value, key)
|
|
}
|
|
params[key] = strVal
|
|
}
|
|
selectorString, found := params["selector"]
|
|
if !found || len(selectorString) == 0 {
|
|
return nil, fmt.Errorf("'selector' is a required parameter")
|
|
}
|
|
selector, err := generate.ParseLabels(selectorString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
labelsString, found := params["labels"]
|
|
var labels map[string]string
|
|
if found && len(labelsString) > 0 {
|
|
labels, err = generate.ParseLabels(labelsString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
name, found := params["name"]
|
|
if !found || len(name) == 0 {
|
|
name, found = params["default-name"]
|
|
if !found || len(name) == 0 {
|
|
return nil, fmt.Errorf("'name' is a required parameter")
|
|
}
|
|
}
|
|
|
|
isHeadlessService := params["cluster-ip"] == "None"
|
|
|
|
ports := []v1.ServicePort{}
|
|
servicePortName, found := params["port-name"]
|
|
if !found {
|
|
// Leave the port unnamed.
|
|
servicePortName = ""
|
|
}
|
|
|
|
protocolsString, found := params["protocols"]
|
|
var portProtocolMap map[string]string
|
|
if found && len(protocolsString) > 0 {
|
|
portProtocolMap, err = generate.ParseProtocols(protocolsString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
// ports takes precedence over port since it will be
|
|
// specified only when the user hasn't specified a port
|
|
// via --port and the exposed object has multiple ports.
|
|
var portString string
|
|
if portString, found = params["ports"]; !found {
|
|
portString, found = params["port"]
|
|
if !found && !isHeadlessService {
|
|
return nil, fmt.Errorf("'ports' or 'port' is a required parameter")
|
|
}
|
|
}
|
|
|
|
if portString != "" {
|
|
portStringSlice := strings.Split(portString, ",")
|
|
for i, stillPortString := range portStringSlice {
|
|
port, err := strconv.Atoi(stillPortString)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
name := servicePortName
|
|
// If we are going to assign multiple ports to a service, we need to
|
|
// generate a different name for each one.
|
|
if len(portStringSlice) > 1 {
|
|
name = fmt.Sprintf("port-%d", i+1)
|
|
}
|
|
protocol := params["protocol"]
|
|
|
|
switch {
|
|
case len(protocol) == 0 && len(portProtocolMap) == 0:
|
|
// Default to TCP, what the flag was doing previously.
|
|
protocol = "TCP"
|
|
case len(protocol) > 0 && len(portProtocolMap) > 0:
|
|
// User has specified the --protocol while exposing a multiprotocol resource
|
|
// We should stomp multiple protocols with the one specified ie. do nothing
|
|
case len(protocol) == 0 && len(portProtocolMap) > 0:
|
|
// no --protocol and we expose a multiprotocol resource
|
|
protocol = "TCP" // have the default so we can stay sane
|
|
if exposeProtocol, found := portProtocolMap[stillPortString]; found {
|
|
protocol = exposeProtocol
|
|
}
|
|
}
|
|
ports = append(ports, v1.ServicePort{
|
|
Name: name,
|
|
Port: int32(port),
|
|
Protocol: v1.Protocol(protocol),
|
|
})
|
|
}
|
|
}
|
|
|
|
service := v1.Service{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: name,
|
|
Labels: labels,
|
|
},
|
|
Spec: v1.ServiceSpec{
|
|
Selector: selector,
|
|
Ports: ports,
|
|
},
|
|
}
|
|
targetPortString := params["target-port"]
|
|
if len(targetPortString) == 0 {
|
|
targetPortString = params["container-port"]
|
|
}
|
|
if len(targetPortString) > 0 {
|
|
var targetPort intstr.IntOrString
|
|
if portNum, err := strconv.Atoi(targetPortString); err != nil {
|
|
targetPort = intstr.FromString(targetPortString)
|
|
} else {
|
|
targetPort = intstr.FromInt(portNum)
|
|
}
|
|
// Use the same target-port for every port
|
|
for i := range service.Spec.Ports {
|
|
service.Spec.Ports[i].TargetPort = targetPort
|
|
}
|
|
} else {
|
|
// If --target-port or --container-port haven't been specified, this
|
|
// should be the same as Port
|
|
for i := range service.Spec.Ports {
|
|
port := service.Spec.Ports[i].Port
|
|
service.Spec.Ports[i].TargetPort = intstr.FromInt(int(port))
|
|
}
|
|
}
|
|
if len(params["external-ip"]) > 0 {
|
|
service.Spec.ExternalIPs = []string{params["external-ip"]}
|
|
}
|
|
if len(params["type"]) != 0 {
|
|
service.Spec.Type = v1.ServiceType(params["type"])
|
|
}
|
|
if service.Spec.Type == v1.ServiceTypeLoadBalancer {
|
|
service.Spec.LoadBalancerIP = params["load-balancer-ip"]
|
|
}
|
|
if len(params["session-affinity"]) != 0 {
|
|
switch v1.ServiceAffinity(params["session-affinity"]) {
|
|
case v1.ServiceAffinityNone:
|
|
service.Spec.SessionAffinity = v1.ServiceAffinityNone
|
|
case v1.ServiceAffinityClientIP:
|
|
service.Spec.SessionAffinity = v1.ServiceAffinityClientIP
|
|
default:
|
|
return nil, fmt.Errorf("unknown session affinity: %s", params["session-affinity"])
|
|
}
|
|
}
|
|
if len(params["cluster-ip"]) != 0 {
|
|
if params["cluster-ip"] == "None" {
|
|
service.Spec.ClusterIP = v1.ClusterIPNone
|
|
} else {
|
|
service.Spec.ClusterIP = params["cluster-ip"]
|
|
}
|
|
}
|
|
return &service, nil
|
|
}
|