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.
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package httpx
|
|
|
|
import (
|
|
"cs-agent/internal/pkg/httpx/params"
|
|
"cs-agent/internal/pkg/openidentity"
|
|
"cs-agent/internal/pkg/tracex"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/mlogclub/simple/common/strs"
|
|
)
|
|
|
|
const (
|
|
ctxKeyExternalUser = "externalUser"
|
|
)
|
|
|
|
func SetExternalUser(ctx *gin.Context, ext *openidentity.ExternalUser) {
|
|
ctx.Set(ctxKeyExternalUser, ext)
|
|
}
|
|
|
|
func GetExternalUser(ctx *gin.Context) *openidentity.ExternalUser {
|
|
v, _ := ctx.Get(ctxKeyExternalUser)
|
|
ext, _ := v.(*openidentity.ExternalUser)
|
|
return ext
|
|
}
|
|
|
|
func GetChannelID(ctx *gin.Context) string {
|
|
if channelID := ctx.GetHeader("X-Channel-ID"); strs.IsNotBlank(channelID) {
|
|
return channelID
|
|
}
|
|
if channelID, _ := params.Get(ctx, "channelId"); strs.IsNotBlank(channelID) {
|
|
return channelID
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func GetRequestID(ctx *gin.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
if value, ok := ctx.Get(tracex.GinRequestIDKey); ok {
|
|
if requestID, ok := value.(string); ok {
|
|
return tracex.NormalizeRequestID(requestID)
|
|
}
|
|
}
|
|
return tracex.NormalizeRequestID(ctx.GetHeader(tracex.RequestIDHeader))
|
|
}
|