Skip to content

Commit

Permalink
Merge branch 'feat_update_depend' of github.com:tkestack/tke into fea…
Browse files Browse the repository at this point in the history
…t_update_depend

# Conflicts:
#	api/openapi/zz_generated.openapi.go
#	cmd/tke-platform-api/app/config/config.go
#	go.mod
#	pkg/auth/authentication/authenticator/apikey.go
#	pkg/auth/controller/config/config_controller.go
#	pkg/auth/controller/group/deletion/grouped_resources_deleter.go
#	pkg/auth/controller/localidentity/deletion/localidentity_resources_deleter.go
#	pkg/auth/controller/policy/deletion/policied_resources_deleter.go
#	pkg/auth/registry/localidentity/storage/group.go
#	pkg/auth/registry/policy/storage/binding.go
#	pkg/auth/util/user.go
#	pkg/business/controller/project/deletion/projected_resources_deleter.go
#	pkg/business/controller/project/project_controller.go
#	pkg/business/registry/portal/storage/storage.go
#	pkg/platform/controller/addon/prometheus/controller.go
  • Loading branch information
choujimmy committed May 12, 2020
2 parents 54c1f49 + 37c791e commit 8a7babb
Show file tree
Hide file tree
Showing 2 changed files with 51 additions and 25 deletions.
54 changes: 29 additions & 25 deletions pkg/platform/controller/addon/prometheus/controller.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ package prometheus

import (
"context"
"encoding/json"
"fmt"
"math/rand"
"reflect"
Expand All @@ -29,12 +30,12 @@ import (

"github.com/coreos/prometheus-operator/pkg/apis/monitoring"
monitoringv1 "github.com/coreos/prometheus-operator/pkg/apis/monitoring/v1"
promopk8sutil "github.com/coreos/prometheus-operator/pkg/k8sutil"
influxApi "github.com/influxdata/influxdb1-client/v2"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
extensionsv1beta1 "k8s.io/api/extensions/v1beta1"
rbacv1 "k8s.io/api/rbac/v1"
apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
"k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand Down Expand Up @@ -128,14 +129,6 @@ const (
specialLabelValue = "controller"
)

var crdKinds = []monitoringv1.CrdKind{
monitoringv1.DefaultCrdKinds.Alertmanager,
monitoringv1.DefaultCrdKinds.Prometheus,
monitoringv1.DefaultCrdKinds.PodMonitor,
monitoringv1.DefaultCrdKinds.PrometheusRule,
monitoringv1.DefaultCrdKinds.ServiceMonitor,
}

type influxdbClient struct {
address string
client influxApi.Client
Expand Down Expand Up @@ -578,17 +571,17 @@ func (c *Controller) installPrometheus(ctx context.Context, prometheus *v1.Prome
if err != nil {
return err
}
kubeClient, err := util.BuildExternalClientSet(cluster, c.client.PlatformV1())
kubeClient, err := util.BuildExternalClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}

crdClient, err := util.BuildExternalExtensionClientSet(cluster, c.client.PlatformV1())
crdClient, err := util.BuildExternalExtensionClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}

mclient, err := util.BuildExternalMonitoringClientSet(cluster, c.client.PlatformV1())
mclient, err := util.BuildExternalMonitoringClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}
Expand Down Expand Up @@ -624,9 +617,14 @@ func (c *Controller) installPrometheus(ctx context.Context, prometheus *v1.Prome
remoteReads = prometheus.Spec.RemoteAddress.ReadAddr
}

for _, crdKind := range crdKinds {
crd := promopk8sutil.NewCustomResourceDefinition(crdKind, monitoring.GroupName, nil, true)
_, err := crdClient.ApiextensionsV1beta1().CustomResourceDefinitions().Create(crd)
crds := getCRDs()
for _, crd := range crds {
crdObj := apiextensionsv1.CustomResourceDefinition{}
err := json.Unmarshal([]byte(crd), &crdObj)
if err != nil {
return err
}
_, err = crdClient.ApiextensionsV1().CustomResourceDefinitions().Create(ctx, &crdObj, metav1.CreateOptions{})
if err != nil {
return err
}
Expand Down Expand Up @@ -688,7 +686,7 @@ func (c *Controller) installPrometheus(ctx context.Context, prometheus *v1.Prome
prometheus.Status.SubVersion[AlertManagerService] = components.AlertManagerService.Tag

// Secret for prometheus-etcd
credential, err := util.GetClusterCredentialV1(c.client.PlatformV1(), cluster)
credential, err := util.GetClusterCredentialV1(ctx, c.client.PlatformV1(), cluster)
if err != nil {
return err
}
Expand Down Expand Up @@ -1925,17 +1923,17 @@ func (c *Controller) uninstallPrometheus(ctx context.Context, prometheus *v1.Pro
if err != nil {
return err
}
kubeClient, err := util.BuildExternalClientSet(cluster, c.client.PlatformV1())
kubeClient, err := util.BuildExternalClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}

crdClient, err := util.BuildExternalExtensionClientSet(cluster, c.client.PlatformV1())
crdClient, err := util.BuildExternalExtensionClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}

mclient, err := util.BuildExternalMonitoringClientSet(cluster, c.client.PlatformV1())
mclient, err := util.BuildExternalMonitoringClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return err
}
Expand Down Expand Up @@ -2079,9 +2077,15 @@ func (c *Controller) uninstallPrometheus(ctx context.Context, prometheus *v1.Pro
errs = append(errs, err)
}

for _, crdKind := range crdKinds {
crd := promopk8sutil.NewCustomResourceDefinition(crdKind, monitoring.GroupName, nil, true)
err := crdClient.ApiextensionsV1beta1().CustomResourceDefinitions().Delete(ctx, crd.Name, metav1.DeleteOptions{})
crds := getCRDs()
for _, crd := range crds {
crdObj := apiextensionsv1.CustomResourceDefinition{}
err := json.Unmarshal([]byte(crd), &crdObj)
if err != nil {
errs = append(errs, err)
continue
}
err = crdClient.ApiextensionsV1().CustomResourceDefinitions().Delete(ctx, crdObj.Name, metav1.DeleteOptions{})
if err != nil {
errs = append(errs, err)
}
Expand Down Expand Up @@ -2140,7 +2144,7 @@ func (c *Controller) watchPrometheusHealth(ctx context.Context, key string) func
log.Info("Prometheus health check over", log.String("prome", key))
return true, nil
}
kubeClient, err := util.BuildExternalClientSet(cluster, c.client.PlatformV1())
kubeClient, err := util.BuildExternalClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return false, err
}
Expand Down Expand Up @@ -2173,7 +2177,7 @@ func (c *Controller) checkPrometheusStatus(ctx context.Context, prometheus *v1.P
log.Info("Prometheus status checking over", log.String("prome", key))
return true, nil
}
kubeClient, err := util.BuildExternalClientSet(cluster, c.client.PlatformV1())
kubeClient, err := util.BuildExternalClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return false, err
}
Expand Down Expand Up @@ -2220,7 +2224,7 @@ func (c *Controller) checkPrometheusUpgrade(ctx context.Context, prometheus *v1.
log.Info("Prometheus upgrade over", log.String("prome", key))
return true, nil
}
kubeClient, err := util.BuildExternalClientSet(cluster, c.client.PlatformV1())
kubeClient, err := util.BuildExternalClientSet(ctx, cluster, c.client.PlatformV1())
if err != nil {
return false, err
}
Expand Down
22 changes: 22 additions & 0 deletions pkg/platform/controller/addon/prometheus/yamls.go

Large diffs are not rendered by default.

0 comments on commit 8a7babb

Please sign in to comment.