client-go gin的简单整合八Service-list初步收尾

Posted 2022-06-21 saynaihe

背景：

完成了client-go gin的简单整合七-继续完善，准备整合一下前面的代码（当然了关于上一节的坑还是先留着…没有想好怎么搞：https://blog.csdn.net/saynaihe/article/details/125282951?spm=1001.2014.3001.5501#t9）。这里以service为例也实现一下list watch…

list watch实现service list:

模仿/src/core/deployment_int.go写一个/src/core/service_int.go:

/src/core/service_int.go

package core

import (
	"fmt"
	corev1 "k8s.io/api/core/v1"
	"log"
	"sync"
)

type ServiceMap struct 
	data sync.Map


func (svcmap *ServiceMap) Add(dep *corev1.Service) 
	if list, ok := svcmap.data.Load(dep.Namespace); ok 
		list = append(list.([]*corev1.Service), dep)
		svcmap.data.Store(dep.Namespace, list)
	 else 
		svcmap.data.Store(dep.Namespace, []*corev1.Servicedep)
	

func (svcmap *ServiceMap) Update(dep *corev1.Service) error 
	if list, ok := svcmap.data.Load(dep.Namespace); ok 
		for i, range_dep := range list.([]*corev1.Service) 
			if range_dep.Name == dep.Name 
				list.([]*corev1.Service)[i] = dep
				svcmap.data.Store(dep.Namespace, list)
			
		
		return nil
	
	return fmt.Errorf("Service-%s not found", dep.Name)


// 删除
func (svcmap *ServiceMap) Delete(dep *corev1.Service) 
	if list, ok := svcmap.data.Load(dep.Namespace); ok 
		for i, range_dep := range list.([]*corev1.Service) 
			if range_dep.Name == dep.Name 
				newList := append(list.([]*corev1.Service)[:i], list.([]*corev1.Service)[i+1:]...)
				svcmap.data.Store(dep.Namespace, newList)
				break
			
		
	

func (svcmap *ServiceMap) ListByNS(ns string) ([]*corev1.Service, error) 
	if list, ok := svcmap.data.Load(ns); ok 
		return list.([]*corev1.Service), nil
	
	return nil, fmt.Errorf("record not found")

func (svcmap *ServiceMap) GetService(ns string, name string) (*corev1.Service, error) 
	depList, err := svcmap.ListByNS(ns)
	if err != nil 
		return nil, fmt.Errorf("Service not found")
	
	for _, item := range depList 
		if item.Name == name 
			return item, nil
		
	
	return nil, fmt.Errorf("Service not found")


var svcmap *ServiceMap

func init() 
	svcmap = &ServiceMap


type SvcHandler struct 


func (d *SvcHandler) OnAdd(obj interface) 
	//fmt.Println(obj.(*corev1.Service).Name)
	svcmap.Add(obj.(*corev1.Service))

func (d *SvcHandler) OnUpdate(oldObj, newObj interface) 
	err := svcmap.Update(newObj.(*corev1.Service))
	if err != nil 
		log.Println(err)
	

func (d *SvcHandler) OnDelete(obj interface) 
	if d, ok := obj.(*corev1.Service); ok 
		svcmap.Delete(d)
	

Int

/src/core/deployment_init.go Init中增加ServiceInformer相关配置：

func InitDeployment() 
	factory := informers.NewSharedInformerFactory(lib.K8sClient, 0)
	depinformer := factory.Apps().V1().Deployments()
	depinformer.Informer().AddEventHandler(&DepHandler)
	Podinformer := factory.Core().V1().Pods()
	Podinformer.Informer().AddEventHandler(&PodHandler)
	Rsinformer := factory.Apps().V1().ReplicaSets()
	Rsinformer.Informer().AddEventHandler(&RSHandler)
	eventInformer := factory.Core().V1().Events()
	eventInformer.Informer().AddEventHandler(&EventHandler)
	ServiceInformer := factory.Core().V1().Services()
	ServiceInformer.Informer().AddEventHandler(&SvcHandler)
	factory.Start(wait.NeverStop)

