client-go實戰之七:準備一個工程管理後續實戰的程式碼

程式設計師欣宸發表於2023-10-16

歡迎訪問我的GitHub

這裡分類和彙總了欣宸的全部原創(含配套原始碼):https://github.com/zq2599/blog_demos

本篇概覽

  • 本文是《client-go實戰》系列的第八篇,主要內容是新建一個golang工程,用於管理程式碼,後面整個系列的程式碼都會儲存在這個工程中

工程結構簡述

  • 此工程打算寫一個簡單的介面,接下來所有實戰的功能程式碼都實現這個介面,在命中輸入引數來決定執行哪個實現類
  • 整個工程結構如下圖所示,後續的實戰中,每個功能類都如綠色箭頭所示,是個獨立的go檔案
    在這裡插入圖片描述

編碼

  • 新建一個名為client-go-tutorials的golang工程
  • 為了便於擴充套件,新建名為action的資料夾,裡面增加action.go檔案,內容如下
package action

import "k8s.io/client-go/kubernetes"

type Action interface {
	DoAction(clientset *kubernetes.Clientset) error
}
  • 後面的實戰都實現這個介面,先把前文的helloworld遷移到本工程中,即list_pod.go
package action

import (
	"context"
	"fmt"

	"k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
)

type ListPod struct{}

func (listPod ListPod) DoAction(clientset *kubernetes.Clientset) error {
	namespace := "kube-system"

	// 查詢pod列表
	pods, err := clientset.CoreV1().Pods(namespace).List(context.TODO(), metav1.ListOptions{})
	if err != nil {
		panic(err.Error())
	}

	nums := len(pods.Items)

	fmt.Printf("There are %d pods in the cluster\n", nums)

	// 如果沒有pod就返回了
	if nums < 1 {
		return nil
	}

	// 遍歷列表中的每個pod
	for index, pod := range pods.Items {
		fmt.Printf("%v. pod name : %v\n", index, pod.Name)

		// 用pod name精確搜尋單個pod
		podObj, err := clientset.CoreV1().Pods(namespace).Get(context.TODO(), pod.Name, metav1.GetOptions{})
		if errors.IsNotFound(err) {
			fmt.Printf("Pod %s in namespace %s not found\n", pod.Name, namespace)
		} else if statusError, isStatus := err.(*errors.StatusError); isStatus {
			fmt.Printf("Error getting pod %s in namespace %s: %v\n",
				pod.Name, namespace, statusError.ErrStatus.Message)
		} else if err != nil {
			panic(err.Error())
		} else {
			fmt.Printf("Found pod %s in namespace %s\n", podObj.Name, namespace)
		}
	}

	return nil
}

程式入口

  • 接下來是main.go,這是程式的入口,程式碼有以下功能:
  1. 載入kubernetes的配置檔案,以便後續的處理
  2. 檢查命令的入參action,根據不同的值執行不同的實現類,在本篇就是list-pod,即獲取pod列表並在控制檯輸出
  • main.go原始碼如下
package main

import (
	"client-go-tutorials/action"
	"flag"
	"fmt"
	"path/filepath"

	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/tools/clientcmd"
	"k8s.io/client-go/util/homedir"
)

func main() {
	var kubeconfig *string
	var actionFlag *string

	// 試圖取到當前賬號的家目錄
	if home := homedir.HomeDir(); home != "" {
		// 如果能取到,就把家目錄下的.kube/config作為預設配置檔案
		kubeconfig = flag.String("kubeconfig", filepath.Join(home, ".kube", "config"), "(optional) absolute path to the kubeconfig file")
	} else {
		// 如果取不到,就沒有預設配置檔案,必須透過kubeconfig引數來指定
		kubeconfig = flag.String("kubeconfig", "", "absolute path to the kubeconfig file")
	}

	actionFlag = flag.String("action", "list-pod", "指定實際操作功能")

	flag.Parse()

	fmt.Println("解析命令完畢,開始載入配置檔案")

	// 載入配置檔案
	config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
	if err != nil {
		panic(err.Error())
	}

	// 用clientset類來執行後續的查詢操作
	clientset, err := kubernetes.NewForConfig(config)
	if err != nil {
		panic(err.Error())
	}

	fmt.Printf("載入配置檔案完畢,即將執行業務 [%v]\n", *actionFlag)

	var actionInterface action.Action

	// 注意,如果有新的功能類實現,就在這裡新增對應的處理
	switch *actionFlag {
	case "list-pod":
		listPod := action.ListPod{}
		actionInterface = &listPod
	case "conflict":
		conflict := action.Confilct{}
		actionInterface = &conflict
	}

	err = actionInterface.DoAction(clientset)
	if err != nil {
		fmt.Printf("err: %v\n", err)
	} else {
		fmt.Println("執行完成")
	}
}

啟動配置

  • 最後,如果您用的是vscode,可以像下面這樣配置launch.json檔案,以便快速執行main.go,以及輸入引數
{
    // Use IntelliSense to learn about possible attributes.
    // Hover to view descriptions of existing attributes.
    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Launch Package",
            "type": "go",
            "request": "launch",
            "mode": "auto",
            "program": "${workspaceFolder}",
            "args": ["-action=list-pod"]
        }
    ]
}
  • 最後,確認.kube/config檔案可以正常使用
  • 按照下圖操作,可以正常執行程式,輸出pod列表資訊
    在這裡插入圖片描述
  • 至此,本篇完成,接下來的實戰都會使用本篇建立的工程,在這裡面新增action介面的實現類

原始碼下載

名稱 連結 備註
專案主頁 https://github.com/zq2599/blog_demos 該專案在GitHub上的主頁
git倉庫地址(https) https://github.com/zq2599/blog_demos.git 該專案原始碼的倉庫地址,https協議
git倉庫地址(ssh) git@github.com:zq2599/blog_demos.git 該專案原始碼的倉庫地址,ssh協議
  • 這個git專案中有多個資料夾,本篇的原始碼在tutorials/client-go-tutorials資料夾下,如下圖紅框所示:
    在這裡插入圖片描述

歡迎關注部落格園:程式設計師欣宸

學習路上,你不孤單,欣宸原創一路相伴...

相關文章