脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → Golang → golang k8s上传yaml

golang简易实现 k8s 的yaml上传并应用示例方案

 更新时间：2023年07月31日 09:36:36   作者：锅包肉  
这篇文章主要为大家介绍了golang简易实现 k8s 的yaml上传并应用示例方案,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪

plan A：调用 os模块

调用 os模块 system的 kubectl apply -f xx.yaml

package main
import (
    "bytes"
    "io/ioutil"
    "net/http"
    "os"
    "os/exec"
    _ "path/filepath"
    "github.com/gin-gonic/gin"
    _ "gopkg.in/yaml.v3"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)
type YAML struct {
    Content string `gorm:"type:text"`
}
func main() {
    // 初始化数据库
    dsn := "root:xixu@tcp(10.23.18x.x1:3306)/texxt?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(any("Failed to connect database"))
    }
    db.AutoMigrate(&YAML{})
    // 初始化 Gin 路由
    r := gin.Default()
    r.LoadHTMLGlob("./temp/*")
    r.GET("/upload", func(c *gin.Context) {
        c.HTML(http.StatusOK, "upload.html", gin.H{
            "mess": "mess",
        })
    })
    r.POST("/upload", func(c *gin.Context) {
        // 从 HTTP 请求中读取 YAML 文件内容
        file, err := c.FormFile("yaml")
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        f, err := file.Open()
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        defer f.Close()
        content, err := ioutil.ReadAll(f)
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        // 将 YAML 文件内容存储到数据库中
        yaml := YAML{Content: string(content)}
        db.Create(&yaml)
        // 应用 YAML 文件内容到 Kubernetes 集群中
        cmd := exec.Command("kubectl", "apply", "-f", "-")
        cmd.Stdout = os.Stdout
        cmd.Stderr = os.Stderr
        cmd.Stdin = bytes.NewBufferString(yaml.Content)
        err = cmd.Run()
        if err != nil {
            c.String(http.StatusInternalServerError, "Internal server error")
            return
        }
        c.String(http.StatusOK, "YAML file uploaded successfully")
    })
    r.Run(":8080")
}
<!DOCTYPE html>
<html lang="zh-CN">
<head>
    <title>上传文件示例</title>
</head>
<body>
<form action="/upload" method="post" enctype="multipart/form-data">
    <input type="file" name="yaml">
    <input type="submit" value="上传">
</form>
</body>
</html>

plan B：dynamic.NewForConfig 函数创建 Kubernetes 客户端对象

我们使用 dynamic.NewForConfig 函数创建 Kubernetes 客户端对象。然后，我们读取 YAML 文件并将其转换为 Unstructured 对象。最后，我们使用 dynamic.Resource 函数来获取对应于 YAML 文件中资源类型的 Kubernetes 资源客户端，并使用 Create 或 Update 方法将资源对象应用到 Kubernetes 中

package main
import (
    "bytes"
    "context"
    "fmt"
    "io/ioutil"
    "k8s.io/apimachinery/pkg/api/meta"
    metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
    "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
    "k8s.io/apimachinery/pkg/runtime"
    "k8s.io/apimachinery/pkg/runtime/serializer/yaml"
    yamlutil "k8s.io/apimachinery/pkg/util/yaml"
    "k8s.io/client-go/dynamic"
    "k8s.io/client-go/kubernetes"
    "k8s.io/client-go/restmapper"
    "k8s.io/client-go/tools/clientcmd"
    "log"
    "net/http"
    "os/user"
    _ "path/filepath"
    "github.com/gin-gonic/gin"
    _ "gopkg.in/yaml.v3"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)
