Function sendLogsV2

log/coder.go:146–184

sendLogsV2 uses the v2 agent API to send logs. Only compatibile with coder versions >= 2.9.

(ctx context.Context, dest agentsdk.LogDest, ls coderLogSender, l slog.Logger)

// sendLogsV2 uses the v2 agent API to send logs. Only compatibile with coder versions >= 2.9.
func sendLogsV2(ctx context.Context, dest agentsdk.LogDest, ls coderLogSender, l slog.Logger) (logger Func, closer func()) {
	sendCtx, sendCancel := context.WithCancel(ctx)
	done := make(chan struct{})
	uid := uuid.New()
	go func() {
		defer close(done)
		if err := ls.SendLoop(sendCtx, dest); err != nil {
			if !errors.Is(err, context.Canceled) {
				l.Warn(ctx, "failed to send logs to Coder", slog.Error(err))
			}
		}
	}()

	var closeOnce sync.Once
	return func(l Level, msg string, args ...any) {
			ls.Enqueue(uid, agentsdk.Log{
				CreatedAt: time.Now(),
				Output:    fmt.Sprintf(msg, args...),
				Level:     codersdk.LogLevel(l),
			})
		}, func() {
			closeOnce.Do(func() {
				// Trigger a flush and wait for logs to be sent.
				ls.Flush(uid)
				ctx, cancel := context.WithTimeout(ctx, logSendGracePeriod)
				defer cancel()
				err := ls.WaitUntilEmpty(ctx)
				if err != nil {
					l.Warn(ctx, "log sender did not empty", slog.Error(err))
				}

				// Stop the send loop.
				sendCancel()
			})

			// Wait for the send loop to finish.
			<-done
		}
}