osamingo/jsonrpc

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)
}