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文件夾下，如下圖紅框所示：
  

歡迎關註博客園：程式員欣宸

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