chore(utis): mv setReqIdHeader to getUID

Signed-off-by: Dwi Siswanto <me@dw1.io>

analyze.go

@@ -23,7 +23,7 @@ func (t *Teler) Analyze(w http.ResponseWriter, r *http.Request) error {
 
 	// If threat detected, set teler request ID to the header
 	if err != nil {
-		setReqIdHeader(w)
+		setCustomHeader(w, xTelerReqId, getUID())
 	}
 
 	return err

teler.go

@@ -358,15 +358,16 @@ func (t *Teler) postAnalyze(w http.ResponseWriter, r *http.Request, k threat.Thr
 		return
 	}
 
-	// Set teler request ID to the header
-	id := setReqIdHeader(w)
+	// Get unique ID
+	id := getUID()
 
 	// Get the error message & convert to string as a message
 	msg := err.Error()
 
-	// Set custom headers ("X-Teler-Msg" and "X-Teler-Threat")
+	// Set custom headers ("X-Teler-Msg", "X-Teler-Threat", "X-Teler-Req-Id")
 	setCustomHeader(w, xTelerMsg, msg)
 	setCustomHeader(w, xTelerThreat, k.String())
+	setCustomHeader(w, xTelerReqId, id)
 
 	// Send the logs
 	t.sendLogs(r, k, id, msg)

utils.go

@@ -211,14 +211,11 @@ func setCustomHeader(w http.ResponseWriter, key, value string) {
 	w.Header().Set(key, value)
 }
 
-// setReqIdHeader to set teler request ID header response and return it
-func setReqIdHeader(w http.ResponseWriter) string {
+// getUID to get unique ID
+func getUID() string {
 	// Generate a unique ID using the gouid package.
 	id := gouid.Bytes(10)
 
-	// Set the "X-Teler-Req-Id" header in the response with the unique ID.
-	setCustomHeader(w, xTelerReqId, id.String())
-
 	return id.String()
 }