etcd: world: Make sure to check if wg is nil or not

We never hit this as far as I know, but might as well be safe.
This commit is contained in:
James Shubin
2023-08-07 23:11:20 -04:00
parent 5d84e33be7
commit 3f403d34a4

View File

@@ -74,9 +74,14 @@ func (obj *World) IdealClusterSizeWatch(ctx context.Context) (chan error, error)
if err := c.Init(); err != nil {
return nil, err
}
util.WgFromCtx(ctx).Add(1)
wg := util.WgFromCtx(ctx)
if wg != nil {
wg.Add(1)
}
go func() {
defer util.WgFromCtx(ctx).Done()
if wg != nil {
defer wg.Done()
}
// This must get closed *after* because it will not finish until
// the Watcher returns, because it contains a wg.Wait() in it...
defer c.Close() // ignore error