golang簡易實現(xiàn)?k8s?的yaml上傳并應用示例方案
plan A:調(diào)用 os模塊
調(diào)用 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() {
// 初始化數(shù)據(jù)庫
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 文件內(nèi)容
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 文件內(nèi)容存儲到數(shù)據(jù)庫中
yaml := YAML{Content: string(content)}
db.Create(&yaml)
// 應用 YAML 文件內(nèi)容到 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 函數(shù)創(chuàng)建 Kubernetes 客戶端對象
我們使用 dynamic.NewForConfig 函數(shù)創(chuàng)建 Kubernetes 客戶端對象。然后,我們讀取 YAML 文件并將其轉換為 Unstructured 對象。最后,我們使用 dynamic.Resource 函數(shù)來獲取對應于 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() {
// 初始化數(shù)據(jù)庫
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 文件內(nèi)容
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 文件內(nèi)容存儲到數(shù)據(jù)庫中
yaml := YAML1{Content: string(content)}
db.Create(&yaml)
// 應用 YAML 文件內(nèi)容到 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
}
// 創(chuàng)建一個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簡易實現(xiàn) k8s 的yaml上傳并應用示例方案的詳細內(nèi)容,更多關于golang k8s上傳yaml的資料請關注腳本之家其它相關文章!
相關文章
Go類型斷言提取測試接口值動態(tài)類型及靜態(tài)轉換確保類型接口編譯安全
這篇文章主要為大家介紹了Go類型斷言提取測試接口值動態(tài)類型及靜態(tài)轉換確保類型實現(xiàn)特定接口的編譯時安全性詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-10-10
一文帶你了解Golang中select的實現(xiàn)原理
select是go提供的一種跟并發(fā)相關的語法,非常有用。本文將介紹?Go?語言中的?select?的實現(xiàn)原理,包括?select?的結構和常見問題、編譯期間的多種優(yōu)化以及運行時的執(zhí)行過程2023-02-02

