Commit6eb1887

authored

Update last connection time on tunnel connections (#401)

* Update last connection time on tunnnel connections* fmt

1 parente641ac5 commit6eb1887Copy full SHA for 6eb1887

File tree

+47

-0

lines changed

+47

-0

lines changed

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,22 @@`
	`1`	`+package coder`
	`2`	`+`
	`3`	`+import (`
	`4`	`+"context"`
	`5`	`+"fmt"`
	`6`	`+"net/http"`
	`7`	`+)`
	`8`	`+`
	`9`	`+// UpdateLastConnectionAt updates the last connection at attribute of a workspace.`
	`10`	`+func (c*DefaultClient)UpdateLastConnectionAt(ctx context.Context,workspaceIDstring)error {`
	`11`	`+reqURL:=fmt.Sprintf("/api/private/envagent/%s/update-last-connection-at",workspaceID)`
	`12`	`+resp,err:=c.request(ctx,http.MethodPost,reqURL,nil)`
	`13`	`+iferr!=nil {`
	`14`	`+returnerr`
	`15`	`+}`
	`16`	`+`
	`17`	`+ifresp.StatusCode!=http.StatusOK {`
	`18`	`+returnNewHTTPError(resp)`
	`19`	`+}`
	`20`	`+`
	`21`	`+returnnil`
	`22`	`+}`

Lines changed: 3 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -250,4 +250,7 @@ type Client interface {`
`250`	`250`
`251`	`251`	`// DeleteSatelliteByID deletes a satellite entity from the Coder control plane.`
`252`	`252`	`DeleteSatelliteByID(ctx context.Context,idstring)error`
	`253`	`+`
	`254`	`+// UpdateLastConnectionAt updates the last connection at attribute of a workspace.`
	`255`	`+UpdateLastConnectionAt(ctx context.Context,workspaceIDstring)error`
`253`	`256`	`}`

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,7 @@ import (`
`8`	`8`	`"net/url"`
`9`	`9`	`"os"`
`10`	`10`	`"strconv"`
	`11`	`+"time"`
`11`	`12`
`12`	`13`	`"cdr.dev/slog"`
`13`	`14`	`"cdr.dev/slog/sloggers/sloghuman"`
`@@ -130,6 +131,27 @@ func (c *tunnneler) start(ctx context.Context) error {`
`130`	`131`	`}`
`131`	`132`	`c.log.Debug(ctx,"Connected to workspace!")`
`132`	`133`
	`134`	`+sdk,err:=newClient(ctx,false)`
	`135`	`+iferr!=nil {`
	`136`	`+returnxerrors.Errorf("getting coder client: %w",err)`
	`137`	`+}`
	`138`	`+`
	`139`	`+// regularly update the last connection at`
	`140`	`+gofunc() {`
	`141`	`+ticker:=time.NewTicker(time.Minute)`
	`142`	`+deferticker.Stop()`
	`143`	`+`
	`144`	`+for {`
	`145`	`+select {`
	`146`	`+case<-ctx.Done():`
	`147`	`+return`
	`148`	`+case<-ticker.C:`
	`149`	`+// silently ignore failures so we don't spam the console`
	`150`	`+_=sdk.UpdateLastConnectionAt(ctx,c.workspaceID)`
	`151`	`+}`
	`152`	`+}`
	`153`	`+}()`
	`154`	`+`
`133`	`155`	`// proxy via stdio`
`134`	`156`	`ifc.stdio {`
`135`	`157`	`gofunc() {`

Comments

(0)