added code to test profiling

Makefile

@@ -25,6 +25,15 @@ delete-postgres:
 	podman rm -f postgres
 	podman volume prune
 
+create-migration:
+	migrate create -ext sql -dir database/migration -seq $(name)
+
+migrate-up:
+	migrate -database ${DATABASE_URL} -path database/migration -verbose up
+
+migrate-down:
+	migrate -database ${DATABASE_URL} -path database/migration -verbose down
+
 create-jaeger:
 	podman run --name jaeger --hostname jaeger -e COLLECTOR_OTLP_ENABLED=true -p 6831:6831/udp -p 6832:6832/udp -p 5778:5778 -p 16686:16686 -p 4317:4317 -p 4318:4318 -p 14250:14250 -p 14268:14268 -p 14269:14269 -p 9411:9411 -d jaegertracing/all-in-one:1.52
 
@@ -46,14 +55,12 @@ delete-alertmanager:
 	podman rm -f alertmanager
 	podman volume prune
 
-create-migration:
-	migrate create -ext sql -dir database/migration -seq $(name)
+create-grafana:
+	podman run --name grafana --hostname grafana -p 3000:3000 -d grafana/grafana-oss:latest
 
-migrate-up:
-	migrate -database ${DATABASE_URL} -path database/migration -verbose up
-
-migrate-down:
-	migrate -database ${DATABASE_URL} -path database/migration -verbose down
+delete-grafana:
+	podman rm -f grafana
+	podman volume prune
 
 generate-dbdocs:
 	dbdocs build database/doc/db.dbml
@@ -75,4 +82,10 @@ generate-mock:
 run-test:
 	go test ./...
 
-.PHONY: create-consul delete-consul proto-generate create-pulsar delete-pulsar create-postgres delete-postgres create-migration migrate-up migrate-down create-jaeger delete-jaeger create-prometheus delete-prometheus create-alertmanager delete-alertmanager generate-dbdocs generate-schema generate-sqlc generate-image generate-mock run-test
+cpu-profiling:
+	go tool pprof cpu.pprof
+
+mem-profiling:
+	go tool pprof mem.pprof
+
+.PHONY: create-consul delete-consul proto-generate create-pulsar delete-pulsar create-postgres delete-postgres create-migration migrate-up migrate-down create-jaeger delete-jaeger create-prometheus delete-prometheus create-alertmanager delete-alertmanager create-grafana delete-grafana generate-dbdocs generate-schema generate-sqlc generate-image generate-mock run-test cpu-profiling mem-profiling

README.md

@@ -1,3 +1,3 @@
 # Go-with-Microservices
 
-Microservice project build using Hashicorp Consul, gRPC, apache Pulsar and PostgreSQL.  
+Complete microservice project build using Consul, gRPC, Pulsar, Jaeger, Prometheus and PostgreSQL. 

metadata/cmd/main.go

@@ -18,6 +18,7 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
+	"runtime/pprof"
 	"sync"
 	"syscall"
 	"time"
@@ -36,11 +37,15 @@ import (
 const serviceName = "metadata"
 
 func main() {
+	// configurations
 	var config string
+	var simulateCPUload bool
 	flag.StringVar(&config, "config", ".env", "Configuration path")
+	flag.BoolVar(&simulateCPUload, "simulatecpuload", false, "Simulate CPU load for profiling")
 	flag.Parse()
 	cfg := util.LoadConfig(config)
 
+	// environments
 	if cfg.Environment == "dev" {
 		var logger = slog.New(slog.NewTextHandler(os.Stdout, nil)).With("service_name", serviceName)
 		slog.SetDefault(logger)
@@ -49,7 +54,26 @@ func main() {
 		slog.SetDefault(logger)
 	}
 
-	// prometheus
+	// profiling
+	cpuFile, err := os.Create("cpu.pprof")
+	if err != nil {
+		panic(err)
+	}
+	memFile, err := os.Create("mem.pprof")
+	if err != nil {
+		panic(err)
+	}
+
+	pprof.StartCPUProfile(cpuFile)
+	pprof.WriteHeapProfile(memFile)
+	defer pprof.StopCPUProfile()
+	defer memFile.Close()
+
+	if simulateCPUload {
+		go util.HeavyOperation()
+	}
+
+	// prometheus metrics
 	reg := prometheus.NewRegistry()
 	counter := promauto.NewCounter(prometheus.CounterOpts{
 		Namespace: serviceName,
@@ -68,6 +92,7 @@ func main() {
 	defer reg.Unregister(counter)
 	counter.Inc()
 
+	// jaeger tracing
 	slog.Info("Starting the movie metadata service on port", slog.Int("port", cfg.MetadataPort))
 
 	ctx, cancel := context.WithCancel(context.Background())
@@ -88,6 +113,7 @@ func main() {
 	otel.SetTracerProvider(tp)
 	otel.SetTextMapPropagator(propagation.TraceContext{})
 
+	// service discovery
 	registry, err := consul.NewRegistry(cfg.ConsulURL)
 	if err != nil {
 		slog.Error("failed to connect consul registry:", slog.String("error", err.Error()))
@@ -112,6 +138,7 @@ func main() {
 	}()
 	defer registry.Deregister(ctx, instanceID, serviceName)
 
+	// services
 	conn, err := pgxpool.New(context.Background(), cfg.DatabaseURL)
 	if err != nil {
 		slog.Error("cannot connect to db:", slog.String("error", err.Error()))

util/profiling.go

@@ -0,0 +1,14 @@
+package util
+
+import (
+	"crypto/md5"
+	"crypto/rand"
+)
+
+func HeavyOperation() {
+	for {
+		token := make([]byte, 1024)
+		rand.Read(token)
+		md5.New().Write(token)
+	}
+}