cmattoon/aws-ssm

View on GitHub
pkg/controller/client.go

Summary

Maintainability
A
0 mins
Test Coverage
/**
 * Copyright 2018 Curtis Mattoon
 *
 * 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 controller

import (
    log "github.com/sirupsen/logrus"
    "k8s.io/client-go/kubernetes"
    "k8s.io/client-go/tools/clientcmd"
    "sync"
)

// ClientGenerator provides clients
type ClientGenerator interface {
    KubeClient() (kubernetes.Interface, error)
}

// SingletonClientGenerator stores provider clients and guarantees that only one instance of client
// will be generated
type SingletonClientGenerator struct {
    KubeConfig string
    KubeMaster string
    client     kubernetes.Interface
    sync.Once
}

// KubeClient generates a kube client if it was not created before
func (p *SingletonClientGenerator) KubeClient() (kubernetes.Interface, error) {
    var err error
    p.Once.Do(func() {
        p.client, err = NewKubeClient(p.KubeConfig, p.KubeMaster)
    })
    return p.client, err
}

// will fallback to restclient.InClusterConfig() if both kubeconfig/master_url == ""
func NewKubeClient(kubeconfig string, master_url string) (*kubernetes.Clientset, error) {
    config, err := clientcmd.BuildConfigFromFlags(master_url, kubeconfig)
    if err != nil {
        return nil, err
    }

    client, err := kubernetes.NewForConfig(config)
    if err != nil {
        return nil, err
    }

    log.Infof("Connected to cluster at %s", config.Host)
    return client, nil
}