mlogclub
c246b85a9e
- Implemented request ID handling in various services and handlers to improve traceability of requests. - Added new AuthOptions endpoint to expose WxWork and OIDC configuration options. - Updated message and event logging to include request ID for better debugging. - Enhanced login form to dynamically show available authentication options based on server configuration. - Introduced utility functions for normalizing and ensuring valid request IDs. - Updated tests to verify request ID functionality in message sending and event logging.
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package tracex
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
RequestIDHeader = "X-Request-Id"
|
|
GinRequestIDKey = "requestId"
|
|
)
|
|
|
|
type requestIDContextKey struct{}
|
|
|
|
func NormalizeRequestID(value string) string {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" || len(value) > 128 {
|
|
return ""
|
|
}
|
|
for _, r := range value {
|
|
if r < 33 || r > 126 {
|
|
return ""
|
|
}
|
|
}
|
|
return value
|
|
}
|
|
|
|
func EnsureRequestID(value string) string {
|
|
if normalized := NormalizeRequestID(value); normalized != "" {
|
|
return normalized
|
|
}
|
|
var b [16]byte
|
|
if _, err := rand.Read(b[:]); err != nil {
|
|
return ""
|
|
}
|
|
return hex.EncodeToString(b[:])
|
|
}
|
|
|
|
func ContextWithRequestID(ctx context.Context, requestID string) context.Context {
|
|
requestID = NormalizeRequestID(requestID)
|
|
if requestID == "" {
|
|
return ctx
|
|
}
|
|
return context.WithValue(ctx, requestIDContextKey{}, requestID)
|
|
}
|
|
|
|
func RequestIDFromContext(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
if value, ok := ctx.Value(requestIDContextKey{}).(string); ok {
|
|
return NormalizeRequestID(value)
|
|
}
|
|
return ""
|
|
}
|