X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=vnfs%2FDAaaS%2Fmicroservices%2Fvisualization-operator%2Fpkg%2Fcontroller%2Fgrafanadatasource%2Fgrafanadatasource_controller.go;fp=vnfs%2FDAaaS%2Fmicroservices%2Fvisualization-operator%2Fpkg%2Fcontroller%2Fgrafanadatasource%2Fgrafanadatasource_controller.go;h=f46cf1b49fecf0a654c09e1ce39a2c8d0eecb2e9;hb=f2d37cb632bb1706f3e223776e72e50253d4025c;hp=0000000000000000000000000000000000000000;hpb=32ec6c9216242b0d9cf76da93eb74b60b6dae7c1;p=demo.git diff --git a/vnfs/DAaaS/microservices/visualization-operator/pkg/controller/grafanadatasource/grafanadatasource_controller.go b/vnfs/DAaaS/microservices/visualization-operator/pkg/controller/grafanadatasource/grafanadatasource_controller.go new file mode 100644 index 00000000..f46cf1b4 --- /dev/null +++ b/vnfs/DAaaS/microservices/visualization-operator/pkg/controller/grafanadatasource/grafanadatasource_controller.go @@ -0,0 +1,167 @@ +package grafanadatasource + +import ( + "bytes" + "context" + "encoding/json" + + onapv1alpha1 "visualization-operator/pkg/apis/onap/v1alpha1" + + "k8s.io/apimachinery/pkg/api/errors" + "k8s.io/apimachinery/pkg/runtime" + "net/http" + "sigs.k8s.io/controller-runtime/pkg/client" + "sigs.k8s.io/controller-runtime/pkg/controller" + "sigs.k8s.io/controller-runtime/pkg/handler" + "sigs.k8s.io/controller-runtime/pkg/manager" + "sigs.k8s.io/controller-runtime/pkg/reconcile" + logf "sigs.k8s.io/controller-runtime/pkg/runtime/log" + "sigs.k8s.io/controller-runtime/pkg/source" +) + +var log = logf.Log.WithName("controller_grafanadatasource") + +// Add creates a new GrafanaDataSource Controller and adds it to the Manager. The Manager will set fields on the Controller +// and Start it when the Manager is Started. +func Add(mgr manager.Manager) error { + return add(mgr, newReconciler(mgr)) +} + +// newReconciler returns a new reconcile.Reconciler +func newReconciler(mgr manager.Manager) reconcile.Reconciler { + return &ReconcileGrafanaDataSource{client: mgr.GetClient(), scheme: mgr.GetScheme()} +} + +// add adds a new Controller to mgr with r as the reconcile.Reconciler +func add(mgr manager.Manager, r reconcile.Reconciler) error { + // Create a new controller + c, err := controller.New("grafanadatasource-controller", mgr, controller.Options{Reconciler: r}) + if err != nil { + return err + } + + // Watch for changes to primary resource GrafanaDataSource + err = c.Watch(&source.Kind{Type: &onapv1alpha1.GrafanaDataSource{}}, &handler.EnqueueRequestForObject{}) + if err != nil { + return err + } + + return nil +} + +// blank assignment to verify that ReconcileGrafanaDataSource implements reconcile.Reconciler +var _ reconcile.Reconciler = &ReconcileGrafanaDataSource{} + +// ReconcileGrafanaDataSource reconciles a GrafanaDataSource object +type ReconcileGrafanaDataSource struct { + // This client, initialized using mgr.Client() above, is a split client + // that reads objects from the cache and writes to the apiserver + client client.Client + scheme *runtime.Scheme +} + +// Reconcile reads that state of the cluster for a GrafanaDataSource object and makes changes based on the state read +// and what is in the GrafanaDataSource.Spec +// Note: +// The Controller will requeue the Request to be processed again if the returned error is non-nil or +// Result.Requeue is true, otherwise upon completion it will remove the work from the queue. +func (r *ReconcileGrafanaDataSource) Reconcile(request reconcile.Request) (reconcile.Result, error) { + reqLogger := log.WithValues("Request.Namespace", request.Namespace, "Request.Name", request.Name) + reqLogger.Info("Reconciling GrafanaDataSource") + + // Fetch the GrafanaDataSource instance + instance := &onapv1alpha1.GrafanaDataSource{} + err := r.client.Get(context.TODO(), request.NamespacedName, instance) + if err != nil { + if errors.IsNotFound(err) { + // Request object not found, could have been deleted after reconcile request. + // Owned objects are automatically garbage collected. For additional cleanup logic use finalizers. + // Return and don't requeue + return reconcile.Result{}, nil + } + // Error reading the object - requeue the request. + + return reconcile.Result{}, err + } + + datasources := instance.Spec.Datasources + grafana := instance.Spec.Grafana + + reqLogger.V(1).Info(" Datasource Name ", "datasources", datasources) + + //loop through all datasources in the spec + for _, datasource := range datasources { + + //check if datasource exists + grafanaURL := grafana["url"] + "/api/datasources/name/" + datasource.Name + grafanaUsername := grafana["username"] + grafanaPassword := grafana["password"] + + client := &http.Client{} + req, err := http.NewRequest("GET", grafanaURL, nil) + if err != nil { + reqLogger.Error(err, "GET REQUEST error") + return reconcile.Result{}, err + } + req.SetBasicAuth(grafanaUsername, grafanaPassword) + getResp, err := client.Do(req) + if err != nil { + reqLogger.Error(err, "GET RESPONSE error") + return reconcile.Result{}, err + } + + defer getResp.Body.Close() + + //add datasource if datasource does not exist already + if getResp.StatusCode == http.StatusNotFound { + reqLogger.Info("Datasource does not exist, creating one...") + // create datasource + if err := createDataSource(grafana, datasource); err != nil { + return reconcile.Result{}, err + } + } else if getResp.StatusCode == http.StatusOK { + //if datasource already exists + reqLogger.V(1).Info("datasource already exists", "datasource", datasource.Name) + } else { + reqLogger.Error(err, "unknown error", datasource.Name) + } + + } + return reconcile.Result{}, nil +} + +func createDataSource(grafana map[string]string, datasource onapv1alpha1.Datasource) error { + reqLogger := log.WithValues("Datasource name", datasource.Name, "Datasource URL", datasource.URL) + reqLogger.Info("creating datasource") + + grafanaURL := grafana["url"] + "/api/datasources" + grafanaUsername := grafana["username"] + grafanaPassword := grafana["password"] + + client := &http.Client{} + postBody, err := json.Marshal(datasource) + if err != nil { + reqLogger.Error(err, "JSON Marshalling error") + return err + } + + req, err := http.NewRequest("POST", grafanaURL, bytes.NewBuffer(postBody)) + if err != nil { + reqLogger.Error(err, "POST REQUEST error") + return err + } + req.Header.Set("Content-Type", "application/json") + req.SetBasicAuth(grafanaUsername, grafanaPassword) + postResp, err := client.Do(req) + if err != nil { + reqLogger.Error(err, "POST RESPONSE error") + return err + } + defer req.Body.Close() + + if postResp.StatusCode == http.StatusOK { + reqLogger.Info("Datasource created") + return nil + } + return err +}