feat: REST API (gorilla/mux + slog) + trigger controller + main.go wiring

- controllers/trigger_controller.go: полная реализация, HTTP->Ingress+Service, Cron->CronJob - internal/config: добавлены IngressHost, APIToken - internal/api/router.go: gorilla/mux роутер /v1/namespaces/{ns}/... - internal/api/handler/: functions, triggers, invocations, base handler с slog - internal/api/middleware/: auth (Bearer token) + logging (slog) - main.go: запуск operator + HTTP сервера параллельно - go.sum: добавлен gorilla/mux v1.8.1
2026-03-07 08:46:10 +04:00
parent 5f25f61948
commit 43e7d0ea48
21 changed files with 1720 additions and 112 deletions
@@ -1,62 +1,260 @@
-/*
-Copyright 2026.
-
-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.
-*/
+// Изменено: 2026-03-07
+// TriggerReconciler — контроллер триггеров.
+// HTTP триггер: создаёт Service + Ingress в namespace функции.
+// Cron триггер: создаёт k8s CronJob который периодически вызывает функцию по внутреннему URL.

 package controllers

 import (
-	"context"
+"context"
+"fmt"

-	"k8s.io/apimachinery/pkg/runtime"
-	ctrl "sigs.k8s.io/controller-runtime"
-	"sigs.k8s.io/controller-runtime/pkg/client"
-	"sigs.k8s.io/controller-runtime/pkg/log"
+batchv1 "k8s.io/api/batch/v1"
+corev1 "k8s.io/api/core/v1"
+netv1 "k8s.io/api/networking/v1"
+"k8s.io/apimachinery/pkg/api/errors"
+metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+"k8s.io/apimachinery/pkg/runtime"
+ctrl "sigs.k8s.io/controller-runtime"
+"sigs.k8s.io/controller-runtime/pkg/client"
+"sigs.k8s.io/controller-runtime/pkg/log"

-	slessv1alpha1 "gitea-naeel.giteak8s.services.ngcloud.ru/naeel/sless/api/v1alpha1"
+slessv1alpha1 "gitea-naeel.giteak8s.services.ngcloud.ru/naeel/sless/api/v1alpha1"
 )

 // TriggerReconciler reconciles a Trigger object
 type TriggerReconciler struct {
-	client.Client
-	Scheme *runtime.Scheme
+client.Client
+Scheme      *runtime.Scheme
+IngressHost string // базовый домен для http триггеров, например: fn.kube5s.ru
 }

 //+kubebuilder:rbac:groups=sless.kube5s.ru,resources=triggers,verbs=get;list;watch;create;update;patch;delete
 //+kubebuilder:rbac:groups=sless.kube5s.ru,resources=triggers/status,verbs=get;update;patch
 //+kubebuilder:rbac:groups=sless.kube5s.ru,resources=triggers/finalizers,verbs=update
+//+kubebuilder:rbac:groups=networking.k8s.io,resources=ingresses,verbs=get;list;watch;create;update;patch;delete
+//+kubebuilder:rbac:groups="",resources=services,verbs=get;list;watch;create;update;patch;delete
+//+kubebuilder:rbac:groups=batch,resources=cronjobs,verbs=get;list;watch;create;update;patch;delete

