sctx.go 365 B

12345678910111213141516171819202122
  1. package sctx
  2. import "context"
  3. type ContextKey int
  4. const (
  5. HTTPRequestIDKey ContextKey = iota
  6. requestHostKey
  7. )
  8. func SetHost(ctx context.Context, domain string) context.Context {
  9. return context.WithValue(ctx, requestHostKey, domain)
  10. }
  11. func GetHost(ctx context.Context) string {
  12. v, ok := ctx.Value(requestHostKey).(string)
  13. if ok {
  14. return v
  15. }
  16. return ""
  17. }