Commit77d3e64

committed

Add WaitForEnvironmentReady utility

1 parentba5d4b8 commit77d3e64Copy full SHA for 77d3e64

File tree

+26

-0

lines changed

+26

-0

lines changed

Lines changed: 26 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -2,11 +2,13 @@ package coder`
`2`	`2`
`3`	`3`	`import (`
`4`	`4`	`"context"`
	`5`	`+"fmt"`
`5`	`6`	`"net/http"`
`6`	`7`	`"time"`
`7`	`8`
`8`	`9`	`"cdr.dev/coder-cli/internal/x/xjson"`
`9`	`10`	`"nhooyr.io/websocket"`
	`11`	`+"nhooyr.io/websocket/wsjson"`
`10`	`12`	`)`
`11`	`13`
`12`	`14`	`// Environment describes a Coder environment`
`@@ -135,3 +137,27 @@ func (c Client) DialEnvironmentStats(ctx context.Context, envID string) (*websoc`
`135`	`137`	`func (cClient)DialResourceLoad(ctx context.Context,envIDstring) (*websocket.Conn,error) {`
`136`	`138`	`returnc.dialWs(ctx,"/api/environments/"+envID+"/watch-resource-load")`
`137`	`139`	`}`
	`140`	`+`
	`141`	`+typebuildLogMsgstruct {`
	`142`	+Typestring`json:"type"`
	`143`	`+}`
	`144`	`+`
	`145`	`+// WaitForEnvironmentReady will watch the build log and return when done`
	`146`	`+func (cClient)WaitForEnvironmentReady(ctx context.Context,env*Environment)error {`
	`147`	`+conn,err:=c.DialEnvironmentBuildLog(ctx,env.ID)`
	`148`	`+iferr!=nil {`
	`149`	`+returnfmt.Errorf("%s: dial build log: %w",env.Name,err)`
	`150`	`+}`
	`151`	`+`
	`152`	`+for {`
	`153`	`+msg:=buildLogMsg{}`
	`154`	`+err:=wsjson.Read(ctx,conn,&msg)`
	`155`	`+iferr!=nil {`
	`156`	`+returnfmt.Errorf("%s: reading build log msg: %w",env.Name,err)`
	`157`	`+}`
	`158`	`+`
	`159`	`+ifmsg.Type=="done" {`
	`160`	`+returnnil`
	`161`	`+}`
	`162`	`+}`
	`163`	`+}`

Comments

(0)