dbmem

ethanndickson · ethanndickson · commit 6bdf3be3fc9a · 2025-06-13T13:54:30.000Z
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -12277,16 +12277,17 @@ func (q *FakeQuerier) UpsertConnectionLog(_ context.Context, arg database.Upsert
 	q.mutex.Lock()
 	defer q.mutex.Unlock()
 
-	if arg.ConnectionAction == "disconnect" {
-		for i, existing := range q.connectionLogs {
-			if existing.ConnectionID == arg.ConnectionID &&
-				existing.WorkspaceID == arg.WorkspaceID &&
-				existing.AgentName == arg.AgentName {
-				// Update existing connection with close time and reason
-				q.connectionLogs[i].CloseTime = sql.NullTime{Valid: true, Time: arg.Time}
-				q.connectionLogs[i].CloseReason = arg.CloseReason
+	for i, existing := range q.connectionLogs {
+		if existing.ConnectionID == arg.ConnectionID &&
+			existing.WorkspaceID == arg.WorkspaceID &&
+			existing.AgentName == arg.AgentName {
+			if arg.ConnectionAction != "disconnect" {
 				return q.connectionLogs[i], nil
 			}
+			// Update existing connection with close time and reason
+			q.connectionLogs[i].CloseTime = sql.NullTime{Valid: true, Time: arg.Time}
+			q.connectionLogs[i].CloseReason = arg.CloseReason
+			return q.connectionLogs[i], nil
 		}
 	}
 
