kubevpn/pkg/handler/reset.go

package handler

import (
	"context"
	"strconv"
	"strings"

	log "github.com/sirupsen/logrus"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"sigs.k8s.io/yaml"

	"github.com/wencaiwulue/kubevpn/pkg/config"
	"github.com/wencaiwulue/kubevpn/pkg/controlplane"
)

// Reset
// 1, get all proxy-resources from configmap
// 2, cleanup all containers
func (c *ConnectOptions) Reset(ctx2 context.Context) error {
	cm, err := c.clientset.CoreV1().ConfigMaps(c.Namespace).Get(ctx2, config.ConfigMapPodTrafficManager, metav1.GetOptions{})
	if err != nil {
		return err
	}
	var v = make([]*controlplane.Virtual, 0)
	if str, ok := cm.Data[config.KeyEnvoy]; ok {
		if err = yaml.Unmarshal([]byte(str), &v); err != nil {
			log.Error(err)
			return err
		}
		for _, virtual := range v {
			// deployments.apps.ry-server --> deployments.apps/ry-server
			lastIndex := strings.LastIndex(virtual.Uid, ".")
			uid := virtual.Uid[:lastIndex] + "/" + virtual.Uid[lastIndex+1:]
			for _, rule := range virtual.Rules {
				err = UnPatchContainer(c.factory, c.clientset.CoreV1().ConfigMaps(c.Namespace), c.Namespace, uid, rule.Headers)
				if err != nil {
					log.Error(err)
					continue
				}
			}
		}
	}
	curCount := 0
	if ref := cm.GetAnnotations()[config.AnnoRefCount]; len(ref) > 0 {
		curCount, err = strconv.Atoi(ref)
	}
	updateRefCount(c.clientset.CoreV1().ConfigMaps(c.Namespace), config.ConfigMapPodTrafficManager, 0-curCount)
	cleanupIfRefCountIsZero(c.clientset, c.Namespace, config.ConfigMapPodTrafficManager)
	return nil
}