| 377 | } |
| 378 | |
| 379 | func (r *Runner) Cleanup(ctx context.Context, id string, logs io.Writer) error { |
| 380 | if r.chatID == uuid.Nil { |
| 381 | return nil |
| 382 | } |
| 383 | |
| 384 | logs = loadtestutil.NewSyncWriter(logs) |
| 385 | logger := slog.Make(sloghuman.Sink(logs)).Leveled(slog.LevelDebug).Named(id).With(slog.F("chat_id", r.chatID)) |
| 386 | r.client.SetLogger(logger) |
| 387 | r.client.SetLogBodies(true) |
| 388 | |
| 389 | archived := true |
| 390 | logger.Info(ctx, "archiving chat session") |
| 391 | if err := r.client.UpdateChat(ctx, r.chatID, codersdk.UpdateChatRequest{Archived: &archived}); err != nil { |
| 392 | logger.Error(ctx, "failed to archive chat", slog.Error(err)) |
| 393 | return xerrors.Errorf("archive chat: %w", err) |
| 394 | } |
| 395 | logger.Info(ctx, "archived chat session") |
| 396 | return nil |
| 397 | } |
| 398 | |
| 399 | func (r *Runner) GetMetrics() map[string]any { |
| 400 | return map[string]any{ |