fix: add create loadb (#2334)

ZhangSetSail · web-flow · commit 5b8d35519d76 · 2025-09-24T18:19:17.000+08:00
Signed-off-by: 张启航 &lt;101104760+ZhangSetSail@users.noreply.github.com&gt;
diff --git a/api/api/api_interface.go b/api/api/api_interface.go
@@ -313,6 +313,7 @@ type GatewayInterface interface {
 	GatewayRouteInterface
 	GatewayServiceInterface
 	GatewayCertInterface
+	GatewayLoadBalancerInterface
 }
 
 // GatewayRouteInterface api gateway route interface
@@ -354,3 +355,10 @@ type GatewayCertInterface interface {
 	UpdateCert(w http.ResponseWriter, r *http.Request)
 	DeleteCert(w http.ResponseWriter, r *http.Request)
 }
+
+// GatewayLoadBalancerInterface api gateway loadbalancer interface
+type GatewayLoadBalancerInterface interface {
+	CreateLoadBalancer(w http.ResponseWriter, r *http.Request)
+	GetLoadBalancer(w http.ResponseWriter, r *http.Request)
+	DeleteLoadBalancer(w http.ResponseWriter, r *http.Request)
+}
diff --git a/api/api_routers/gateway/gateway.go b/api/api_routers/gateway/gateway.go
@@ -26,6 +26,13 @@ func Routes() chi.Router {
 
 	})
 
