Move packages to proejct/pkg to follow go project folder structure convention (#1190)

2019-05-31 16:28:55 +08:00
parent 1c5fd92ad6
commit a0e9a39511
196 changed files with 1672 additions and 1716 deletions
@@ -0,0 +1,160 @@
+/*
+Copyright 2016 The Fission 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 router
+
+import (
+	"os"
+
+	"go.uber.org/zap"
+	"k8s.io/api/extensions/v1beta1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/util/intstr"
+	"k8s.io/client-go/kubernetes"
+
+	fv1 "github.com/fission/fission/pkg/apis/fission.io/v1"
+)
+
+var podNamespace string
+
+func init() {
+	podNamespace = os.Getenv("POD_NAMESPACE")
+	if podNamespace == "" {
+		podNamespace = "fission"
+	}
+}
+
+func createIngress(logger *zap.Logger, trigger *fv1.HTTPTrigger, kubeClient *kubernetes.Clientset) {
+
+	if !trigger.Spec.CreateIngress {
+		logger.Info("skipping creation of ingress for trigger", zap.String("trigger", trigger.Metadata.Name))
+		return
+	}
+
+	_, err := kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Get(trigger.Metadata.Name, v1.GetOptions{})
+	if err == nil {
+		logger.Info("ingress for trigger exists already", zap.String("trigger", trigger.Metadata.Name))
+		return
+	}
+
+	ing := &v1beta1.Ingress{
+		ObjectMeta: metav1.ObjectMeta{
+			Labels: getDeployLabels(trigger),
+			Name:   trigger.Metadata.Name,
+			// The Ingress NS MUST be same as Router NS, check long discussion:
+			// https://github.com/kubernetes/kubernetes/issues/17088
+			// We need to revisit this in future, once Kubernetes supports cross namespace ingress
+			Namespace: podNamespace,
+		},
+		Spec: v1beta1.IngressSpec{
+			Rules: []v1beta1.IngressRule{
+				{
+					Host: trigger.Spec.Host,
+					IngressRuleValue: v1beta1.IngressRuleValue{
+						HTTP: &v1beta1.HTTPIngressRuleValue{
+							Paths: []v1beta1.HTTPIngressPath{
+								{
+									Backend: v1beta1.IngressBackend{
+										ServiceName: "router",
+										ServicePort: intstr.IntOrString{
+											Type:   intstr.Int,
+											IntVal: 80,
+										},
+									},
+									Path: trigger.Spec.RelativeURL,
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	_, err = kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Create(ing)
+	if err != nil {
+		logger.Error("failed to create ingress", zap.Error(err))
+		return
+	}
+	logger.Info("created ingress successfully for trigger", zap.String("trigger", trigger.Metadata.Name))
+}
+
+func getDeployLabels(trigger *fv1.HTTPTrigger) map[string]string {
+	return map[string]string{
+		"triggerName":      trigger.Metadata.Name,
+		"functionName":     trigger.Spec.FunctionReference.Name,
+		"triggerNamespace": trigger.Metadata.Namespace,
+	}
+}
+
+func deleteIngress(logger *zap.Logger, trigger *fv1.HTTPTrigger, kubeClient *kubernetes.Clientset) {
+	if !trigger.Spec.CreateIngress {
+		return
+	}
+
+	ingress, err := kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Get(trigger.Metadata.Name, v1.GetOptions{})
+	if err != nil {
+		logger.Error("failed to get ingress when deleting trigger", zap.Error(err), zap.String("trigger", trigger.Metadata.Name))
+	}
+
+	err = kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Delete(ingress.Name, &v1.DeleteOptions{})
+
+	if err != nil {
+		logger.Error("failed to delete ingress for trigger",
+			zap.Error(err),
+			zap.Any("ingress", ingress),
+			zap.String("trigger", trigger.Metadata.Name))
+	}
+
+}
+
+func updateIngress(logger *zap.Logger, oldT *fv1.HTTPTrigger, newT *fv1.HTTPTrigger, kubeClient *kubernetes.Clientset) {
+
+	if oldT.Spec.CreateIngress == false && newT.Spec.CreateIngress == true {
+		createIngress(logger, newT, kubeClient)
+		return
+	}
+
+	if newT.Spec.CreateIngress == false && oldT.Spec.CreateIngress == true {
+		deleteIngress(logger, oldT, kubeClient)
+		return
+	}
+
+	if newT.Spec.Host != oldT.Spec.Host || newT.Spec.RelativeURL != oldT.Spec.RelativeURL {
+		logger.Info("updating ingress for trigger", zap.String("trigger", oldT.Metadata.Name))
+		ingress, err := kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Get(oldT.Metadata.Name, v1.GetOptions{})
+		if err != nil {
+			logger.Error("failed to get ingress when updating trigger",
+				zap.Error(err),
+				zap.String("trigger", oldT.Metadata.Name))
+		}
+
+		if newT.Spec.Host != oldT.Spec.Host {
+			ingress.Spec.Rules[0].Host = newT.Spec.Host
+		}
+
+		if newT.Spec.RelativeURL != oldT.Spec.RelativeURL {
+			ingress.Spec.Rules[0].HTTP.Paths[0].Path = newT.Spec.RelativeURL
+		}
+
+		_, err = kubeClient.ExtensionsV1beta1().Ingresses(podNamespace).Update(ingress)
+		if err != nil {
+			logger.Error("failed to update ingress for trigger", zap.String("trigger", oldT.Metadata.Name))
+		}
+	}
+
+}