ctrlclient.go

Documentation: github.com/chaos-mesh/chaos-mesh/pkg/ctrl/client

     1  // Copyright 2021 Chaos Mesh Authors.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  // http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  //
    15  
    16  package client
    17  
    18  import (
    19  	"context"
    20  
    21  	"github.com/hasura/go-graphql-client"
    22  )
    23  
    24  type CtrlClient struct {
    25  	QueryClient        *graphql.Client
    26  	SubscriptionClient *graphql.SubscriptionClient
    27  }
    28  
    29  func NewCtrlClient(url string) *CtrlClient {
    30  	return &CtrlClient{
    31  		QueryClient:        graphql.NewClient(url, nil),
    32  		SubscriptionClient: graphql.NewSubscriptionClient(url),
    33  	}
    34  }
    35  
    36  func (c *CtrlClient) ListNamespace(ctx context.Context) ([]string, error) {
    37  	namespaceQuery := new(struct {
    38  		Namespace []struct {
    39  			Ns string
    40  		}
    41  	})
    42  
    43  	err := c.QueryClient.Query(ctx, namespaceQuery, nil)
    44  	if err != nil {
    45  		return nil, err
    46  	}
    47  
    48  	var namespaces []string
    49  	for _, ns := range namespaceQuery.Namespace {
    50  		namespaces = append(namespaces, ns.Ns)
    51  	}
    52  
    53  	return namespaces, nil
    54  }
    55