feat: Added profefe.com/service annotation

Fixes #19

By default the service name is the pod name. As consequence you have to
know it when querying from profefe. It is not always easy to achieve
because pod name changes frequently.

When specified the annotation `profefe.com/service` override the pod
name as service name.

Signed-off-by: Gianluca Arbezzano <gianarb92@gmail.com>

Author: Gianluca Arbezzano

pkg/cmd/capture.go

@@ -177,7 +177,7 @@ func writeProfiles(ctx context.Context, pClient *profefe.Client, profiles map[pp
 		// if the profefe client is not null the profile needs to be pushed to
 		// profefe server, otherwise it is written into a file locally
 		if pClient != nil {
-			saved, err := pClient.SavePprof(ctx, profefe.SavePprofRequest{
+			req := profefe.SavePprofRequest{
 				Profile:    profile,
 				Service:    target.Name,
 				InstanceID: target.Status.HostIP,
@@ -186,7 +186,12 @@ func writeProfiles(ctx context.Context, pClient *profefe.Client, profiles map[pp
 					"namespace": target.Namespace,
 					"from":      "kube-profefe",
 				},
-			})
+			}
+			if serviceName, ok := target.Annotations["profefe.com/service"]; ok && serviceName != "" {
+				req.Service = serviceName
+				req.Labels["pod"] = target.Name
+			}
+			saved, err := pClient.SavePprof(context.Background(), req)
 			if err != nil {
 				println(fmt.Sprintf("%s type=%s profile_type=%s", err.Error(), profefeType, profile.PeriodType.Type))
 			} else {

pkg/cmd/kprofefe.go

@@ -163,7 +163,7 @@ func do(ctx context.Context, l *zap.Logger, pClient *profefe.Client, target core
 			logger.Warn("Unknown profile type it can not be sent to profefe. Skip this profile")
 			continue
 		}
-		saved, err := pClient.SavePprof(context.Background(), profefe.SavePprofRequest{
+		req := profefe.SavePprofRequest{
 			Profile:    profile,
 			Service:    target.Name,
 			InstanceID: target.Status.HostIP,
@@ -172,7 +172,12 @@ func do(ctx context.Context, l *zap.Logger, pClient *profefe.Client, target core
 				"namespace": target.Namespace,
 				"from":      "kube-profefe",
 			},
-		})
+		}
+		if serviceName, ok := target.Annotations["profefe.com/service"]; ok && serviceName != "" {
+			req.Service = serviceName
+			req.Labels["pod"] = target.Name
+		}
+		saved, err := pClient.SavePprof(context.Background(), req)
 		if err != nil {
 			logger.Warn("Unknown profile type it can not be sent to profefe. Skip this profile", zap.Error(err))
 		} else {