feat: add server custom options

jrpc/options.go

@@ -0,0 +1,31 @@
+package jrpc
+
+import "time"
+
+type jrpcOpt struct {
+	CustomHeadersKey  []string
+	ReadHeaderTimeout time.Duration
+}
+
+type Option func(*jrpcOpt)
+
+func defaultOpt() *jrpcOpt {
+	return &jrpcOpt{
+		ReadHeaderTimeout: 3 * time.Second,
+		CustomHeadersKey:  []string{"Authorization"},
+	}
+}
+
+// WithCustomHeaders add custom headers to metadata.MD
+func WithCustomHeaders(headersKey ...string) Option {
+	return func(opt *jrpcOpt) {
+		opt.CustomHeadersKey = append(opt.CustomHeadersKey, headersKey...)
+	}
+}
+
+// WithReadHeaderTimeout set custom read header timeout
+func WithReadHeaderTimeout(timeout time.Duration) Option {
+	return func(opt *jrpcOpt) {
+		opt.ReadHeaderTimeout = timeout
+	}
+}

jrpc/options_test.go

@@ -0,0 +1,25 @@
+package jrpc
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+	"time"
+)
+
+func TestOptions(t *testing.T) {
+	dur := 5 * time.Second
+
+	opts := []Option{
+		WithCustomHeaders("foobar"),
+		WithReadHeaderTimeout(dur),
+	}
+
+	def := defaultOpt()
+
+	for _, opt := range opts {
+		opt(def)
+	}
+
+	assert.Len(t, def.CustomHeadersKey, 2)
+	assert.Equal(t, def.ReadHeaderTimeout.String(), dur.String())
+}