263 lines
7.3 KiB
Go
263 lines
7.3 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"fmt"
|
|||
|
|
"text/template"
|
|||
|
|
|
|||
|
|
applicationv1 "code.gitea.io/gitea/modules/k8s/api/application/v1"
|
|||
|
|
apps_v1 "k8s.io/api/apps/v1"
|
|||
|
|
core_v1 "k8s.io/api/core/v1"
|
|||
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|||
|
|
"k8s.io/apimachinery/pkg/runtime/serializer"
|
|||
|
|
"k8s.io/apimachinery/pkg/util/intstr"
|
|||
|
|
"k8s.io/client-go/kubernetes/scheme"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
TemplatePath = "modules/k8s/controller/application/templates/"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// parseTemplate 解析 Go Template 模板文件
|
|||
|
|
func parseTemplate(templateName string, app *applicationv1.Application) ([]byte, error) {
|
|||
|
|
tmpl, err := template.
|
|||
|
|
New(templateName + ".yaml").
|
|||
|
|
Funcs(template.FuncMap{"default": DefaultFunc}).
|
|||
|
|
ParseFiles(TemplatePath + templateName + ".yaml")
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to parse template %s: %w", templateName, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
b := new(bytes.Buffer)
|
|||
|
|
err = tmpl.Execute(b, app)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to execute template %s: %w", templateName, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return b.Bytes(), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewDeployment 使用模板创建 Deployment
|
|||
|
|
func NewDeployment(app *applicationv1.Application) (*apps_v1.Deployment, error) {
|
|||
|
|
yamlBytes, err := parseTemplate("deployment", app)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to parse deployment template: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
deployment := &apps_v1.Deployment{}
|
|||
|
|
decoder := serializer.NewCodecFactory(scheme.Scheme).UniversalDecoder()
|
|||
|
|
err = runtime.DecodeInto(decoder, yamlBytes, deployment)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to decode deployment YAML: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 ObjectMeta
|
|||
|
|
deployment.Name = app.Name
|
|||
|
|
deployment.Namespace = app.Namespace
|
|||
|
|
|
|||
|
|
return deployment, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewService 使用模板创建 Service
|
|||
|
|
func NewService(app *applicationv1.Application) (*core_v1.Service, error) {
|
|||
|
|
// 检查是否需要创建 Service
|
|||
|
|
if !shouldCreateService(app) {
|
|||
|
|
return nil, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
yamlBytes, err := parseTemplate("service", app)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to parse service template: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
service := &core_v1.Service{}
|
|||
|
|
decoder := serializer.NewCodecFactory(scheme.Scheme).UniversalDecoder()
|
|||
|
|
err = runtime.DecodeInto(decoder, yamlBytes, service)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to decode service YAML: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 ObjectMeta
|
|||
|
|
service.Name = app.Name + "-svc"
|
|||
|
|
service.Namespace = app.Namespace
|
|||
|
|
|
|||
|
|
// 后处理：根据新的 Service 配置更新
|
|||
|
|
updateServiceWithConfig(service, app)
|
|||
|
|
|
|||
|
|
return service, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewStatefulSet 使用模板创建 StatefulSet
|
|||
|
|
func NewStatefulSet(app *applicationv1.Application) (*apps_v1.StatefulSet, error) {
|
|||
|
|
yamlBytes, err := parseTemplate("statefulset", app)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to parse statefulset template: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
statefulSet := &apps_v1.StatefulSet{}
|
|||
|
|
decoder := serializer.NewCodecFactory(scheme.Scheme).UniversalDecoder()
|
|||
|
|
err = runtime.DecodeInto(decoder, yamlBytes, statefulSet)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to decode statefulset YAML: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 ObjectMeta
|
|||
|
|
statefulSet.Name = app.Name
|
|||
|
|
statefulSet.Namespace = app.Namespace
|
|||
|
|
|
|||
|
|
return statefulSet, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// updateServiceWithConfig 根据新的 Service 配置更新 Service
|
|||
|
|
func updateServiceWithConfig(service *core_v1.Service, app *applicationv1.Application) {
|
|||
|
|
if app.Spec.Service == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
config := app.Spec.Service
|
|||
|
|
|
|||
|
|
// 更新服务类型
|
|||
|
|
if config.Type != "" {
|
|||
|
|
service.Spec.Type = core_v1.ServiceType(config.Type)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加自定义注解
|
|||
|
|
if config.Annotations != nil {
|
|||
|
|
if service.Annotations == nil {
|
|||
|
|
service.Annotations = make(map[string]string)
|
|||
|
|
}
|
|||
|
|
for k, v := range config.Annotations {
|
|||
|
|
service.Annotations[k] = v
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加自定义标签
|
|||
|
|
if config.Labels != nil {
|
|||
|
|
if service.Labels == nil {
|
|||
|
|
service.Labels = make(map[string]string)
|
|||
|
|
}
|
|||
|
|
for k, v := range config.Labels {
|
|||
|
|
service.Labels[k] = v
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置特定配置
|
|||
|
|
switch config.Type {
|
|||
|
|
case "LoadBalancer":
|
|||
|
|
if config.LoadBalancerIP != "" {
|
|||
|
|
service.Spec.LoadBalancerIP = config.LoadBalancerIP
|
|||
|
|
}
|
|||
|
|
if len(config.LoadBalancerSourceRanges) > 0 {
|
|||
|
|
service.Spec.LoadBalancerSourceRanges = config.LoadBalancerSourceRanges
|
|||
|
|
}
|
|||
|
|
case "ExternalName":
|
|||
|
|
if config.ExternalName != "" {
|
|||
|
|
service.Spec.ExternalName = config.ExternalName
|
|||
|
|
}
|
|||
|
|
// ExternalName 类型不需要 selector 和 ports
|
|||
|
|
service.Spec.Selector = nil
|
|||
|
|
service.Spec.Ports = nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置会话亲和性
|
|||
|
|
if config.SessionAffinity != "" {
|
|||
|
|
service.Spec.SessionAffinity = core_v1.ServiceAffinity(config.SessionAffinity)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 更新端口配置（如果有自定义端口配置）
|
|||
|
|
if len(config.Ports) > 0 {
|
|||
|
|
service.Spec.Ports = getServicePorts(app, config)
|
|||
|
|
} else if config.NodePorts != nil {
|
|||
|
|
// 如果只配置了 NodePorts，更新现有端口的 NodePort
|
|||
|
|
updateServiceNodePorts(service, config)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// updateServiceNodePorts 更新服务的 NodePort 配置
|
|||
|
|
func updateServiceNodePorts(service *core_v1.Service, config *applicationv1.ServiceConfig) {
|
|||
|
|
for i, port := range service.Spec.Ports {
|
|||
|
|
if nodePort, exists := config.NodePorts[port.Name]; exists {
|
|||
|
|
service.Spec.Ports[i].NodePort = nodePort
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getServicePorts 获取 Service 端口配置
|
|||
|
|
func getServicePorts(app *applicationv1.Application, config *applicationv1.ServiceConfig) []core_v1.ServicePort {
|
|||
|
|
var servicePorts []core_v1.ServicePort
|
|||
|
|
|
|||
|
|
// 如果配置了自定义端口，使用自定义端口
|
|||
|
|
if len(config.Ports) > 0 {
|
|||
|
|
for _, port := range config.Ports {
|
|||
|
|
servicePort := core_v1.ServicePort{
|
|||
|
|
Name: port.Name,
|
|||
|
|
Port: port.Port,
|
|||
|
|
Protocol: core_v1.Protocol(getPortProtocol(port.Protocol)),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置目标端口
|
|||
|
|
if port.TargetPort != "" {
|
|||
|
|
servicePort.TargetPort = intstr.FromString(port.TargetPort)
|
|||
|
|
} else {
|
|||
|
|
servicePort.TargetPort = intstr.FromInt(int(port.Port))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置 NodePort（仅适用于 NodePort 和 LoadBalancer 类型）
|
|||
|
|
if (config.Type == "NodePort" || config.Type == "LoadBalancer") && port.NodePort > 0 {
|
|||
|
|
servicePort.NodePort = port.NodePort
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
servicePorts = append(servicePorts, servicePort)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 使用模板中的端口配置
|
|||
|
|
for _, port := range app.Spec.Template.Ports {
|
|||
|
|
servicePort := core_v1.ServicePort{
|
|||
|
|
Name: port.Name,
|
|||
|
|
Port: port.Port,
|
|||
|
|
TargetPort: intstr.FromInt(int(port.Port)),
|
|||
|
|
Protocol: core_v1.Protocol(getPortProtocol(port.Protocol)),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果是 NodePort 类型，检查是否有指定的 NodePort
|
|||
|
|
if (config.Type == "NodePort" || config.Type == "LoadBalancer") &&
|
|||
|
|
config.NodePorts != nil {
|
|||
|
|
if nodePort, exists := config.NodePorts[port.Name]; exists {
|
|||
|
|
servicePort.NodePort = nodePort
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
servicePorts = append(servicePorts, servicePort)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return servicePorts
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// shouldCreateService 判断是否需要创建 Service
|
|||
|
|
func shouldCreateService(app *applicationv1.Application) bool {
|
|||
|
|
// 优先使用新的 Service 配置
|
|||
|
|
if app.Spec.Service != nil {
|
|||
|
|
return app.Spec.Service.Enabled
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 向后兼容：使用旧的 expose 配置
|
|||
|
|
return app.Spec.Expose && len(app.Spec.Template.Ports) > 0
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// getPortProtocol 获取端口协议，设置默认值
|
|||
|
|
func getPortProtocol(protocol string) string {
|
|||
|
|
if protocol == "" {
|
|||
|
|
return "TCP"
|
|||
|
|
}
|
|||
|
|
return protocol
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DefaultFunc 函数用于实现默认值
|
|||
|
|
func DefaultFunc(value interface{}, defaultValue interface{}) interface{} {
|
|||
|
|
if value == nil || value == "" {
|
|||
|
|
return defaultValue
|
|||
|
|
}
|
|||
|
|
return value
|
|||
|
|
}
|