refactor: move otlp to custom file out of the scope from plug instance
This commit is contained in:
parent
117c15e371
commit
130f6928fc
31
pkg/plug/otlp.go
Normal file
31
pkg/plug/otlp.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package plug
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"git.anthrove.art/Anthrove/plug-sdk/v3/pkg/telemetry"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"go.opentelemetry.io/otel/codes"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SetupOpenTelemtry(ctx context.Context, serviceName string) error {
|
||||||
|
ctx, span := tracer.Start(ctx, "SetupOpenTelemtry")
|
||||||
|
defer span.End()
|
||||||
|
|
||||||
|
err := telemetry.SetupMeterProvider(serviceName)
|
||||||
|
if err != nil {
|
||||||
|
span.RecordError(err)
|
||||||
|
span.SetStatus(codes.Error, err.Error())
|
||||||
|
log.WithContext(ctx).WithError(err).Error("Failed to setup meter provider")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = telemetry.SetupTraceProvider(ctx, serviceName)
|
||||||
|
if err != nil {
|
||||||
|
span.RecordError(err)
|
||||||
|
span.SetStatus(codes.Error, err.Error())
|
||||||
|
log.WithContext(ctx).WithError(err).Error("Failed to setup trace provider")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
@ -8,7 +8,6 @@ import (
|
|||||||
"git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/database"
|
"git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/database"
|
||||||
"git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/models"
|
"git.anthrove.art/Anthrove/otter-space-sdk/v3/pkg/models"
|
||||||
pb "git.anthrove.art/Anthrove/plug-sdk/v3/pkg/grpc"
|
pb "git.anthrove.art/Anthrove/plug-sdk/v3/pkg/grpc"
|
||||||
"git.anthrove.art/Anthrove/plug-sdk/v3/pkg/telemetry"
|
|
||||||
log "github.com/sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"go.opentelemetry.io/otel"
|
"go.opentelemetry.io/otel"
|
||||||
"go.opentelemetry.io/otel/attribute"
|
"go.opentelemetry.io/otel/attribute"
|
||||||
@ -158,26 +157,3 @@ func (p *Plug) GetMessageExecution(function GetMessageExecution) {
|
|||||||
p.getMessageExecution = function
|
p.getMessageExecution = function
|
||||||
span.AddEvent("Get message execution function set")
|
span.AddEvent("Get message execution function set")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p Plug) SetupOpenTelemtry() error {
|
|
||||||
ctx, span := tracer.Start(p.ctx, "SetupOpenTelemtry")
|
|
||||||
defer span.End()
|
|
||||||
|
|
||||||
err := telemetry.SetupMeterProvider(p.plugName)
|
|
||||||
if err != nil {
|
|
||||||
span.RecordError(err)
|
|
||||||
span.SetStatus(codes.Error, err.Error())
|
|
||||||
log.WithContext(ctx).WithError(err).Error("Failed to setup meter provider")
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
err = telemetry.SetupTraceProvider(ctx, p.plugName)
|
|
||||||
if err != nil {
|
|
||||||
span.RecordError(err)
|
|
||||||
span.SetStatus(codes.Error, err.Error())
|
|
||||||
log.WithContext(ctx).WithError(err).Error("Failed to setup trace provider")
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user