Update last connection time on tunnel connections (#401)

f0ssel · web-flow · commit 6eb1887cd421 · 2021-07-26T12:59:59.000-05:00
* Update last connection time on tunnnel connections

* fmt
diff --git a/coder-sdk/agent.go b/coder-sdk/agent.go
@@ -0,0 +1,22 @@
+package coder
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+)
+
+// UpdateLastConnectionAt updates the last connection at attribute of a workspace.
+func (c *DefaultClient) UpdateLastConnectionAt(ctx context.Context, workspaceID string) error {
+	reqURL := fmt.Sprintf("/api/private/envagent/%s/update-last-connection-at", workspaceID)
+	resp, err := c.request(ctx, http.MethodPost, reqURL, nil)
+	if err != nil {
+		return err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return NewHTTPError(resp)
+	}
+
+	return nil
+}
diff --git a/coder-sdk/interface.go b/coder-sdk/interface.go
@@ -250,4 +250,7 @@ type Client interface {
 
 	// DeleteSatelliteByID deletes a satellite entity from the Coder control plane.
 	DeleteSatelliteByID(ctx context.Context, id string) error
+
+	// UpdateLastConnectionAt updates the last connection at attribute of a workspace.
+	UpdateLastConnectionAt(ctx context.Context, workspaceID string) error
 }
diff --git a/internal/cmd/tunnel.go b/internal/cmd/tunnel.go
@@ -8,6 +8,7 @@ import (
 	"net/url"
 	"os"
 	"strconv"
+	"time"
 
 	"cdr.dev/slog"
 	"cdr.dev/slog/sloggers/sloghuman"
@@ -130,6 +131,27 @@ func (c *tunnneler) start(ctx context.Context) error {
 	}
 	c.log.Debug(ctx, "Connected to workspace!")
 
+	sdk, err := newClient(ctx, false)
+	if err != nil {
+		return xerrors.Errorf("getting coder client: %w", err)
+	}
+
+	// regularly update the last connection at
+	go func() {
+		ticker := time.NewTicker(time.Minute)
+		defer ticker.Stop()
+
+		for {
+			select {
+			case <-ctx.Done():
+				return
+			case <-ticker.C:
+				// silently ignore failures so we don't spam the console
+				_ = sdk.UpdateLastConnectionAt(ctx, c.workspaceID)
+			}
+		}
+	}()
+
 	// proxy via stdio
 	if c.stdio {
 		go func() {

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, id string) error`
	`253`	`+`
	`254`	`+ // UpdateLastConnectionAt updates the last connection at attribute of a workspace.`
	`255`	`+ UpdateLastConnectionAt(ctx context.Context, workspaceID string) error`
`253`	`256`	`}`