(ctx context.Context, event wspubsub.WorkspaceEvent, err error)
| 58 | } |
| 59 | |
| 60 | func (s *sub) handleEvent(ctx context.Context, event wspubsub.WorkspaceEvent, err error) { |
| 61 | s.mu.Lock() |
| 62 | defer s.mu.Unlock() |
| 63 | |
| 64 | switch event.Kind { |
| 65 | case wspubsub.WorkspaceEventKindStateChange: |
| 66 | case wspubsub.WorkspaceEventKindAgentConnectionUpdate: |
| 67 | case wspubsub.WorkspaceEventKindAgentTimeout: |
| 68 | case wspubsub.WorkspaceEventKindAgentLifecycleUpdate: |
| 69 | default: |
| 70 | if err == nil { |
| 71 | return |
| 72 | } |
| 73 | // Always attempt an update if the pubsub lost connection |
| 74 | s.logger.Warn(ctx, "failed to handle workspace event", slog.Error(err)) |
| 75 | } |
| 76 | |
| 77 | // Use context containing actor |
| 78 | rows, err := s.db.GetWorkspacesAndAgentsByOwnerID(s.ctx, s.userID) |
| 79 | if err != nil { |
| 80 | s.logger.Warn(ctx, "failed to get workspaces and agents by owner ID", slog.Error(err)) |
| 81 | return |
| 82 | } |
| 83 | latest := convertRows(rows) |
| 84 | |
| 85 | out, updated := produceUpdate(s.prev, latest) |
| 86 | if !updated { |
| 87 | return |
| 88 | } |
| 89 | |
| 90 | s.prev = latest |
| 91 | select { |
| 92 | case <-s.ctx.Done(): |
| 93 | return |
| 94 | case s.ch <- out: |
| 95 | } |
| 96 | } |
| 97 | |
| 98 | func (s *sub) start(ctx context.Context) (err error) { |
| 99 | rows, err := s.db.GetWorkspacesAndAgentsByOwnerID(ctx, s.userID) |
no test coverage detected