Commit3ba357f

committed

Add heartbeat

1 parent6f726cd commit3ba357fCopy full SHA for 3ba357f

File tree

+19

-0

lines changed

+19

-0

lines changed

Lines changed: 19 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -12,6 +12,7 @@ import (`
`12`	`12`	`"golang.org/x/sys/unix"`
`13`	`13`	`"golang.org/x/time/rate"`
`14`	`14`	`"golang.org/x/xerrors"`
	`15`	`+"nhooyr.io/websocket"`
`15`	`16`
`16`	`17`	`"go.coder.com/cli"`
`17`	`18`	`"go.coder.com/flog"`
`@@ -126,6 +127,7 @@ func runCommand(ctx context.Context, envName string, command string, args []stri`
`126`	`127`	`iferr!=nil {`
`127`	`128`	`returnerr`
`128`	`129`	`}`
	`130`	`+goheatbeat(ctx,conn,15*time.Second)`
`129`	`131`
`130`	`132`	`execer:=wsep.RemoteExecer(conn)`
`131`	`133`	`process,err:=execer.Start(ctx, wsep.Command{`
`@@ -172,4 +174,21 @@ func runCommand(ctx context.Context, envName string, command string, args []stri`
`172`	`174`	`returnerr`
`173`	`175`	`}`
`174`	`176`
	`177`	`+funcheatbeat(ctx context.Context,c*websocket.Conn,interval time.Duration) {`
	`178`	`+ticker:=time.NewTicker(interval)`
	`179`	`+deferticker.Stop()`
	`180`	`+`
	`181`	`+for {`
	`182`	`+select {`
	`183`	`+case<-ctx.Done():`
	`184`	`+return`
	`185`	`+case<-ticker.C:`
	`186`	`+err:=c.Ping(ctx)`
	`187`	`+iferr!=nil {`
	`188`	`+flog.Error("failed to ping websocket: %v",err)`
	`189`	`+}`
	`190`	`+}`
	`191`	`+}`
	`192`	`+}`
	`193`	`+`
`175`	`194`	`constsshActivityName="ssh"`

Comments

(0)