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

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.Service{dep})
	}
}
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)
	}
}

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)
}

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
}

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
}

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()
}

[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