pkg/kube/apisix_consumer.go

// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License.  You may obtain a copy of the License at
//
//	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package kube

import (
	"errors"

	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"

	"github.com/apache/apisix-ingress-controller/pkg/config"
	configv2 "github.com/apache/apisix-ingress-controller/pkg/kube/apisix/apis/config/v2"
	listersv2 "github.com/apache/apisix-ingress-controller/pkg/kube/apisix/client/listers/config/v2"
)

// ApisixConsumerLister is an encapsulation for the lister of ApisixConsumer,
// it aims at to be compatible with different ApisixConsumer versions.
type ApisixConsumerLister interface {
	// V2 gets the ApisixConsumer in apisix.apache.org/v2.
	V2(string, string) (ApisixConsumer, error)
}

// ApisixConsumerInformer is an encapsulation for the informer of ApisixConsumer,
// it aims at to be compatible with different ApisixConsumer versions.
type ApisixConsumerInformer interface {
	Run(chan struct{})
}

// ApisixConsumer is an encapsulation for ApisixConsumer resource with different
// versions, for now, they only includes  apisix.apache.org/v2
type ApisixConsumer interface {
	// GroupVersion returns the api group version of the
	// real ApisixConsumer.
	GroupVersion() string
	// V2 returns the ApisixConsumer in apisix.apache.org/v2, the real
	// ApisixConsumer must be in this group version, otherwise will panic.
	V2() *configv2.ApisixConsumer
	// ResourceVersion returns the the resource version field inside
	// the real ApisixConsumer.
	ResourceVersion() string

	metav1.Object
}

// ApisixConsumerEvent contains the ApisixConsumer key (namespace/name)
// and the group version message.
type ApisixConsumerEvent struct {
	Key          string
	OldObject    ApisixConsumer
	GroupVersion string
}

type apisixConsumer struct {
	groupVersion string
	v2           *configv2.ApisixConsumer

	metav1.Object
}

func (ac *apisixConsumer) V2() *configv2.ApisixConsumer {
	if ac.groupVersion != config.ApisixV2 {
		panic("not a apisix.apache.org/v2 Consumer")
	}
	return ac.v2
}

func (ac *apisixConsumer) GroupVersion() string {
	return ac.groupVersion
}

func (ac *apisixConsumer) ResourceVersion() string {
	return ac.V2().ResourceVersion
}

type apisixConsumerLister struct {
	v2Lister listersv2.ApisixConsumerLister
}

func (l *apisixConsumerLister) V2(namespace, name string) (ApisixConsumer, error) {
	ac, err := l.v2Lister.ApisixConsumers(namespace).Get(name)
	if err != nil {
		return nil, err
	}
	return &apisixConsumer{
		groupVersion: config.ApisixV2,
		v2:           ac,
		Object:       ac,
	}, nil
}

// MustNewApisixConsumer creates a kube.ApisixConsumer object according to the
// type of obj.
func MustNewApisixConsumer(obj interface{}) ApisixConsumer {
	switch ac := obj.(type) {
	case *configv2.ApisixConsumer:
		return &apisixConsumer{
			groupVersion: config.ApisixV2,
			v2:           ac,
			Object:       ac,
		}
	default:
		panic("invalid ApisixConsumer type")
	}
}

// NewApisixConsumer creates a kube.ApisixConsumer object according to the
// type of obj. It returns nil and the error reason when the
// type assertion fails.
func NewApisixConsumer(obj interface{}) (ApisixConsumer, error) {
	switch ac := obj.(type) {
	case *configv2.ApisixConsumer:
		return &apisixConsumer{
			groupVersion: config.ApisixV2,
			v2:           ac,
			Object:       ac,
		}, nil
	default:
		return nil, errors.New("invalid ApisixConsumer type")
	}
}

func NewApisixConsumerLister(v2 listersv2.ApisixConsumerLister) ApisixConsumerLister {
	return &apisixConsumerLister{
		v2Lister: v2,
	}
}