From e7b465c6ce2a72c4f9fb1c2192c7dff0f24bbef5 Mon Sep 17 00:00:00 2001
From: Vitaliy Guschin <vitaliy.guschin@spirent.com>
Date: Thu, 23 May 2024 12:17:38 +0400
Subject: [PATCH] Add an image with profiling enabled.

Signed-off-by: Vitaliy Guschin <vitaliy.guschin@spirent.com>
---
 .github/workflows/release.yml |  9 +++++++
 Dockerfile                    |  3 ++-
 main.go                       |  3 +++
 profiler_off.go               | 23 +++++++++++++++++
 profiler_on.go                | 48 +++++++++++++++++++++++++++++++++++
 5 files changed, 85 insertions(+), 1 deletion(-)
 create mode 100644 profiler_off.go
 create mode 100644 profiler_on.go

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 8fcf39e9..db390305 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -43,6 +43,15 @@ jobs:
     secrets:
       token: ${{ secrets.GITHUB_TOKEN }}
 
+  docker-profiler:
+    needs: [get-tag, check-gomod-deps]
+    uses: networkservicemesh/.github/.github/workflows/docker-release.yaml@main
+    with:
+      tag: ${{ needs.get-tag.outputs.tag }}-pprof
+      build-args: BUILD_TAGS=profiler
+    secrets:
+      token: ${{ secrets.GITHUB_TOKEN }}
+
   update-deployments-k8s:
     name: Update deployments-k8s
     needs: [get-tag, create-release]
diff --git a/Dockerfile b/Dockerfile
index e0006515..62da39e8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,6 +6,7 @@ ENV GO111MODULE=on
 ENV CGO_ENABLED=0
 ENV GOBIN=/bin
 ARG BUILDARCH=amd64
+ARG BUILD_TAGS=""
 RUN rm -r /etc/vpp
 RUN go install github.com/go-delve/delve/cmd/dlv@v1.21.0
 RUN go install github.com/grpc-ecosystem/grpc-health-probe@v0.4.25
@@ -23,7 +24,7 @@ COPY ./internal/afxdp/afxdp.c ./internal/afxdp/
 RUN clang -O3 -g -Wextra -Wall -target bpf -I/usr/include/$(uname -m)-linux-gnu -I/usr/include -c -o /bin/afxdp.o ./internal/afxdp/afxdp.c
 RUN go build ./internal/imports
 COPY . .
-RUN go build -o /bin/forwarder .
+RUN go build -tags "$BUILD_TAGS" -o /bin/forwarder .
 
 FROM build as test
 CMD go test -test.v ./...
diff --git a/main.go b/main.go
index dcea1705..a23ae462 100644
--- a/main.go
+++ b/main.go
@@ -92,6 +92,9 @@ func main() {
 	logrus.SetFormatter(&nested.Formatter{})
 	ctx = log.WithLog(ctx, logruslogger.New(ctx, map[string]interface{}{"cmd": os.Args[0]}))
 
+	// Start profiling server
+	startProfiler(ctx)
+
 	// ********************************************************************************
 	// Debug self if necessary
 	// ********************************************************************************
diff --git a/profiler_off.go b/profiler_off.go
new file mode 100644
index 00000000..2f22abfa
--- /dev/null
+++ b/profiler_off.go
@@ -0,0 +1,23 @@
+// Copyright (c) 2024 Pragmagic Inc. and/or its affiliates.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+
+//go:build !profiler
+
+package main
+
+import "context"
+
+func startProfiler(_ context.Context) {}
diff --git a/profiler_on.go b/profiler_on.go
new file mode 100644
index 00000000..883a3d5c
--- /dev/null
+++ b/profiler_on.go
@@ -0,0 +1,48 @@
+// Copyright (c) 2024 Pragmagic Inc. and/or its affiliates.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+// 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.
+
+//go:build profiler
+
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	_ "net/http/pprof" // #nosec
+	"time"
+
+	"github.com/networkservicemesh/sdk/pkg/tools/log"
+)
+
+func startProfiler(ctx context.Context) {
+	go func() {
+		profilerHTTPPort := 6060
+		log.FromContext(ctx).Infof("Profiler is enabled. Starting HTTP server on %d", profilerHTTPPort)
+
+		address := fmt.Sprintf("localhost:%d", profilerHTTPPort)
+
+		server := &http.Server{
+			Addr:         address,
+			ReadTimeout:  10 * time.Second,
+			WriteTimeout: 10 * time.Second,
+		}
+
+		if err := server.ListenAndServe(); err != nil {
+			log.FromContext(ctx).Errorf("Failed to start profiler: %s", err.Error())
+		}
+	}()
+}