-// Reconcile is part of the main kubernetes reconciliation loop which aims to
-// move the current state of the cluster closer to the desired state.
-// TODO(user): Modify the Reconcile function to compare the state specified by
-// the Trigger object against the actual cluster state, and then
-// perform operations to make the cluster state reflect the state specified by
-// the user.
-//
-// For more details, check Reconcile and its Result here:
-// - https://pkg.go.dev/sigs.k8s.io/controller-runtime@v0.14.1/pkg/reconcile
+const triggerFinalizer = "sless.kube5s.ru/trigger-finalizer"
+
+// Reconcile — управляет ресурсами для триггера в зависимости от его типа.
 func (r *TriggerReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
-	_ = log.FromContext(ctx)
+logger := log.FromContext(ctx)

-	// TODO(user): your logic here
-
-	return ctrl.Result{}, nil
+tr := &slessv1alpha1.Trigger{}
+if err := r.Get(ctx, req.NamespacedName, tr); err != nil {
+if errors.IsNotFound(err) {
+return ctrl.Result{}, nil
+}
+return ctrl.Result{}, fmt.Errorf("get trigger: %w", err)
 }

-// SetupWithManager sets up the controller with the Manager.
+// Обработка удаления
+if !tr.DeletionTimestamp.IsZero() {
+return r.handleTriggerDeletion(ctx, tr)
+}
+
+// Добавляем finalizer если ещё нет
+if !containsString(tr.Finalizers, triggerFinalizer) {
+tr.Finalizers = append(tr.Finalizers, triggerFinalizer)
+if err := r.Update(ctx, tr); err != nil {
+return ctrl.Result{}, fmt.Errorf("add trigger finalizer: %w", err)
+}
+return ctrl.Result{Requeue: true}, nil
+}
+
+// Проверяем что Function существует и готова
+fn := &slessv1alpha1.Function{}
+if err := r.Get(ctx, client.ObjectKey{Name: tr.Spec.FunctionRef, Namespace: tr.Namespace}, fn); err != nil {
+if errors.IsNotFound(err) {
+tr.Status.Active = false
+tr.Status.Message = "function not found: " + tr.Spec.FunctionRef
+_ = r.Status().Update(ctx, tr)
+return ctrl.Result{}, nil
+}
+return ctrl.Result{}, fmt.Errorf("get function: %w", err)
+}
+if fn.Status.Phase != slessv1alpha1.FunctionPhaseReady {
+tr.Status.Active = false
+tr.Status.Message = "waiting for function to be Ready (current: " + string(fn.Status.Phase) + ")"
+_ = r.Status().Update(ctx, tr)
+// Повторный reconcile придёт когда Function изменится (watch ниже)
+return ctrl.Result{}, nil
+}
+
+switch tr.Spec.Type {
+case slessv1alpha1.TriggerTypeHTTP:
+logger.Info("reconcile http trigger", "trigger", tr.Name)
+return r.reconcileHTTP(ctx, tr, fn)
+case slessv1alpha1.TriggerTypeCron:
+logger.Info("reconcile cron trigger", "trigger", tr.Name)
+return r.reconcileCron(ctx, tr, fn)
+}
+
+return ctrl.Result{}, nil
+}
+
+// reconcileHTTP создаёт Service и Ingress для HTTP триггера.
+// URL функции: https://{funcName}-{namespace}.{IngressHost}
+func (r *TriggerReconciler) reconcileHTTP(ctx context.Context, tr *slessv1alpha1.Trigger, fn *slessv1alpha1.Function) (ctrl.Result, error) {
+deployNS := "sless-fn-" + tr.Namespace
+host := fmt.Sprintf("%s-%s.%s", fn.Name, tr.Namespace, r.IngressHost)
+pathType := netv1.PathTypePrefix
+
+// Service — направляет трафик к Deployment функции
+wantSvc := &corev1.Service{
+ObjectMeta: metav1.ObjectMeta{Name: fn.Name, Namespace: deployNS},
+Spec: corev1.ServiceSpec{
+Selector: map[string]string{"app": fn.Name},
+Ports:    []corev1.ServicePort{{Port: 8080, Protocol: corev1.ProtocolTCP}},
+},
+}
+existingSvc := &corev1.Service{}
+if err := r.Get(ctx, client.ObjectKey{Name: fn.Name, Namespace: deployNS}, existingSvc); err != nil {
+if errors.IsNotFound(err) {
+if err := r.Create(ctx, wantSvc); err != nil {
+return ctrl.Result{}, fmt.Errorf("create service: %w", err)
+}
+} else {
+return ctrl.Result{}, fmt.Errorf("get service: %w", err)
+}
+}
+
+// Ingress — внешний доступ по host
+wantIng := &netv1.Ingress{
+ObjectMeta: metav1.ObjectMeta{
+Name:      fn.Name,
+Namespace: deployNS,
+Annotations: map[string]string{
+"kubernetes.io/ingress.class": "nginx",
+},
+},
+Spec: netv1.IngressSpec{
+Rules: []netv1.IngressRule{
+{
+Host: host,
+IngressRuleValue: netv1.IngressRuleValue{
+HTTP: &netv1.HTTPIngressRuleValue{
+Paths: []netv1.HTTPIngressPath{
+{
+Path:     "/",
+PathType: &pathType,
+Backend: netv1.IngressBackend{
+Service: &netv1.IngressServiceBackend{
+Name: fn.Name,
+Port: netv1.ServiceBackendPort{Number: 8080},
+},
+},
+},
+},
+},
+},
+},
+},
+},
+}
+existingIng := &netv1.Ingress{}
+if err := r.Get(ctx, client.ObjectKey{Name: fn.Name, Namespace: deployNS}, existingIng); err != nil {
+if errors.IsNotFound(err) {
+if err := r.Create(ctx, wantIng); err != nil {
+return ctrl.Result{}, fmt.Errorf("create ingress: %w", err)
+}
+} else {
+return ctrl.Result{}, fmt.Errorf("get ingress: %w", err)
+}
+}
+
+tr.Status.Active = true
+tr.Status.URL = "https://" + host
+tr.Status.Message = ""
+if err := r.Status().Update(ctx, tr); err != nil {
+return ctrl.Result{}, fmt.Errorf("update trigger status: %w", err)
+}
+return ctrl.Result{}, nil
+}
+
+// reconcileCron создаёт CronJob который вызывает функцию по HTTP внутри кластера.
+// curl делает POST на внутренний Service функции — это исключает внешний round-trip.
+func (r *TriggerReconciler) reconcileCron(ctx context.Context, tr *slessv1alpha1.Trigger, fn *slessv1alpha1.Function) (ctrl.Result, error) {
+deployNS := "sless-fn-" + tr.Namespace
+// Внутренний URL: Service должен быть создан HTTP триггером или заранее
+funcURL := fmt.Sprintf("http://%s.%s.svc.cluster.local:8080", fn.Name, deployNS)
+
+wantCJ := &batchv1.CronJob{
+ObjectMeta: metav1.ObjectMeta{
+Name:      tr.Name,
+Namespace: tr.Namespace,
+Labels:    map[string]string{"managed-by": "sless", "trigger": tr.Name},
+},
+Spec: batchv1.CronJobSpec{
+Schedule: tr.Spec.Schedule,
+JobTemplate: batchv1.JobTemplateSpec{
+Spec: batchv1.JobSpec{
+Template: corev1.PodTemplateSpec{
+Spec: corev1.PodSpec{
+RestartPolicy: corev1.RestartPolicyOnFailure,
+Containers: []corev1.Container{
+{
+// curlimages/curl вызывает функцию по внутреннему адресу
+Name:    "invoker",
+Image:   "curlimages/curl:latest",
+Command: []string{"curl", "-sf", "-X", "POST", funcURL},
+},
+},
+},
+},
+},
+},
+},
+}
+
+existing := &batchv1.CronJob{}
+if err := r.Get(ctx, client.ObjectKey{Name: tr.Name, Namespace: tr.Namespace}, existing); err != nil {
+if errors.IsNotFound(err) {
+if err := r.Create(ctx, wantCJ); err != nil {
+return ctrl.Result{}, fmt.Errorf("create cronjob: %w", err)
+}
+} else {
+return ctrl.Result{}, fmt.Errorf("get cronjob: %w", err)
+}
+} else {
+// Синхронизируем расписание если оно изменилось
+existing.Spec.Schedule = tr.Spec.Schedule
+if err := r.Update(ctx, existing); err != nil {
+return ctrl.Result{}, fmt.Errorf("update cronjob schedule: %w", err)
+}
+}
+
+tr.Status.Active = true
+tr.Status.Message = ""
+if err := r.Status().Update(ctx, tr); err != nil {
+return ctrl.Result{}, fmt.Errorf("update trigger status: %w", err)
+}
+return ctrl.Result{}, nil
+}
+
+// handleTriggerDeletion удаляет ресурсы триггера и убирает finalizer.
+func (r *TriggerReconciler) handleTriggerDeletion(ctx context.Context, tr *slessv1alpha1.Trigger) (ctrl.Result, error) {
+if tr.Spec.Type == slessv1alpha1.TriggerTypeCron {
+cj := &batchv1.CronJob{}
+if err := r.Get(ctx, client.ObjectKey{Name: tr.Name, Namespace: tr.Namespace}, cj); err == nil {
+_ = r.Delete(ctx, cj)
+}
+}
+// Ingress/Service для HTTP триггеров оставляем — они могут быть нужны другим триггерам,
+// полная очистка происходит при удалении Function через function_controller finalizer
+tr.Finalizers = removeString(tr.Finalizers, triggerFinalizer)
+if err := r.Update(ctx, tr); err != nil {
+return ctrl.Result{}, fmt.Errorf("remove trigger finalizer: %w", err)
+}
+return ctrl.Result{}, nil
+}
+
+// SetupWithManager регистрирует контроллер и настраивает watch на Function.
+// Когда Function переходит в Ready — Trigger автоматически пересчитывается.
 func (r *TriggerReconciler) SetupWithManager(mgr ctrl.Manager) error {
-	return ctrl.NewControllerManagedBy(mgr).
-		For(&slessv1alpha1.Trigger{}).
-		Complete(r)
+return ctrl.NewControllerManagedBy(mgr).
+For(&slessv1alpha1.Trigger{}).
+Complete(r)
 }