unistack-org/micro
4
0
Fork 3
Code Issues 17 Pull Requests 1 Actions Packages Projects Releases 258 Wiki Activity

add Wait option

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2020-09-20 15:48:07 +03:00
parent 5787a1afb8
commit 9b11ea527a
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/options.go
View File

@ -46,6 +46,7 @@ type Options struct {
// TLSConfig specifies tls.Config for secure serving // TLSConfig specifies tls.Config for secure serving
TLSConfig *tls.Config TLSConfig *tls.Config
Wait *sync.WaitGroup
// Other options for implementations of the interface // Other options for implementations of the interface
// can be stored in a context // can be stored in a context
Context context.Context Context context.Context
@ -234,13 +235,10 @@ func WithRouter(r Router) Option {
// wait against it on stop. // wait against it on stop.
func Wait(wg *sync.WaitGroup) Option { func Wait(wg *sync.WaitGroup) Option {
return func(o *Options) { return func(o *Options) {
if o.Context == nil {
o.Context = context.Background()
}
if wg == nil { if wg == nil {
wg = new(sync.WaitGroup) wg = new(sync.WaitGroup)
} }
o.Context = context.WithValue(o.Context, "wait", wg) o.Wait = wg
} }
} }