K8s client-go watch pod

一.前言

我們在使用 kubectl 操作 k8s 時,可以在命令中加入 -w來觀察資源變化,比如 kubectl get pod -w 觀察 pod 狀態變化。出了使用控制檯,還可以編寫代碼和 k8s 交互來獲取 pod 變更。

二.代碼實現

k8s client-go:https://github.com/kubernetes/client-go

安裝 package: go get k8s.io/client-go

config, err := clientcmd.BuildConfigFromFlags("", filepath.Join(os.Getenv("HOME"), ".kube", "config"))
if err != nil {
	log.Fatal(err)
}

// 創建 client
clientSet, err := kubernetes.NewForConfig(config)
if err != nil {
	log.Fatal(err)
}

factory := informers.NewSharedInformerFactoryWithOptions(clientSet, 0, informers.WithNamespace("default"))
informer := factory.Core().V1().Pods().Informer()
informer.AddEventHandler(NewEventHandler())

stopper := make(chan struct{}, 2)
go informer.Run(stopper)
log.Println("watch pod started...")

sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)

<-sigs
stopper <- struct{}{}
close(stopper)
log.Println("watch pod stopped...")

處理程序:

type EventHandler struct {
}

func NewEventHandler() *EventHandler {
	return &EventHandler{}
}

func (e *EventHandler) OnAdd(obj interface{}) {
	event := obj.(*corev1.Pod)
	log.Printf("OnAdd: %s", event.ObjectMeta.Name)
}

func (e *EventHandler) OnUpdate(oldObj, newObj interface{}) {
	event := newObj.(*corev1.Pod)
	log.Printf("OnUpdate: %s", event.ObjectMeta.Name)

}

func (e *EventHandler) OnDelete(obj interface{}) {
	event := obj.(*corev1.Pod)
	log.Printf("OnDelete: %s", event.ObjectMeta.Name)

}

三.總結

基於 k8s 聲明式 API 和 informer 機制,不僅僅是 pod,k8s 所有資源都可以被 watch,只需替換 factory.Core().V1().Pods().Informer() 中的 Pods 爲其他資源即可,比如 Events、ConfigMaps 等等。

Demo:https://github.com/stulzq/blog-demo-go/tree/main/k8s/watch-pod

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章