Refactor

server/run_option.go

@@ -0,0 +1,21 @@
+package server
+
+import (
+	"context"
+)
+
+// RunOptions holds a set of configurations for Server.Run().
+type RunOptions struct {
+	gracefullCtx context.Context
+}
+
+// RunOption is an option of Server.Run().
+type RunOption func(*RunOptions)
+
+// WithGracefullContext accepts a context to shutdown a Server
+// with care for existing client connections.
+func WithGracefullContext(ctx context.Context) RunOption {
+	return func(options *RunOptions) {
+		options.gracefullCtx = ctx
+	}
+}