context.go

package jsonrpc

import (
	"context"

	"github.com/goccy/go-json"
)

type (
	requestIDKey  struct{}
	metadataIDKey struct{}
	methodNameKey struct{}
)

// RequestID takes request id from context.
func RequestID(c context.Context) *json.RawMessage {
	v, _ := c.Value(requestIDKey{}).(*json.RawMessage)

	return v
}

// WithRequestID adds request id to context.
func WithRequestID(c context.Context, id *json.RawMessage) context.Context {
	return context.WithValue(c, requestIDKey{}, id)
}

// GetMetadata takes jsonrpc metadata from context.
func GetMetadata(c context.Context) Metadata {
	v, _ := c.Value(metadataIDKey{}).(Metadata)

	return v
}

// WithMetadata adds jsonrpc metadata to context.
func WithMetadata(c context.Context, md Metadata) context.Context {
	return context.WithValue(c, metadataIDKey{}, md)
}

// MethodName takes method name from context.
func MethodName(c context.Context) string {
	v, _ := c.Value(methodNameKey{}).(string)

	return v
}

// WithMethodName adds method name to context.
func WithMethodName(c context.Context, name string) context.Context {
	return context.WithValue(c, methodNameKey{}, name)
}