support customize notification receiver
Signed-off-by: wanjunlei <wanjunlei@yunify.com>
This commit is contained in:
wanjunlei
302
pkg/models/resources/v1alpha3/notification/notification.go
Normal file
302
pkg/models/resources/v1alpha3/notification/notification.go
Normal file
@@ -0,0 +1,302 @@
|
||||
/*
|
||||
Copyright 2019 The KubeSphere Authors.
|
||||
|
||||
Licensed 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 notification
|
||||
|
||||
import (
|
||||
"k8s.io/apimachinery/pkg/api/meta"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"kubesphere.io/kubesphere/pkg/api"
|
||||
"kubesphere.io/kubesphere/pkg/apiserver/query"
|
||||
ksinformers "kubesphere.io/kubesphere/pkg/client/informers/externalversions"
|
||||
"kubesphere.io/kubesphere/pkg/models/resources/v1alpha3"
|
||||
)
|
||||
|
||||
type dingtalkConfigGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewDingTalkConfigGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &dingtalkConfigGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *dingtalkConfigGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().DingTalkConfigs().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *dingtalkConfigGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().DingTalkConfigs().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type dingtalkReceiverGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewDingTalkReceiverGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &dingtalkReceiverGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *dingtalkReceiverGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().DingTalkReceivers().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *dingtalkReceiverGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().DingTalkReceivers().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type emailConfigGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewEmailConfigGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &emailConfigGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *emailConfigGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().EmailConfigs().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *emailConfigGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().EmailConfigs().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type emailReceiverGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewEmailReceiverGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &emailReceiverGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *emailReceiverGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().EmailReceivers().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *emailReceiverGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().EmailReceivers().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type slackConfigGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewSlackConfigGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &slackConfigGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *slackConfigGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().SlackConfigs().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *slackConfigGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().SlackConfigs().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type slackReceiverGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewSlackReceiverGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &slackReceiverGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *slackReceiverGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().SlackReceivers().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *slackReceiverGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().SlackReceivers().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type webhookConfigGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewWebhookConfigGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &webhookConfigGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *webhookConfigGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().WebhookConfigs().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *webhookConfigGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().WebhookConfigs().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type webhookReceiverGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewWebhookReceiverGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &webhookReceiverGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *webhookReceiverGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().WebhookReceivers().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *webhookReceiverGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().WebhookReceivers().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type wechatConfigGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewWechatConfigGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &wechatConfigGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *wechatConfigGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().WechatConfigs().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *wechatConfigGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().WechatConfigs().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
type wechatReceiverGetter struct {
|
||||
ksInformer ksinformers.SharedInformerFactory
|
||||
}
|
||||
|
||||
func NewWechatReceiverGetter(informer ksinformers.SharedInformerFactory) v1alpha3.Interface {
|
||||
return &wechatReceiverGetter{ksInformer: informer}
|
||||
}
|
||||
|
||||
func (g *wechatReceiverGetter) Get(_, name string) (runtime.Object, error) {
|
||||
return g.ksInformer.Notification().V2().WechatReceivers().Lister().Get(name)
|
||||
}
|
||||
|
||||
func (g *wechatReceiverGetter) List(_ string, query *query.Query) (*api.ListResult, error) {
|
||||
objs, err := g.ksInformer.Notification().V2().WechatReceivers().Lister().List(query.Selector())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result []runtime.Object
|
||||
for _, obj := range objs {
|
||||
result = append(result, obj)
|
||||
}
|
||||
return v1alpha3.DefaultList(result, query, compare, filter), nil
|
||||
}
|
||||
|
||||
func compare(left runtime.Object, right runtime.Object, field query.Field) bool {
|
||||
|
||||
leftObj, err := meta.Accessor(left)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
rightObj, err := meta.Accessor(right)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
return v1alpha3.DefaultObjectMetaCompare(meta.AsPartialObjectMetadata(leftObj).ObjectMeta,
|
||||
meta.AsPartialObjectMetadata(rightObj).ObjectMeta, field)
|
||||
}
|
||||
|
||||
func filter(object runtime.Object, filter query.Filter) bool {
|
||||
|
||||
accessor, err := meta.Accessor(object)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
return v1alpha3.DefaultObjectMetaFilter(meta.AsPartialObjectMetadata(accessor).ObjectMeta, filter)
|
||||
}
|
||||
204
pkg/models/resources/v1alpha3/notification/notification_test.go
Normal file
204
pkg/models/resources/v1alpha3/notification/notification_test.go
Normal file
@@ -0,0 +1,204 @@
|
||||
/*
|
||||
Copyright 2019 The KubeSphere Authors.
|
||||
|
||||
Licensed 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 notification
|
||||
|
||||
import (
|
||||
"github.com/google/go-cmp/cmp"
|
||||
"github.com/google/uuid"
|
||||
"k8s.io/apimachinery/pkg/api/meta"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"k8s.io/client-go/tools/cache"
|
||||
"kubesphere.io/kubesphere/pkg/api"
|
||||
v2 "kubesphere.io/kubesphere/pkg/apis/notification/v2"
|
||||
"kubesphere.io/kubesphere/pkg/apiserver/query"
|
||||
"kubesphere.io/kubesphere/pkg/client/clientset/versioned/fake"
|
||||
ksinformers "kubesphere.io/kubesphere/pkg/client/informers/externalversions"
|
||||
"kubesphere.io/kubesphere/pkg/models/resources/v1alpha3"
|
||||
"kubesphere.io/kubesphere/pkg/server/errors"
|
||||
"math/rand"
|
||||
"sort"
|
||||
"testing"
|
||||
)
|
||||
|
||||
const (
|
||||
Prefix = "foo"
|
||||
LengthMin = 3
|
||||
LengthMax = 10
|
||||
)
|
||||
|
||||
func TestListObjects(t *testing.T) {
|
||||
tests := []struct {
|
||||
description string
|
||||
key string
|
||||
}{
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralDingTalkConfig,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralDingTalkReceiver,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralEmailConfig,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralEmailReceiver,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralSlackConfig,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralSlackReceiver,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralWebhookConfig,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralWebhookReceiver,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralWechatConfig,
|
||||
},
|
||||
{
|
||||
"test name filter",
|
||||
v2.ResourcesPluralWechatReceiver,
|
||||
},
|
||||
}
|
||||
|
||||
q := &query.Query{
|
||||
Pagination: &query.Pagination{
|
||||
Limit: 10,
|
||||
Offset: 0,
|
||||
},
|
||||
SortBy: query.FieldName,
|
||||
Ascending: true,
|
||||
Filters: map[query.Field]query.Value{query.FieldName: query.Value(Prefix)},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
|
||||
getter, objects, err := prepare(test.key)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
got, err := getter.List("", q)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
expected := &api.ListResult{
|
||||
Items: objects,
|
||||
TotalItems: len(objects),
|
||||
}
|
||||
|
||||
if diff := cmp.Diff(got, expected); diff != "" {
|
||||
t.Errorf("[%s] %T differ (-got, +want): %s", test.description, expected, diff)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func prepare(key string) (v1alpha3.Interface, []interface{}, error) {
|
||||
client := fake.NewSimpleClientset()
|
||||
informer := ksinformers.NewSharedInformerFactory(client, 0)
|
||||
|
||||
var obj runtime.Object
|
||||
var indexer cache.Indexer
|
||||
var getter func(informer ksinformers.SharedInformerFactory) v1alpha3.Interface
|
||||
switch key {
|
||||
case v2.ResourcesPluralDingTalkConfig:
|
||||
indexer = informer.Notification().V2().DingTalkConfigs().Informer().GetIndexer()
|
||||
getter = NewDingTalkConfigGetter
|
||||
obj = &v2.DingTalkConfig{}
|
||||
case v2.ResourcesPluralDingTalkReceiver:
|
||||
indexer = informer.Notification().V2().DingTalkReceivers().Informer().GetIndexer()
|
||||
getter = NewDingTalkReceiverGetter
|
||||
obj = &v2.DingTalkReceiver{}
|
||||
case v2.ResourcesPluralEmailConfig:
|
||||
indexer = informer.Notification().V2().EmailConfigs().Informer().GetIndexer()
|
||||
getter = NewEmailConfigGetter
|
||||
obj = &v2.EmailConfig{}
|
||||
case v2.ResourcesPluralEmailReceiver:
|
||||
indexer = informer.Notification().V2().EmailReceivers().Informer().GetIndexer()
|
||||
getter = NewEmailReceiverGetter
|
||||
obj = &v2.EmailReceiver{}
|
||||
case v2.ResourcesPluralSlackConfig:
|
||||
indexer = informer.Notification().V2().SlackConfigs().Informer().GetIndexer()
|
||||
getter = NewSlackConfigGetter
|
||||
obj = &v2.SlackConfig{}
|
||||
case v2.ResourcesPluralSlackReceiver:
|
||||
indexer = informer.Notification().V2().SlackReceivers().Informer().GetIndexer()
|
||||
getter = NewSlackReceiverGetter
|
||||
obj = &v2.SlackReceiver{}
|
||||
case v2.ResourcesPluralWebhookConfig:
|
||||
indexer = informer.Notification().V2().WebhookConfigs().Informer().GetIndexer()
|
||||
getter = NewWebhookConfigGetter
|
||||
obj = &v2.WebhookConfig{}
|
||||
case v2.ResourcesPluralWebhookReceiver:
|
||||
indexer = informer.Notification().V2().WebhookReceivers().Informer().GetIndexer()
|
||||
getter = NewWebhookReceiverGetter
|
||||
obj = &v2.WebhookReceiver{}
|
||||
case v2.ResourcesPluralWechatConfig:
|
||||
indexer = informer.Notification().V2().WechatConfigs().Informer().GetIndexer()
|
||||
getter = NewWechatConfigGetter
|
||||
obj = &v2.WechatConfig{}
|
||||
case v2.ResourcesPluralWechatReceiver:
|
||||
indexer = informer.Notification().V2().WechatReceivers().Informer().GetIndexer()
|
||||
getter = NewWechatReceiverGetter
|
||||
obj = &v2.WechatReceiver{}
|
||||
default:
|
||||
return nil, nil, errors.New("unkonwed type %s", key)
|
||||
}
|
||||
|
||||
num := rand.Intn(LengthMax)
|
||||
if num < LengthMin {
|
||||
num = LengthMin
|
||||
}
|
||||
|
||||
var suffix []string
|
||||
for i := 0; i < num; i++ {
|
||||
s := uuid.New().String()
|
||||
suffix = append(suffix, s)
|
||||
}
|
||||
sort.Strings(suffix)
|
||||
|
||||
var objects []interface{}
|
||||
for i := 0; i < num; i++ {
|
||||
val := obj.DeepCopyObject()
|
||||
accessor, err := meta.Accessor(val)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
accessor.SetName(Prefix + "-" + suffix[i])
|
||||
err = indexer.Add(accessor)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
objects = append(objects, val)
|
||||
}
|
||||
|
||||
return getter(informer), objects, nil
|
||||
}
|
||||
Reference in New Issue
Block a user