mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
package telemetrylogger
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/mattermost/mattermost/server/public/pluginapi/experimental/bot/logger"
|
|
"github.com/mattermost/mattermost/server/public/pluginapi/experimental/common"
|
|
"github.com/mattermost/mattermost/server/public/pluginapi/experimental/telemetry"
|
|
)
|
|
|
|
type telemetryLogger struct {
|
|
logger.Logger
|
|
logLevel logger.LogLevel
|
|
tracker telemetry.Tracker
|
|
}
|
|
|
|
/*
|
|
New promotes the provided logger into a telemetry logger, storing all events below the logLevel
|
|
through the tracker.
|
|
|
|
- l Logger: A logger to promote.
|
|
|
|
- logLevel: The highest type of message to be stored in telemetry.
|
|
|
|
- tracker: The telemetry tracker to store the messages.
|
|
*/
|
|
func New(l logger.Logger, logLevel logger.LogLevel, tracker telemetry.Tracker) logger.Logger {
|
|
return &telemetryLogger{
|
|
Logger: l,
|
|
logLevel: logLevel,
|
|
tracker: tracker,
|
|
}
|
|
}
|
|
|
|
// NewFromAPI creates a telemetryLogger directly from a LogAPI instead of passing a logger.
|
|
func NewFromAPI(api common.LogAPI, logLevel logger.LogLevel, tracker telemetry.Tracker) logger.Logger {
|
|
return New(logger.New(api), logLevel, tracker)
|
|
}
|
|
|
|
func (l *telemetryLogger) Debugf(format string, args ...any) {
|
|
l.Logger.Debugf(format, args...)
|
|
message := fmt.Sprintf(format, args...)
|
|
if logger.Level(l.logLevel) >= 4 {
|
|
l.logToTelemetry("DEBUG", message)
|
|
}
|
|
}
|
|
|
|
func (l *telemetryLogger) Errorf(format string, args ...any) {
|
|
l.Logger.Errorf(format, args...)
|
|
message := fmt.Sprintf(format, args...)
|
|
if logger.Level(l.logLevel) >= 1 {
|
|
l.logToTelemetry("ERROR", message)
|
|
}
|
|
}
|
|
|
|
func (l *telemetryLogger) Infof(format string, args ...any) {
|
|
l.Logger.Infof(format, args...)
|
|
message := fmt.Sprintf(format, args...)
|
|
if logger.Level(l.logLevel) >= 3 {
|
|
l.logToTelemetry("INFO", message)
|
|
}
|
|
}
|
|
|
|
func (l *telemetryLogger) Warnf(format string, args ...any) {
|
|
l.Logger.Warnf(format, args...)
|
|
message := fmt.Sprintf(format, args...)
|
|
if logger.Level(l.logLevel) >= 2 {
|
|
l.logToTelemetry("WARN", message)
|
|
}
|
|
}
|
|
|
|
func (l *telemetryLogger) logToTelemetry(level, message string) {
|
|
properties := map[string]any{}
|
|
properties["message"] = message
|
|
for k, v := range l.Context() {
|
|
properties["context_"+k] = fmt.Sprintf("%v", v)
|
|
}
|
|
|
|
_ = l.tracker.TrackEvent("logger_"+level, properties)
|
|
}
|