initial

1 files changed, 39 insertions, 0 deletions

diff --git a/request_id.go b/request_id.go
new file mode 100644
index 0000000..67a0c82
--- /dev/null
+++ b/request_id.go
@@ -0,0 +1,39 @@
+package middleware
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/google/uuid"
+)
+
+type ctxKeyRequestID int
+
+const (
+	RequestIDKey    ctxKeyRequestID = 0
+	RequestIDHeader string          = "X-Request-ID"
+)
+
+func RequestID(handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		requestID := r.Header.Get(RequestIDHeader)
+		if requestID == "" {
+			requestID = uuid.NewString()
+		}
+
+		handler.ServeHTTP(w, r.WithContext(context.WithValue(r.Context(), RequestIDKey, requestID)))
+	})
+}
+
+func GetRequestID(r *http.Request) string {
+	rid := r.Context().Value(RequestIDKey)
+	if rid == nil {
+		return ""
+	}
+	srid, ok := rid.(string)
+	if !ok {
+		return ""
+	}
+
+	return srid
+}