type YAML1 struct {
    Content string `gorm:"type:text"`
}
func main() {
    // 初始化数据库
    dsn := "root:xiaoxhu@tcp(10.2x.1x.1x:330x)/txst?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(any("Failed to connect database"))
    }
    db.AutoMigrate(&YAML1{})
    // 初始化 Gin 路由
    r := gin.Default()
    r.LoadHTMLGlob("./temp/*")
    r.GET("/upload", func(c *gin.Context) {
        c.HTML(http.StatusOK, "upload.html", gin.H{
            "mess": "mess",
        })
    })
    r.POST("/upload", func(c *gin.Context) {
        // 从 HTTP 请求中读取 YAML 文件内容
        file, err := c.FormFile("yaml")
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        f, err := file.Open()
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        defer f.Close()
        content, err := ioutil.ReadAll(f)
        if err != nil {
            c.String(http.StatusBadRequest, "Bad request")
            return
        }
        // 将 YAML 文件内容存储到数据库中
        yaml := YAML1{Content: string(content)}
        db.Create(&yaml)
        // 应用 YAML 文件内容到 Kubernetes 集群中
        err = ApplyYaml(content)
        if err != nil {
            c.String(http.StatusInternalServerError, "Internal server error")
            return
        }
        c.String(http.StatusOK, "YAML file uploaded successfully")
    })
    r.Run(":8080")
}
func ApplyYaml(filebytes []byte) error{
    home := GetHomePath()
    nameSpace := "default"
    k8sConfig, err := clientcmd.BuildConfigFromFlags("", fmt.Sprintf("%s/.kube/config", home))  // 使用 kubectl 默认配置 ~/.kube/config
    if err != nil {
        fmt.Printf("%v",err)
        return err
    }
    // 创建一个k8s客户端
    clientSet, err := kubernetes.NewForConfig(k8sConfig)
    if err != nil {
        fmt.Printf("%v",err)
        return err
    }
    dd, err := dynamic.NewForConfig(k8sConfig)
    if err != nil {
        log.Fatal(err)
    }
    decoder := yamlutil.NewYAMLOrJSONDecoder(bytes.NewReader(filebytes), 100)
    for {
        var rawObj runtime.RawExtension
        if err = decoder.Decode(&rawObj); err != nil {
            break
        }
        obj, gvk, err := yaml.NewDecodingSerializer(unstructured.UnstructuredJSONScheme).Decode(rawObj.Raw, nil, nil)
        unstructuredMap, err := runtime.DefaultUnstructuredConverter.ToUnstructured(obj)
        if err != nil {
            log.Fatal(err)
        }
        unstructuredObj := &unstructured.Unstructured{ Object: unstructuredMap}
        gr, err := restmapper.GetAPIGroupResources(clientSet.Discovery())
        if err != nil {
            log.Fatal(err)
        }
        mapper := restmapper.NewDiscoveryRESTMapper(gr)
        mapping, err := mapper.RESTMapping(gvk.GroupKind(), gvk.Version)
        if err != nil {
            log.Fatal(err)
        }
        var dri dynamic.ResourceInterface
        if mapping.Scope.Name() == meta.RESTScopeNameNamespace {
            if unstructuredObj.GetNamespace() == "" {
                unstructuredObj.SetNamespace(nameSpace)
            }
            dri = dd.Resource(mapping.Resource).Namespace(unstructuredObj.GetNamespace())
        } else {
            dri = dd.Resource(mapping.Resource)
        }
        obj2, err := dri.Create(context.Background(), unstructuredObj, metav1.CreateOptions{ })
        if  err != nil {
            log.Fatal(err)
        }
        fmt.Printf("%s/%s created", obj2.GetKind(), obj2.GetName())
    }
    return nil
}
func GetHomePath() string {
    u , err := user.Current()
    if err == nil {
        return u.HomeDir
    }
    return ""
}

以上就是golang简易实现 k8s 的yaml上传并应用示例方案的详细内容，更多关于golang k8s上传yaml的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • GORM不定参数的用法最佳实践

    GORM不定参数的用法最佳实践

    这篇文章主要为大家介绍了GORM不定参数的用法最佳实践,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-12-12
  • golang框架gin的日志处理和zap lumberjack日志使用方式

    golang框架gin的日志处理和zap lumberjack日志使用方式

    这篇文章主要介绍了golang框架gin的日志处理和zap lumberjack日志使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2024-01-01
  • Go gRPC环境安装教程示例详解

    Go gRPC环境安装教程示例详解

    这篇文章主要为大家介绍了Go gRPC环境安装的教程示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-06-06
  • GoFrame框架garray并发安全数组使用开箱体验

    GoFrame框架garray并发安全数组使用开箱体验

    这篇文章主要介绍了GoFrame框架garray并发安全数组使用开箱体验,有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2022-06-06
  • golang gorm错误处理事务以及日志用法示例

    golang gorm错误处理事务以及日志用法示例

    这篇文章主要为大家介绍了golang gorm错误处理事务以及日志用法示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步早日升职加薪
    2022-04-04
  • 在 Golang 中使用 Cobra 创建 CLI 应用

    在 Golang 中使用 Cobra 创建 CLI 应用

    这篇文章主要介绍了在 Golang 中使用 Cobra 创建 CLI 应用,来看下 Cobra 的使用，这里我们使用的 go1.13.3 版本，使用 Go Modules 来进行包管理,需要的朋友可以参考下
    2022-01-01
  • golang 跳出多重循环的高级break用法说明

    golang 跳出多重循环的高级break用法说明

    这篇文章主要介绍了golang 跳出多重循环的高级break用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • 基于Go语言构建RESTful API服务

    基于Go语言构建RESTful API服务

    在实际开发项目中，你编写的服务可以被其他服务使用，这样就组成了微服务的架构；也可以被前端调用，这样就可以前后端分离。那么，本文主要介绍什么是 RESTful API，以及 Go 语言是如何玩转 RESTful API 的
    2021-07-07
  • golang逐行读取文件的操作

    golang逐行读取文件的操作

    这篇文章主要介绍了golang逐行读取文件的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • 浅谈golang package中init方法的多处定义及运行顺序问题

    浅谈golang package中init方法的多处定义及运行顺序问题

    这篇文章主要介绍了浅谈golang package中init方法的多处定义及运行顺序问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2021-05-05

最新评论