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

歡迎存取我的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介面的實現類

原始碼下載

• 上述完整原始碼可在GitHub下載到，地址和連結資訊如下表所示(https://github.com/zq2599/blog_demos)：

• 這個git專案中有多個資料夾，本篇的原始碼在tutorials/client-go-tutorials資料夾下，如下圖紅框所示：
  

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

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