generated from TBD54566975/tbd-project-template
-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor: Move timeline log sink under timelineclient package (#4774)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
- Loading branch information
1 parent
f2f94d8
commit f7d5881
Showing
4 changed files
with
90 additions
and
73 deletions.
There are no files selected for viewing
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,85 @@ | ||
| package timelineclient | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
|
|
||
| "github.com/alecthomas/types/optional" | ||
|
|
||
| "github.com/block/ftl/internal/channels" | ||
| "github.com/block/ftl/internal/key" | ||
| "github.com/block/ftl/internal/log" | ||
| ) | ||
|
|
||
| // LogSink is a log sink that sends logs to the timeline client. | ||
| // | ||
| // It needs to be run in a separate goroutine after creation by calling RunLogLoop. | ||
| type LogSink struct { | ||
| client *Client | ||
| logQueue chan log.Entry | ||
| level log.Level | ||
| } | ||
|
|
||
| var _ log.Sink = (*LogSink)(nil) | ||
|
|
||
| func NewLogSink(client *Client, level log.Level) *LogSink { | ||
| return &LogSink{ | ||
| client: client, | ||
| logQueue: make(chan log.Entry, 10000), | ||
| level: level, | ||
| } | ||
| } | ||
|
|
||
| // Log implements Sink | ||
| func (l *LogSink) Log(entry log.Entry) error { | ||
| if entry.Level < l.level { | ||
| return nil | ||
| } | ||
|
|
||
| select { | ||
| case l.logQueue <- entry: | ||
| default: | ||
| // Drop log entry if queue is full | ||
| return fmt.Errorf("log queue is full") | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| // RunLogLoop runs the log loop. | ||
| // | ||
| // It will run until the context is cancelled. | ||
| func (l *LogSink) RunLogLoop(ctx context.Context) { | ||
| for entry := range channels.IterContext(ctx, l.logQueue) { | ||
| dep, ok := entry.Attributes["deployment"] | ||
| var deploymentKey key.Deployment | ||
| var err error | ||
| if ok { | ||
| deploymentKey, err = key.ParseDeploymentKey(dep) | ||
| if err != nil { | ||
| continue | ||
| } | ||
| } | ||
|
|
||
| var errorString *string | ||
| if entry.Error != nil { | ||
| errStr := entry.Error.Error() | ||
| errorString = &errStr | ||
| } | ||
| var request optional.Option[key.Request] | ||
| if reqStr, ok := entry.Attributes["request"]; ok { | ||
| req, err := key.ParseRequestKey(reqStr) //nolint:errcheck // best effort | ||
| if err == nil { | ||
| request = optional.Some(req) | ||
| } | ||
| } | ||
| l.client.Publish(ctx, &Log{ | ||
| DeploymentKey: deploymentKey, | ||
| RequestKey: request, | ||
| Level: int32(entry.Level), //nolint:gosec | ||
| Time: entry.Time, | ||
| Attributes: entry.Attributes, | ||
| Message: entry.Message, | ||
| Error: optional.Ptr(errorString), | ||
| }) | ||
| } | ||
| } |