add server Context option to pass own context

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
ajunlonglive · Dec 5, 2019 · 1c5a4c4 · 1c5a4c4
1 parent 18cf025
commit 1c5a4c4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/server/options.go b/server/options.go
@@ -136,6 +136,15 @@ func Codec(contentType string, c codec.NewCodec) Option {
 	}
 }
 
+// Context specifies a context for the service.
+// Can be used to signal shutdown of the service
+// Can be used for extra option values.
+func Context(ctx context.Context) Option {
+	return func(o *Options) {
+		o.Context = ctx
+	}
+}
+
 // Registry used for discovery
 func Registry(r registry.Registry) Option {
 	return func(o *Options) {

diff --git a/web/options.go b/web/options.go
@@ -126,18 +126,21 @@ func Context(ctx context.Context) Option {
 	}
 }
 
+// Registry used for discovery
 func Registry(r registry.Registry) Option {
 	return func(o *Options) {
 		o.Registry = r
 	}
 }
 
+// Register the service with a TTL
 func RegisterTTL(t time.Duration) Option {
 	return func(o *Options) {
 		o.RegisterTTL = t
 	}
 }
 
+// Register the service with at interval
 func RegisterInterval(t time.Duration) Option {
 	return func(o *Options) {
 		o.RegisterInterval = t