+	// 创建 LoadBalancer 接口
+	r.Route("/loadbalancer", func(r chi.Router) {
+		r.Post("/", controller.GetManager().CreateLoadBalancer)
+		r.Get("/", controller.GetManager().GetLoadBalancer)
+		r.Delete("/{name}", controller.GetManager().DeleteLoadBalancer)
+	})
+
 	// 关于路由的接口
 	r.Route("/routes/tcp", func(r chi.Router) {
 		r.Get("/domains", controller.GetManager().GetTCPBindDomains)
diff --git a/api/controller/apigateway/api_gateway_loadbalancer.go b/api/controller/apigateway/api_gateway_loadbalancer.go
@@ -0,0 +1,286 @@
+// RAINBOND, Application Management Platform
+// Copyright (C) 2014-2017 Goodrain Co., Ltd.
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version. For any non-GPL usage of Rainbond,
+// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
+// must be obtained first.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+package apigateway
+
+import (
+	"fmt"
+	"net/http"
+	"strings"
+	"time"
+
+	"github.com/go-chi/chi"
+	"github.com/goodrain/rainbond/api/model"
+	"github.com/goodrain/rainbond/api/util/bcode"
+	ctxutil "github.com/goodrain/rainbond/api/util/ctx"
+	dbmodel "github.com/goodrain/rainbond/db/model"
+	"github.com/goodrain/rainbond/pkg/component/k8s"
+	httputil "github.com/goodrain/rainbond/util/http"
+	"github.com/sirupsen/logrus"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/errors"
+	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/util/intstr"
+)
+
+// CreateLoadBalancer 创建LoadBalancer服务
+func (g Struct) CreateLoadBalancer(w http.ResponseWriter, r *http.Request) {
+	tenant := r.Context().Value(ctxutil.ContextKey("tenant")).(*dbmodel.Tenants)
+	serviceID := r.URL.Query().Get("service_id")
+	appID := r.URL.Query().Get("appID")
+	k := k8s.Default().Clientset.CoreV1()
+
+	var createLBReq model.CreateLoadBalancerStruct
+	if !httputil.ValidatorRequestStructAndErrorResponse(r, w, &createLBReq, nil) {
+		return
+	}
+
+	// 验证协议类型
+	protocol := strings.ToUpper(createLBReq.Protocol)
+	if protocol != "TCP" && protocol != "UDP" {
+		httputil.ReturnBcodeError(r, w, bcode.NewBadRequest("protocol must be TCP or UDP"))
+		return
+	}
+
+	// 生成服务名称
+	serviceName := fmt.Sprintf("%s-lb-%d", createLBReq.ServiceName, createLBReq.ServicePort)
+
+	// 检查服务是否已存在
+	_, err := k.Services(tenant.Namespace).Get(r.Context(), serviceName, v1.GetOptions{})
+	if err == nil {
+		httputil.ReturnBcodeError(r, w, bcode.NewBadRequest("LoadBalancer service already exists"))
+		return
+	}
+	if !errors.IsNotFound(err) {
+		logrus.Errorf("check LoadBalancer service error %s", err.Error())
+		httputil.ReturnBcodeError(r, w, bcode.ServerErr)
+		return
+	}
+
+	// 创建服务规格
+	spec := corev1.ServiceSpec{
+		Type: corev1.ServiceTypeLoadBalancer,
+		Ports: []corev1.ServicePort{
+			{
+				Name:       serviceName,
+				Protocol:   corev1.Protocol(protocol),
+				Port:       int32(createLBReq.ServicePort),
+				TargetPort: intstr.FromInt(createLBReq.ServicePort),
+			},
+		},
+	}
+
+	// 直接生成服务选择器
+	spec.Selector = map[string]string{
+		"service_alias": createLBReq.ServiceName,
+	}
+
+	// 创建标签
+	labels := make(map[string]string)
+	labels["creator"] = "Rainbond"
+	labels["loadbalancer"] = "true"
+	if appID != "" {
+		labels["app_id"] = appID
+	}
+	if serviceID != "" {
+		labels["service_id"] = serviceID
+	}
+	labels["service_alias"] = createLBReq.ServiceName
+	labels["port"] = fmt.Sprintf("%d", createLBReq.ServicePort)
+
+	// 创建注解
+	annotations := make(map[string]string)
+	if createLBReq.Annotations != nil {
+		for k, v := range createLBReq.Annotations {
+			annotations[k] = v
+		}
+	}
+
+	// 创建LoadBalancer服务
+	service := &corev1.Service{
+		ObjectMeta: v1.ObjectMeta{
+			Name:        serviceName,
+			Labels:      labels,
+			Annotations: annotations,
+		},
+		Spec: spec,
+	}
+
+	createdService, err := k.Services(tenant.Namespace).Create(r.Context(), service, v1.CreateOptions{})
+	if err != nil {
+		logrus.Errorf("create LoadBalancer service error %s", err.Error())
+		httputil.ReturnBcodeError(r, w, fmt.Errorf("create LoadBalancer service error: %s", err.Error()))
+		return
+	}
+
+	// 构造响应
+	response := &model.LoadBalancerResponse{
+		Name:           createdService.Name,
+		Namespace:      createdService.Namespace,
+		ServiceName:    createLBReq.ServiceName,
+		ServicePort:    createLBReq.ServicePort,
+		Protocol:       protocol,
+		LoadBalancerIP: createdService.Spec.LoadBalancerIP,
+		ExternalIPs:    createdService.Spec.ExternalIPs,
+		Annotations:    createdService.Annotations,
+		Status:         "Creating",
+		CreatedAt:      createdService.CreationTimestamp.Format(time.RFC3339),
+	}
+
+	// 如果LoadBalancer已分配IP，更新状态
+	if len(createdService.Status.LoadBalancer.Ingress) > 0 {
+		response.Status = "Ready"
+		var externalIPs []string
+		for _, ingress := range createdService.Status.LoadBalancer.Ingress {
+			if ingress.IP != "" {
+				externalIPs = append(externalIPs, ingress.IP)
+			}
+			if ingress.Hostname != "" {
+				externalIPs = append(externalIPs, ingress.Hostname)
+			}
+		}
+		response.ExternalIPs = externalIPs
+	}
+
+	logrus.Infof("LoadBalancer service created successfully: %s", serviceName)
+	httputil.ReturnSuccess(r, w, response)
+}
+
+// GetLoadBalancer 获取LoadBalancer服务
+func (g Struct) GetLoadBalancer(w http.ResponseWriter, r *http.Request) {
+	tenant := r.Context().Value(ctxutil.ContextKey("tenant")).(*dbmodel.Tenants)
+	k := k8s.Default().Clientset.CoreV1()
+
+	// 获取查询参数
+	appID := r.URL.Query().Get("appID")
+	serviceName := r.URL.Query().Get("service_name")
+
+	// 构建标签选择器
+	labelSelector := "loadbalancer=true"
+	if appID != "" {
+		labelSelector += ",app_id=" + appID
+	}
+	if serviceName != "" {
+		labelSelector += ",service_alias=" + serviceName
+	}
+
+	// 列出LoadBalancer服务
+	list, err := k.Services(tenant.Namespace).List(r.Context(), v1.ListOptions{
+		LabelSelector: labelSelector,
+	})
+	if err != nil {
+		logrus.Errorf("get LoadBalancer services error %s", err.Error())
+		httputil.ReturnBcodeError(r, w, bcode.ServerErr)
+		return
+	}
+
+	var responses []*model.LoadBalancerResponse
+	for _, service := range list.Items {
+		if service.Spec.Type != corev1.ServiceTypeLoadBalancer {
+			continue
+		}
+
+		response := &model.LoadBalancerResponse{
+			Name:           service.Name,
+			Namespace:      service.Namespace,
+			ServiceName:    service.Labels["service_alias"],
+			ServicePort:    int(service.Spec.Ports[0].Port),
+			Protocol:       string(service.Spec.Ports[0].Protocol),
+			LoadBalancerIP: service.Spec.LoadBalancerIP,
+			ExternalIPs:    service.Spec.ExternalIPs,
+			Annotations:    service.Annotations,
+			Status:         "Creating",
+			CreatedAt:      service.CreationTimestamp.Format(time.RFC3339),
+		}
+
+		// 检查LoadBalancer状态
+		if len(service.Status.LoadBalancer.Ingress) > 0 {
+			response.Status = "Ready"
+			var externalIPs []string
+			for _, ingress := range service.Status.LoadBalancer.Ingress {
+				if ingress.IP != "" {
+					externalIPs = append(externalIPs, ingress.IP)
+				}
+				if ingress.Hostname != "" {
+					externalIPs = append(externalIPs, ingress.Hostname)
+				}
+			}
+			if len(externalIPs) > 0 {
+				response.ExternalIPs = externalIPs
+			}
+		}
+
+		responses = append(responses, response)
+	}
+
+	httputil.ReturnSuccess(r, w, responses)
+}
+
+// DeleteLoadBalancer 删除LoadBalancer服务
+func (g Struct) DeleteLoadBalancer(w http.ResponseWriter, r *http.Request) {
+	tenant := r.Context().Value(ctxutil.ContextKey("tenant")).(*dbmodel.Tenants)
+	serviceName := chi.URLParam(r, "name")
+	if serviceName == "" {
+		serviceName = r.URL.Query().Get("name")
+	}
+
+	if serviceName == "" {
+		httputil.ReturnBcodeError(r, w, bcode.NewBadRequest("service name is required"))
+		return
+	}
+
+	k := k8s.Default().Clientset.CoreV1()
+
+	// 检查服务是否存在且为LoadBalancer类型
+	service, err := k.Services(tenant.Namespace).Get(r.Context(), serviceName, v1.GetOptions{})
+	if err != nil {
+		if errors.IsNotFound(err) {
+			httputil.ReturnBcodeError(r, w, bcode.NotFound)
+			return
+		}
+		logrus.Errorf("get LoadBalancer service error %s", err.Error())
+		httputil.ReturnBcodeError(r, w, bcode.ServerErr)
+		return
+	}
+
+	// 验证是否为LoadBalancer服务
+	if service.Spec.Type != corev1.ServiceTypeLoadBalancer {
+		httputil.ReturnBcodeError(r, w, bcode.NewBadRequest("service is not a LoadBalancer type"))
+		return
+	}
+
+	// 验证是否为Rainbond创建的LoadBalancer
+	if service.Labels["creator"] != "Rainbond" || service.Labels["loadbalancer"] != "true" {
+		httputil.ReturnBcodeError(r, w, bcode.NewBadRequest("service is not a Rainbond LoadBalancer"))
+		return
+	}
+
+	// 删除服务
+	err = k.Services(tenant.Namespace).Delete(r.Context(), serviceName, v1.DeleteOptions{})
+	if err != nil && !errors.IsNotFound(err) {
+		logrus.Errorf("delete LoadBalancer service error %s", err.Error())
+		httputil.ReturnBcodeError(r, w, bcode.ServerErr)
+		return
+	}
+
+	logrus.Infof("LoadBalancer service deleted successfully: %s", serviceName)
+	httputil.ReturnSuccess(r, w, map[string]string{
+		"message": "LoadBalancer service deleted successfully",
+		"name":    serviceName,
+	})
+}
diff --git a/api/model/gateway_model.go b/api/model/gateway_model.go
@@ -385,3 +385,25 @@ type UpdCertificateReq struct {
 	Certificate     string `json:"certificate"`
 	PrivateKey      string `json:"private_key"`
 }
+
+// CreateLoadBalancerStruct 创建LoadBalancer的请求结构体
+type CreateLoadBalancerStruct struct {
+	ServiceName    string            `json:"service_name" validate:"required"` // 后端服务名称
+	ServicePort    int               `json:"service_port" validate:"required"` // 后端服务端口
+	Protocol       string            `json:"protocol" validate:"required"`     // 协议类型 TCP/UDP
+	Annotations    map[string]string `json:"annotations,omitempty"`            // 注解
+}
+
+// LoadBalancerResponse LoadBalancer响应结构体
+type LoadBalancerResponse struct {
+	Name           string            `json:"name"`
+	Namespace      string            `json:"namespace"`
+	ServiceName    string            `json:"service_name"`
+	ServicePort    int               `json:"service_port"`
+	Protocol       string            `json:"protocol"`
+	LoadBalancerIP string            `json:"load_balancer_ip,omitempty"`
+	ExternalIPs    []string          `json:"external_ips,omitempty"`
+	Annotations    map[string]string `json:"annotations,omitempty"`
+	Status         string            `json:"status"`
+	CreatedAt      string            `json:"created_at"`
+}
diff --git a/go.mod b/go.mod
@@ -74,7 +74,6 @@ require (
 	github.com/stretchr/testify v1.10.0
 	github.com/thejerf/suture v3.0.3+incompatible
 	github.com/tidwall/gjson v1.9.3
-	github.com/twinj/uuid v0.0.0-00010101000000-000000000000
 	github.com/urfave/cli v1.22.4
 	golang.org/x/crypto v0.37.0
 	golang.org/x/net v0.39.0
diff --git a/go.sum b/go.sum
@@ -998,7 +998,6 @@ github.com/google/uuid v1.1.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+
 github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.2.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
-github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
