github.com/maeglindeveloper/gqlgen@v0.13.1-0.20210413081235-57808b12a0a0/graphql/handler/transport/websocket_init.go (about)

     1  package transport
     2  
     3  import "context"
     4  
     5  type key string
     6  
     7  const (
     8  	initpayload key = "ws_initpayload_context"
     9  )
    10  
    11  // InitPayload is a structure that is parsed from the websocket init message payload. TO use
    12  // request headers for non-websocket, instead wrap the graphql handler in a middleware.
    13  type InitPayload map[string]interface{}
    14  
    15  // GetString safely gets a string value from the payload. It returns an empty string if the
    16  // payload is nil or the value isn't set.
    17  func (p InitPayload) GetString(key string) string {
    18  	if p == nil {
    19  		return ""
    20  	}
    21  
    22  	if value, ok := p[key]; ok {
    23  		res, _ := value.(string)
    24  		return res
    25  	}
    26  
    27  	return ""
    28  }
    29  
    30  // Authorization is a short hand for getting the Authorization header from the
    31  // payload.
    32  func (p InitPayload) Authorization() string {
    33  	if value := p.GetString("Authorization"); value != "" {
    34  		return value
    35  	}
    36  
    37  	if value := p.GetString("authorization"); value != "" {
    38  		return value
    39  	}
    40  
    41  	return ""
    42  }
    43  
    44  func withInitPayload(ctx context.Context, payload InitPayload) context.Context {
    45  	return context.WithValue(ctx, initpayload, payload)
    46  }
    47  
    48  // GetInitPayload gets a map of the data sent with the connection_init message, which is used by
    49  // graphql clients as a stand-in for HTTP headers.
    50  func GetInitPayload(ctx context.Context) InitPayload {
    51  	payload, ok := ctx.Value(initpayload).(InitPayload)
    52  	if !ok {
    53  		return nil
    54  	}
    55  
    56  	return payload
    57  }