From bb65ece97b796df893962fff4407a8a9991699be Mon Sep 17 00:00:00 2001
From: Gimmy <gheorghe.popescu@aserto.com>
Date: Fri, 22 Nov 2024 16:02:54 +0200
Subject: [PATCH] fix trace message

---
 lib/util/connect.ts | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/lib/util/connect.ts b/lib/util/connect.ts
index 38b17ae..7bb5dae 100644
--- a/lib/util/connect.ts
+++ b/lib/util/connect.ts
@@ -1,3 +1,4 @@
+import { DescMessage } from "@bufbuild/protobuf";
 import {
   Code,
   ConnectError,
@@ -16,7 +17,9 @@ import {
 import { log } from "../log";
 
 export const setHeader = (
-  req: UnaryRequest | StreamRequest,
+  req:
+    | UnaryRequest<DescMessage, DescMessage>
+    | StreamRequest<DescMessage, DescMessage>,
   key: string,
   value: string,
 ) => {
@@ -24,7 +27,15 @@ export const setHeader = (
 };
 
 export const traceMessage: Interceptor = (next) => async (req) => {
-  log(JSON.stringify(req));
+  log(
+    JSON.stringify({
+      message: req.message,
+      method: req.method.toString(),
+      requestMethod: req.requestMethod.toString(),
+      service: req.service.toString(),
+      url: req.url.toString(),
+    }),
+  );
   return await next(req);
 };