You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
crate/logotel/processor.go

102 lines
2.9 KiB
Go

// otellog 提供 git.blauwelle.com/go/crate/log 的 opentelemetry 处理功能
package logotel
import (
"context"
"fmt"
"strconv"
"git.blauwelle.com/go/crate/log/logsdk"
"git.blauwelle.com/go/crate/log/logsdk/logjson"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/codes"
semconv "go.opentelemetry.io/otel/semconv/v1.17.0"
"go.opentelemetry.io/otel/trace"
)
// New 创建 log/opentelemetry 处理器
func New(opts ...Option) *Processor {
cfg := newConfig(opts...)
return &Processor{
bufferPool: cfg.bytesBufferPool,
defaultSpan: cfg.defaultSpan,
}
}
var _ logsdk.EntryProcessor = &Processor{}
// Processor 用于把日志和 opentelemetry 对接
type Processor struct {
bufferPool logjson.BytesBufferPool
defaultSpan bool
}
func (processor *Processor) Process(ctx context.Context, entry logsdk.ReadonlyEntry) {
span := trace.SpanFromContext(ctx)
if !span.IsRecording() {
if processor.defaultSpan {
ctx, span = otel.Tracer("git.blauwelle.com/go/crate/logotel").Start(ctx, "default") //nolint:ineffassign,staticcheck,wastedassign
defer span.End()
}
}
if !span.IsRecording() {
return
}
const entryInitSize = 6
attrs := make([]attribute.KeyValue, 0, len(entry.Fields)+entryInitSize)
attrs = append(attrs, attribute.String("log.severity", entry.Level.String()))
attrs = append(attrs, attribute.String("log.message", entry.Message))
if entry.Caller.IsValid() {
attrs = append(attrs, semconv.CodeFunctionKey.String(entry.Caller.Function))
attrs = append(attrs, semconv.CodeFilepathKey.String(entry.Caller.File))
attrs = append(attrs, semconv.CodeLineNumberKey.Int(entry.Caller.Line))
}
if len(entry.Stack) > 0 {
buf := processor.bufferPool.Get()
for _, frame := range entry.Stack {
buf.WriteString(frame.Function)
buf.WriteByte('\n')
buf.WriteByte('\t')
buf.WriteString(frame.File)
buf.WriteByte(':')
buf.WriteString(strconv.Itoa(frame.Line))
buf.WriteByte('\n')
}
processor.bufferPool.Put(buf)
attrs = append(attrs, attribute.String("zz.stack", buf.String()))
}
for _, field := range entry.Fields {
attrs = append(attrs, fieldToKV(field))
}
span.AddEvent("log", trace.WithTimestamp(entry.Time), trace.WithAttributes(attrs...))
if entry.Level <= logsdk.LevelError {
span.SetStatus(codes.Error, entry.Message)
}
}
func fieldToKV(field logsdk.KV) attribute.KeyValue {
switch value := field.Value.(type) {
case nil:
return attribute.String(field.Key, "<nil>")
case string:
return attribute.String(field.Key, value)
case int:
return attribute.Int(field.Key, value)
case int64:
return attribute.Int64(field.Key, value)
case float64:
return attribute.Float64(field.Key, value)
case bool:
return attribute.Bool(field.Key, value)
case error:
return attribute.String(field.Key, value.Error())
case fmt.Stringer:
return attribute.String(field.Key, value.String())
}
return attribute.String(field.Key, fmt.Sprint(field.Value))
}