NotificationsYou must be signed in to change notification settings
Fork1k
Star11.2k

Commit57f619c

committed

implement dual barrier for notifications command

1 parent3410c76 commit57f619cCopy full SHA for 57f619c

File tree

1 file changed

+44

-17

lines changed

cli
- exp_scaletest.go

1 file changed

+44

-17

lines changed

`‎cli/exp_scaletest.go‎`

Lines changed: 44 additions & 17 deletions

Original file line number	Diff line number	Diff line change
`@@ -2013,8 +2013,10 @@ func (r RootCmd) scaletestNotifications() serpent.Command {`
`2013`	`2013`
`2014`	`2014`	`_,_=fmt.Fprintln(inv.Stderr,"Creating users...")`
`2015`	`2015`
`2016`		`-dialBarrier:=&sync.WaitGroup{}`
`2017`		`-dialBarrier.Add(int(userCount))`
	`2016`	`+ownerDialBarrier:=&sync.WaitGroup{}`
	`2017`	`+regularDialBarrier:=&sync.WaitGroup{}`
	`2018`	`+ownerDialBarrier.Add(int(ownerUserCount))`
	`2019`	`+regularDialBarrier.Add(int(regularUserCount))`
`2018`	`2020`
`2019`	`2021`	`configs:=make([]notifications.Config,0,userCount)`
`2020`	`2022`	`forrangeownerUserCount {`
`@@ -2025,7 +2027,7 @@ func (r RootCmd) scaletestNotifications() serpent.Command {`
`2025`	`2027`	`IsOwner:true,`
`2026`	`2028`	`NotificationTimeout:notificationTimeout,`
`2027`	`2029`	`DialTimeout:dialTimeout,`
`2028`		`-DialBarrier:dialBarrier,`
	`2030`	`+DialBarrier:ownerDialBarrier,`
`2029`	`2031`	`Metrics:metrics,`
`2030`	`2032`	`}`
`2031`	`2033`	`iferr:=config.Validate();err!=nil {`
`@@ -2041,7 +2043,8 @@ func (r RootCmd) scaletestNotifications() serpent.Command {`
`2041`	`2043`	`IsOwner:false,`
`2042`	`2044`	`NotificationTimeout:notificationTimeout,`
`2043`	`2045`	`DialTimeout:dialTimeout,`
`2044`		`-DialBarrier:dialBarrier,`
	`2046`	`+DialBarrier:regularDialBarrier,`
	`2047`	`+OwnerDialBarrier:ownerDialBarrier,`
`2045`	`2048`	`Metrics:metrics,`
`2046`	`2049`	`}`
`2047`	`2050`	`iferr:=config.Validate();err!=nil {`
`@@ -2051,31 +2054,55 @@ func (r RootCmd) scaletestNotifications() serpent.Command {`
`2051`	`2054`	`}`
`2052`	`2055`
`2053`	`2056`	`gofunc() {`
`2054`		`-logger.Info(ctx,"waiting forall users to connect")`
	`2057`	`+logger.Info(ctx,"waiting forowner users to connect")`
`2055`	`2058`
`2056`		`-//Create timeout contextforthe entire trigger operation`
`2057`		`-waitCtx,cancel:=context.WithTimeout(ctx,dialTimeout+30*time.Second)`
	`2059`	`+//Waitforowner users to connect`
	`2060`	`+ownerWaitCtx,cancel:=context.WithTimeout(ctx,dialTimeout+30*time.Second)`
`2058`	`2061`	`defercancel()`
`2059`	`2062`
`2060`		`-// Wait for all runners to reach the barrier`
`2061`		`-done:=make(chanstruct{})`
	`2063`	`+ownerDone:=make(chanstruct{})`
`2062`	`2064`	`gofunc() {`
`2063`		`-dialBarrier.Wait()`
`2064`		`-close(done)`
	`2065`	`+ownerDialBarrier.Wait()`
	`2066`	`+close(ownerDone)`
`2065`	`2067`	`}()`
`2066`	`2068`
`2067`	`2069`	`select {`
`2068`		`-case<-done:`
`2069`		`-logger.Info(ctx,"all users connected")`
`2070`		`-case<-waitCtx.Done():`
`2071`		`-ifwaitCtx.Err()==context.DeadlineExceeded {`
`2072`		`-logger.Error(ctx,"timeout waiting forall users to connect")`
	`2070`	`+case<-ownerDone:`
	`2071`	`+logger.Info(ctx,"allownerusers connected")`
	`2072`	`+case<-ownerWaitCtx.Done():`
	`2073`	`+ifownerWaitCtx.Err()==context.DeadlineExceeded {`
	`2074`	`+logger.Error(ctx,"timeout waiting forowner users to connect")`
`2073`	`2075`	`}else {`
`2074`		`-logger.Info(ctx,"context canceled while waiting for users")`
	`2076`	`+logger.Info(ctx,"context canceled while waiting forownerusers")`
`2075`	`2077`	`}`
`2076`	`2078`	`return`
`2077`	`2079`	`}`
`2078`	`2080`
	`2081`	`+// Wait for regular users to connect`
	`2082`	`+logger.Info(ctx,"waiting for regular users to connect")`
	`2083`	`+regularWaitCtx,cancel:=context.WithTimeout(ctx,dialTimeout+30*time.Second)`
	`2084`	`+defercancel()`
	`2085`	`+`
	`2086`	`+regularDone:=make(chanstruct{})`
	`2087`	`+gofunc() {`
	`2088`	`+regularDialBarrier.Wait()`
	`2089`	`+close(regularDone)`
	`2090`	`+}()`
	`2091`	`+`
	`2092`	`+select {`
	`2093`	`+case<-regularDone:`
	`2094`	`+logger.Info(ctx,"all regular users connected")`
	`2095`	`+case<-regularWaitCtx.Done():`
	`2096`	`+ifregularWaitCtx.Err()==context.DeadlineExceeded {`
	`2097`	`+logger.Error(ctx,"timeout waiting for regular users to connect")`
	`2098`	`+}else {`
	`2099`	`+logger.Info(ctx,"context canceled while waiting for regular users")`
	`2100`	`+}`
	`2101`	`+return`
	`2102`	`+}`
	`2103`	`+`
	`2104`	`+logger.Info(ctx,"all users connected, triggering notifications")`
	`2105`	`+`
`2079`	`2106`	`const (`
`2080`	`2107`	`triggerUsername="scaletest-trigger-user"`
`2081`	`2108`	`triggerEmail="scaletest-trigger@example.com"`

0 commit comments

Comments

(0)

Movatterモバイル変換

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit57f619c

File tree

1 file changed

1 file changed

`‎cli/exp_scaletest.go‎`

0 commit comments