opentel-optional

agent/container/main.go

@@ -14,16 +14,25 @@ import (
 func main() {
 	log.SetFlags(log.LstdFlags | log.Lshortfile)
 
-	tp, err := opentelemetry.InitTracer()
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
 	if err != nil {
-		log.Fatal(err)
+		log.Println("Unable to read open telemetry configurations")
 	}
-	defer func() {
-		if err := tp.Shutdown(context.Background()); err != nil {
-			log.Printf("Error shutting down tracer provider: %v", err)
+	if opentelconfig.IsEnabled {
+		tp, err := opentelemetry.InitTracer()
+		if err != nil {
+			log.Fatal(err)
 		}
-	}()
-
+		defer func() {
+			if err := tp.Shutdown(context.Background()); err != nil {
+				log.Printf("Error shutting down tracer provider: %v", err)
+			}
+		}()
+	} else {
+		log.Println("OpenTelemetry is disabled. Tracing will not be enabled.")
+	}
+
 	app := application.New()
 	go app.GithubContainerWatch()
 	go app.Start()

agent/container/pkg/application/application.go

@@ -46,13 +46,15 @@ func New() *Application {
 
 	r := gin.Default()
 
-	config, err := opentelemetry.GetConfigurations()
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
 	if err != nil {
 		log.Println("Unable to read open telemetry configurations")
 	}
+	if opentelconfig.IsEnabled {
+		r.Use(otelgin.Middleware(opentelconfig.ServiceName))
+	}
 
-	r.Use(otelgin.Middleware(config.ServiceName))
-
 	apiServer.BindRequest(r)
 
 	httpServer := &http.Server{

agent/container/pkg/application/handlers.go

@@ -1,26 +1,15 @@
 package application
 
 import (
-	"context"
 	"io"
 	"log"
 
 	"net/http"
-
-	"github.com/intelops/kubviz/pkg/opentelemetry"
-	"go.opentelemetry.io/otel"
-	"go.opentelemetry.io/otel/attribute"
 )
 
-//githubHandler handles the github webhooks post requests.
+// githubHandler handles the github webhooks post requests.
 func (app *Application) localRegistryHandler(w http.ResponseWriter, r *http.Request) {
 
-	ctx:=context.Background()
-	tracer := otel.Tracer("container-gitlab")
-	_, span := tracer.Start(opentelemetry.BuildContext(ctx), "localRegistryHandler")
-	span.SetAttributes(attribute.String("http.method", "POST"))
-	defer span.End()
-
 	event, err := io.ReadAll(r.Body)
 	if err != nil {
 		log.Printf("Event body read failed: %v", err)

agent/container/pkg/clients/nats_client.go

@@ -1,16 +1,12 @@
 package clients
 
 import (
-	"context"
 	"fmt"
 	"log"
 	"time"
 
 	"github.com/intelops/kubviz/agent/container/pkg/config"
 	"github.com/intelops/kubviz/pkg/mtlsnats"
-	"github.com/intelops/kubviz/pkg/opentelemetry"
-	"go.opentelemetry.io/otel"
-	"go.opentelemetry.io/otel/attribute"
 
 	"github.com/nats-io/nats.go"
 )
@@ -143,12 +139,6 @@ func (n *NATSContext) Close() {
 // An error is returned if the publishing process fails, such as if the connection is lost or if there are issues with the JetStream.
 func (n *NATSContext) Publish(event []byte, repo string) error {
 
-	ctx := context.Background()
-	tracer := otel.Tracer("container-nats-client")
-	_, span := tracer.Start(opentelemetry.BuildContext(ctx), "ContainerPublish")
-	span.SetAttributes(attribute.String("repo-name", repo))
-	defer span.End()
-
 	msg := nats.NewMsg(eventSubject)
 	msg.Data = event
 	msg.Header.Set("REPO_NAME", repo)

agent/container/pkg/handler/api_handler.go

@@ -33,13 +33,15 @@ func NewAPIHandler(conn *clients.NATSContext) (*APIHandler, error) {
 
 func (ah *APIHandler) BindRequest(r *gin.Engine) {
 
-	config, err := opentelemetry.GetConfigurations()
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
 	if err != nil {
 		log.Println("Unable to read open telemetry configurations")
 	}
+	if opentelconfig.IsEnabled {
+		r.Use(otelgin.Middleware(opentelconfig.ServiceName))
+	}
 
-	r.Use(otelgin.Middleware(config.ServiceName))
-
 	apiGroup := r.Group("/")
 	{
 		apiGroup.GET("/api-docs", ah.GetApiDocs)

agent/container/pkg/handler/azure_container.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/intelops/kubviz/model"
+	"github.com/intelops/kubviz/pkg/opentelemetry"
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
 )
@@ -22,11 +23,18 @@ var ErrInvalidPayload = errors.New("invalid or malformed Azure Container Registr
 // If the payload is invalid or the publishing process fails, an error response is returned.
 func (ah *APIHandler) PostEventAzureContainer(c *gin.Context) {
 
-	tracer := otel.Tracer("azure-container")
-	_, span := tracer.Start(c.Request.Context(), "PostEventAzureContainer")
-	span.SetAttributes(attribute.String("http.method", "POST"))
-	defer span.End()
-
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
+	if err != nil {
+		log.Println("Unable to read open telemetry configurations")
+	}
+	if opentelconfig.IsEnabled {
+		tracer := otel.Tracer("azure-container")
+		_, span := tracer.Start(c.Request.Context(), "PostEventAzureContainer")
+		span.SetAttributes(attribute.String("http.method", "POST"))
+		defer span.End()
+	}
+
 	defer func() {
 		_, _ = io.Copy(io.Discard, c.Request.Body)
 		_ = c.Request.Body.Close()

agent/container/pkg/handler/docker_event_dockerhub.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 
 	"github.com/gin-gonic/gin"
+	"github.com/intelops/kubviz/pkg/opentelemetry"
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
 )
@@ -18,12 +19,18 @@ var (
 )
 
 func (ah *APIHandler) PostEventDockerHub(c *gin.Context) {
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
+	if err != nil {
+		log.Println("Unable to read open telemetry configurations")
+	}
+	if opentelconfig.IsEnabled {
+		tracer := otel.Tracer("dockerhub-container")
+		_, span := tracer.Start(c.Request.Context(), "PostEventDockerHub")
+		span.SetAttributes(attribute.String("http.method", "POST"))
+		defer span.End()
+	}
 
-	tracer := otel.Tracer("dockerhub-container")
-	_, span := tracer.Start(c.Request.Context(), "PostEventDockerHub")
-	span.SetAttributes(attribute.String("http.method", "POST"))
-	defer span.End()
-
 	defer func() {
 		_, _ = io.Copy(io.Discard, c.Request.Body)
 		_ = c.Request.Body.Close()

agent/container/pkg/handler/jfrog_container.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/intelops/kubviz/model"
+	"github.com/intelops/kubviz/pkg/opentelemetry"
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
 )
@@ -17,11 +18,18 @@ var ErrInvalidPayloads = errors.New("invalid or malformed jfrog Container Regist
 
 func (ah *APIHandler) PostEventJfrogContainer(c *gin.Context) {
 
-	tracer := otel.Tracer("jfrog-container")
-	_, span := tracer.Start(c.Request.Context(), "PostEventJfrogContainer")
-	span.SetAttributes(attribute.String("http.method", "POST"))
-	defer span.End()
-
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
+	if err != nil {
+		log.Println("Unable to read open telemetry configurations")
+	}
+	if opentelconfig.IsEnabled {
+		tracer := otel.Tracer("jfrog-container")
+		_, span := tracer.Start(c.Request.Context(), "PostEventJfrogContainer")
+		span.SetAttributes(attribute.String("http.method", "POST"))
+		defer span.End()
+	}
+
 	defer func() {
 		_, _ = io.Copy(io.Discard, c.Request.Body)
 		_ = c.Request.Body.Close()

agent/container/pkg/handler/quay_handler.go

@@ -8,17 +8,25 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/intelops/kubviz/model"
+	"github.com/intelops/kubviz/pkg/opentelemetry"
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/attribute"
 )
 
 func (ah *APIHandler) PostEventQuayContainer(c *gin.Context) {
 
-	tracer := otel.Tracer("quay-container")
-	_, span := tracer.Start(c.Request.Context(), "PostEventQuayContainer")
-	span.SetAttributes(attribute.String("http.method", "POST"))
-	defer span.End()
-
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
+	if err != nil {
+		log.Println("Unable to read open telemetry configurations")
+	}
+	if opentelconfig.IsEnabled {
+		tracer := otel.Tracer("quay-container")
+		_, span := tracer.Start(c.Request.Context(), "PostEventQuayContainer")
+		span.SetAttributes(attribute.String("http.method", "POST"))
+		defer span.End()
+	}
+
 	defer func() {
 		_, _ = io.Copy(io.Discard, c.Request.Body)
 		_ = c.Request.Body.Close()
@@ -45,4 +53,4 @@ func (ah *APIHandler) PostEventQuayContainer(c *gin.Context) {
 		return
 	}
 	c.Status(http.StatusOK)
-}
+}

agent/git/main.go

@@ -22,15 +22,24 @@ func main() {
 		log.Fatalf("Could not parse env Config: %v", err)
 	}
 
-	tp, err := opentelemetry.InitTracer()
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
 	if err != nil {
-		log.Fatal(err)
+		log.Println("Unable to read open telemetry configurations")
 	}
-	defer func() {
-		if err := tp.Shutdown(context.Background()); err != nil {
-			log.Printf("Error shutting down tracer provider: %v", err)
+	if opentelconfig.IsEnabled {
+		tp, err := opentelemetry.InitTracer()
+		if err != nil {
+			log.Fatal(err)
 		}
-	}()
+		defer func() {
+			if err := tp.Shutdown(context.Background()); err != nil {
+				log.Printf("Error shutting down tracer provider: %v", err)
+			}
+		}()
+	} else {
+		log.Println("OpenTelemetry is disabled. Tracing will not be enabled.")
+	}
 
 	// Connect to NATS
 	natsContext, err := clients.NewNATSContext(cfg)

agent/git/pkg/application/application.go

@@ -44,13 +44,15 @@ func New(conf *config.Config, conn *clients.NATSContext) *Application {
 func (app *Application) Routes() *gin.Engine {
 	router := gin.New()
 
-	config, err := opentelemetry.GetConfigurations()
+	//opentelemetry
+	opentelconfig, err := opentelemetry.GetConfigurations()
 	if err != nil {
 		log.Println("Unable to read open telemetry configurations")
 	}
+	if opentelconfig.IsEnabled {
+		router.Use(otelgin.Middleware(opentelconfig.ServiceName))
+	}
 
-	router.Use(otelgin.Middleware(config.ServiceName))
-
 	api.RegisterHandlers(router, app)
 	return router
 }