注意：只截取了相关配置，没有复制整个文件！

修改Service.go为list watch模式

/src/service/Service.go not list-watch原代码：

package service

import (
	"context"
	"github.com/gin-gonic/gin"
	. "k8s-demo1/src/lib"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

type Service struct 
	Name       string
	Type       string
	ClusterIp  string
	ExternalIp []string
	Ports      []string
	Select     map[string]string


func ListService(g *gin.Context) 
	ns := g.Query("ns")
	svc, err := K8sClient.CoreV1().Services(ns).List(context.Background(), metav1.ListOptions)
	if err != nil 
		g.Error(err)
		return
	
	ret := make([]*Service, 0)
	for _, item := range svc.Items 
		ret = append(ret, &Service
			Name:       item.Name,
			Type:       string(item.Spec.Type),
			ClusterIp:  item.Spec.ClusterIP,
			ExternalIp: item.Spec.ExternalIPs,
			Select:     item.Spec.Selector,
		)

	
	g.JSON(200, ret)
	return

修改成list watch方式：

package service

import (
	"github.com/gin-gonic/gin"
	"k8s-demo1/src/core"
	corev1 "k8s.io/api/core/v1"
)

type Service struct 
	Name       string
	Type       string
	ClusterIp  string
	ExternalIp []string
	Ports      []corev1.ServicePort
	Select     map[string]string


func ListService(g *gin.Context) 
	ns := g.Query("ns")
	//svc, err := K8sClient.CoreV1().Services(ns).List(context.Background(), metav1.ListOptions)
	svclist, _ := core.SVCMap.ListByNS(ns)
	ret := make([]*Service, 0)
	for _, item := range svclist 
		ret = append(ret, &Service
			Name:       item.Name,
			Type:       string(item.Spec.Type),
			ClusterIp:  item.Spec.ClusterIP,
			ExternalIp: item.Spec.ExternalIPs,
			Ports:      item.Spec.Ports,
			Select:     item.Spec.Selector,
		)

	
	g.JSON(200, ret)
	return

运行main.go

package main

import (
	"github.com/gin-gonic/gin"
	"k8s-demo1/src/core"
	"k8s-demo1/src/service"
)

func main() 
	r := gin.Default()
	r.GET("/", func(context *gin.Context) 
		context.JSON(200, "hello")
	)
	r.GET("/namespaces", service.ListNamespace)
	r.GET("/deployments", service.ListDeployment)
	r.GET("/service", service.ListService)
	r.GET("pods", service.ListallPod)
	core.InitDeployment()
	r.Run()

做一个svc

[zhangpeng@zhangpeng k8s]$ cat nginx1-deployment.yaml 
apiVersion: v1
kind: Service
metadata:
  name: nginx1
  labels:
    app: nginx
spec:
  ports:
  - port: 80
    protocol: TCP
    targetPort: 80
  selector:
    app: nginx
[zhangpeng@zhangpeng k8s]$ kubectl apply -f  nginx1-deployment.yaml 
service/nginx1 created
[zhangpeng@zhangpeng k8s]$ kubectl get svc
NAME         TYPE        CLUSTER-IP     EXTERNAL-IP   PORT(S)   AGE
kubernetes   ClusterIP   10.96.0.1      <none>        443/TCP   47d
nginx1       ClusterIP   10.111.134.5   <none>        80/TCP    48s

注：访问 svc就免了，我这里肯定会有问题的。我的app label都是nginx。所以我不能精确指定到deployment1的（我的初衷），所以尽量是app label去区分deployment！。随便修改一下nginx中index.html就发现他都会调度的几个deployments。所以我前面整deployment的时候也是一种耍流氓了…不过我还是想区分一下搞出来！

总结：

1. list watch 就实现一个service了 statefulset configmap 其他的最后实现吧,curd先实现一下简单的
2. deployment 的坑还是先留着…
3. 下面继续准备实现一下create delete update的方法…