190 lines
6.7 KiB
Go
190 lines
6.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
|
|
// Import all Kubernetes client auth plugins (e.g. Azure, GCP, OIDC, etc.)
|
|
_ "k8s.io/client-go/plugin/pkg/client/auth"
|
|
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/apis"
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/controller/jenkins"
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/controller/jenkins/constants"
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/event"
|
|
"github.com/jenkinsci/kubernetes-operator/pkg/log"
|
|
"github.com/jenkinsci/kubernetes-operator/version"
|
|
|
|
"github.com/operator-framework/operator-sdk/pkg/k8sutil"
|
|
kubemetrics "github.com/operator-framework/operator-sdk/pkg/kube-metrics"
|
|
"github.com/operator-framework/operator-sdk/pkg/leader"
|
|
"github.com/operator-framework/operator-sdk/pkg/log/zap"
|
|
"github.com/operator-framework/operator-sdk/pkg/metrics"
|
|
"github.com/operator-framework/operator-sdk/pkg/restmapper"
|
|
sdkVersion "github.com/operator-framework/operator-sdk/version"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/pflag"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/util/intstr"
|
|
"k8s.io/client-go/kubernetes"
|
|
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
|
|
"k8s.io/client-go/rest"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/config"
|
|
"sigs.k8s.io/controller-runtime/pkg/manager"
|
|
"sigs.k8s.io/controller-runtime/pkg/runtime/signals"
|
|
)
|
|
|
|
// Change below variables to serve metrics on different host or port.
|
|
var (
|
|
metricsHost = "0.0.0.0"
|
|
metricsPort int32 = 8383
|
|
operatorMetricsPort int32 = 8686
|
|
)
|
|
|
|
//var log = logf.Log.WithName("cmd")
|
|
|
|
func printInfo() {
|
|
log.Log.Info(fmt.Sprintf("Version: %s", version.Version))
|
|
log.Log.Info(fmt.Sprintf("Git commit: %s", version.GitCommit))
|
|
log.Log.Info(fmt.Sprintf("Go Version: %s", runtime.Version()))
|
|
log.Log.Info(fmt.Sprintf("Go OS/Arch: %s/%s", runtime.GOOS, runtime.GOARCH))
|
|
log.Log.Info(fmt.Sprintf("operator-sdk Version: %v", sdkVersion.Version))
|
|
}
|
|
|
|
func main() {
|
|
// Add the zap logger flag set to the CLI. The flag set must
|
|
// be added before calling pflag.Parse().
|
|
pflag.CommandLine.AddFlagSet(zap.FlagSet())
|
|
|
|
// Add flags registered by imported packages (e.g. glog and
|
|
// controller-runtime)
|
|
pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
|
|
|
|
minikube := pflag.Bool("minikube", false, "Use minikube as a Kubernetes platform")
|
|
local := pflag.Bool("local", false, "Run operator locally")
|
|
debug := pflag.Bool("debug", false, "Set log level to debug")
|
|
pflag.Parse()
|
|
|
|
log.SetupLogger(*debug)
|
|
printInfo()
|
|
|
|
namespace, err := k8sutil.GetWatchNamespace()
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to get watch namespace"), *debug)
|
|
}
|
|
log.Log.Info(fmt.Sprintf("Watch namespace: %v", namespace))
|
|
|
|
// get a config to talk to the apiserver
|
|
cfg, err := config.GetConfig()
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to get config"), *debug)
|
|
}
|
|
|
|
ctx := context.TODO()
|
|
|
|
// Become the leader before proceeding
|
|
err = leader.Become(ctx, "jenkins-operator-lock")
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to become leader"), *debug)
|
|
}
|
|
|
|
// Create a new Cmd to provide shared dependencies and start components
|
|
mgr, err := manager.New(cfg, manager.Options{
|
|
Namespace: namespace,
|
|
MapperProvider: restmapper.NewDynamicRESTMapper,
|
|
MetricsBindAddress: fmt.Sprintf("%s:%d", metricsHost, metricsPort),
|
|
})
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to create manager"), *debug)
|
|
}
|
|
|
|
log.Log.Info("Registering Components.")
|
|
|
|
// setup Scheme for all resources
|
|
if err := apis.AddToScheme(mgr.GetScheme()); err != nil {
|
|
fatal(errors.Wrap(err, "failed to setup scheme"), *debug)
|
|
}
|
|
|
|
// setup events
|
|
events, err := event.New(cfg, constants.OperatorName)
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to create manager"), *debug)
|
|
}
|
|
|
|
clientSet, err := kubernetes.NewForConfig(cfg)
|
|
if err != nil {
|
|
fatal(errors.Wrap(err, "failed to create Kubernetes client set"), *debug)
|
|
}
|
|
|
|
// setup Jenkins controller
|
|
if err := jenkins.Add(mgr, *local, *minikube, events, *clientSet, *cfg); err != nil {
|
|
fatal(errors.Wrap(err, "failed to setup controllers"), *debug)
|
|
}
|
|
|
|
if err = serveCRMetrics(cfg); err != nil {
|
|
log.Log.V(log.VWarn).Info("Could not generate and serve custom resource metrics", "error", err.Error())
|
|
}
|
|
|
|
// Add to the below struct any other metrics ports you want to expose.
|
|
servicePorts := []v1.ServicePort{
|
|
{Port: metricsPort, Name: metrics.OperatorPortName, Protocol: v1.ProtocolTCP, TargetPort: intstr.IntOrString{Type: intstr.Int, IntVal: metricsPort}},
|
|
{Port: operatorMetricsPort, Name: metrics.CRPortName, Protocol: v1.ProtocolTCP, TargetPort: intstr.IntOrString{Type: intstr.Int, IntVal: operatorMetricsPort}},
|
|
}
|
|
// Create Service object to expose the metrics port(s).
|
|
service, err := metrics.CreateMetricsService(ctx, cfg, servicePorts)
|
|
if err != nil {
|
|
log.Log.V(log.VWarn).Info("Could not create metrics Service", "error", err.Error())
|
|
}
|
|
|
|
// CreateServiceMonitors will automatically create the prometheus-operator ServiceMonitor resources
|
|
// necessary to configure Prometheus to scrape metrics from this operator.
|
|
services := []*v1.Service{service}
|
|
_, err = metrics.CreateServiceMonitors(cfg, namespace, services)
|
|
if err != nil {
|
|
log.Log.V(log.VWarn).Info("Could not create ServiceMonitor object", "error", err.Error())
|
|
// If this operator is deployed to a cluster without the prometheus-operator running, it will return
|
|
// ErrServiceMonitorNotPresent, which can be used to safely skip ServiceMonitor creation.
|
|
if err == metrics.ErrServiceMonitorNotPresent {
|
|
log.Log.V(log.VWarn).Info("Install prometheus-operator in your cluster to create ServiceMonitor objects", "error", err.Error())
|
|
}
|
|
}
|
|
|
|
log.Log.Info("Starting the Cmd.")
|
|
|
|
// start the Cmd
|
|
if err := mgr.Start(signals.SetupSignalHandler()); err != nil {
|
|
fatal(errors.Wrap(err, "failed to start cmd"), *debug)
|
|
}
|
|
}
|
|
|
|
// serveCRMetrics gets the Operator/CustomResource GVKs and generates metrics based on those types.
|
|
// It serves those metrics on "http://metricsHost:operatorMetricsPort".
|
|
func serveCRMetrics(cfg *rest.Config) error {
|
|
// Below function returns filtered operator/CustomResource specific GVKs.
|
|
// For more control override the below GVK list with your own custom logic.
|
|
filteredGVK, err := k8sutil.GetGVKsFromAddToScheme(apis.AddToScheme)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Get the namespace the operator is currently deployed in.
|
|
operatorNs, err := k8sutil.GetOperatorNamespace()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// To generate metrics in other namespaces, add the values below.
|
|
ns := []string{operatorNs}
|
|
// Generate and serve custom resource specific metrics.
|
|
return kubemetrics.GenerateAndServeCRMetrics(cfg, ns, filteredGVK, metricsHost, operatorMetricsPort)
|
|
}
|
|
|
|
func fatal(err error, debug bool) {
|
|
if debug {
|
|
log.Log.Error(nil, fmt.Sprintf("%+v", err))
|
|
} else {
|
|
log.Log.Error(nil, fmt.Sprintf("%s", err))
|
|
}
|
|
os.Exit(-1)
|
|
}
|