Skip to content
This repository has been archived by the owner on Jun 21, 2022. It is now read-only.

Commit

Permalink
Fix IA APIs registration (#534)
Browse files Browse the repository at this point in the history
  • Loading branch information
AlekSi authored Nov 24, 2020
1 parent bc3c383 commit aa87f50
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 3 deletions.
1 change: 1 addition & 0 deletions docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ services:
- TRAVIS_PULL_REQUEST=${TRAVIS_PULL_REQUEST}
- REVIEWDOG_GITHUB_API_TOKEN=${REVIEWDOG_GITHUB_API_TOKEN}
- PERCONA_TEST_DBAAS=${PERCONA_TEST_DBAAS:-0}
- PERCONA_TEST_IA=1

# for delve
cap_add:
Expand Down
10 changes: 9 additions & 1 deletion main.go
Original file line number Diff line number Diff line change
Expand Up @@ -180,14 +180,15 @@ func runGRPCServer(ctx context.Context, deps *gRPCServerDeps) {
managementpb.RegisterSecurityChecksServer(gRPCServer, managementgrpc.NewChecksServer(checksSvc))

// TODO remove PERCONA_TEST_IA once IA is out of beta: https://jira.percona.com/browse/PMM-7001
if enable, _ := strconv.ParseBool(os.Getenv("PERCONA_TEST_IA")); enable {
if enable, err := strconv.ParseBool(os.Getenv("PERCONA_TEST_IA")); err == nil && enable {
l.Warnf("Enabling experimental IA APIs.")
iav1beta1.RegisterAlertsServer(gRPCServer, ia.NewAlertsService())
iav1beta1.RegisterChannelsServer(gRPCServer, ia.NewChannelsService())
iav1beta1.RegisterRulesServer(gRPCServer, ia.NewRulesService())
iav1beta1.RegisterTemplatesServer(gRPCServer, ia.NewTemplatesService())
}

// TODO Remove once changing settings.DBaaS.Enabled is possible via API.
if deps.settings.DBaaS.Enabled {
dbaasv1beta1.RegisterKubernetesServer(gRPCServer, managementdbaas.NewKubernetesServer(deps.db, deps.dbaasControllerClient))
dbaasv1beta1.RegisterXtraDBClusterServer(gRPCServer, managementdbaas.NewXtraDBClusterService(deps.db, deps.dbaasControllerClient))
Expand Down Expand Up @@ -285,6 +286,11 @@ func runHTTP1Server(ctx context.Context, deps *http1ServerDeps) {
managementpb.RegisterAnnotationHandlerFromEndpoint,
managementpb.RegisterSecurityChecksHandlerFromEndpoint,

iav1beta1.RegisterAlertsHandlerFromEndpoint,
iav1beta1.RegisterChannelsHandlerFromEndpoint,
iav1beta1.RegisterRulesHandlerFromEndpoint,
iav1beta1.RegisterTemplatesHandlerFromEndpoint,

dbaasv1beta1.RegisterKubernetesHandlerFromEndpoint,
dbaasv1beta1.RegisterXtraDBClusterHandlerFromEndpoint,
dbaasv1beta1.RegisterPSMDBClusterHandlerFromEndpoint,
Expand Down Expand Up @@ -470,9 +476,11 @@ func getQANClient(ctx context.Context, sqlDB *sql.DB, dbName, qanAPIAddr string)
}

func getDBaaSControllerClient(ctx context.Context, dbaasControllerAPIAddr string, settings *models.Settings) *dbaas.Client {
// TODO Remove once changing settings.DBaaS.Enabled is possible via API.
if !settings.DBaaS.Enabled {
return dbaas.NewClient(nil)
}

opts := []grpc.DialOption{
grpc.WithInsecure(),
grpc.WithConnectParams(grpc.ConnectParams{Backoff: backoff.Config{MaxDelay: 10 * time.Second}, MinConnectTimeout: 10 * time.Second}),
Expand Down
4 changes: 2 additions & 2 deletions services/dbaas/dbaas_client_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ func TestClient(t *testing.T) {
return c
}
t.Run("InvalidKubeConfig", func(t *testing.T) {
if os.Getenv("PERCONA_TEST_DBAAS") != "1" {
if os.Getenv("PERCONA_TEST_DBAAS") != "1" { // FIXME use strconv.ParseBool
t.Skip("PERCONA_TEST_DBAAS env variable is not passed, skipping")
}
kubeConfig := os.Getenv("PERCONA_TEST_DBAAS_KUBECONFIG")
Expand All @@ -59,7 +59,7 @@ func TestClient(t *testing.T) {
})

t.Run("InvalidKubeConfig", func(t *testing.T) {
if os.Getenv("PERCONA_TEST_DBAAS") != "1" {
if os.Getenv("PERCONA_TEST_DBAAS") != "1" { // FIXME use strconv.ParseBool
t.Skip("PERCONA_TEST_DBAAS env variable is not passed, skipping")
}

Expand Down

0 comments on commit aa87f50

Please sign in to comment.