rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add graphql log

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-10-11 23:14:45 +02:00
parent 8398f587de
commit e109cb602e
3 changed files with 52 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/server/api/console/v1/resolver.go
View File

@@ -233,7 +233,7 @@ func graphqlHandler(logger *log.Logger, proboSvc *probo.Service, usrmgrSvc *usrm
},
)
srv.Use(extension.Introspection{})
srv.Use(tracingExtension{})
srv.Use(gqlutils.NewTracingExtension(logger))
srv.SetRecoverFunc(gqlutils.RecoverFunc)
srv.AroundOperations(

1
pkg/server/api/trust/v1/resolver.go
View File

@@ -133,6 +133,7 @@ func graphqlHandler(logger *log.Logger, usrmgrSvc *usrmgr.Service, trustSvc *tru
srv.AddTransport(transport.Options{})
srv.Use(extension.Introspection{})
srv.Use(gqlutils.NewTracingExtension(logger))
srv.SetRecoverFunc(gqlutils.RecoverFunc)

61
pkg/server/api/console/v1/tracing_extensions.go → pkg/server/graphql/tracing.go
View File

@@ -12,28 +12,37 @@
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.
package console_v1
package graphql
import (
"context"
"fmt"
"time"
"github.com/99designs/gqlgen/graphql"
"go.gearno.de/kit/log"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/trace"
)
type tracingExtension struct{}
type TracingExtension struct {
logger *log.Logger
}
func (t tracingExtension) ExtensionName() string {
func NewTracingExtension(logger *log.Logger) TracingExtension {
return TracingExtension{logger: logger}
}
func (t TracingExtension) ExtensionName() string {
return "Tracing"
}
func (t tracingExtension) Validate(schema graphql.ExecutableSchema) error {
func (t TracingExtension) Validate(schema graphql.ExecutableSchema) error {
return nil
}
func (t tracingExtension) InterceptField(ctx context.Context, next graphql.Resolver) (interface{}, error) {
func (t TracingExtension) InterceptField(ctx context.Context, next graphql.Resolver) (interface{}, error) {
rootSpan := trace.SpanFromContext(ctx)
if rootSpan.IsRecording() {
@@ -61,18 +70,20 @@ func (t tracingExtension) InterceptField(ctx context.Context, next graphql.Resol
return next(ctx)
}
func (t tracingExtension) InterceptOperation(ctx context.Context, next graphql.OperationHandler) graphql.ResponseHandler {
func (t TracingExtension) InterceptOperation(ctx context.Context, next graphql.OperationHandler) graphql.ResponseHandler {
rootSpan := trace.SpanFromContext(ctx)
requestContext := graphql.GetOperationContext(ctx)
startTime := time.Now()
if rootSpan.IsRecording() {
tracer := otel.Tracer("graphql-operation")
requestContext := graphql.GetOperationContext(ctx)
operationName := "GraphQL Operation"
if requestContext.OperationName != "" {
operationName = "GraphQL " + requestContext.OperationName
}
ctx, span := tracer.Start(ctx, operationName)
var span trace.Span
ctx, span = tracer.Start(ctx, operationName)
defer span.End()
span.SetAttributes(
@@ -80,9 +91,37 @@ func (t tracingExtension) InterceptOperation(ctx context.Context, next graphql.O
attribute.String("graphql.operation_type", string(requestContext.Operation.Operation)),
attribute.String("graphql.query", requestContext.RawQuery),
)
return next(ctx)
}
return next(ctx)
handler := next(ctx)
return func(ctx context.Context) *graphql.Response {
resp := handler(ctx)
duration := time.Since(startTime)
operationType := string(requestContext.Operation.Operation)
operationName := requestContext.OperationName
if operationName == "" {
operationName = "unnamed"
}
if resp.Errors != nil {
t.logger.ErrorCtx(ctx,
fmt.Sprintf("%s %s failed %s", operationType, operationName, duration.String()),
log.String("graphql_operation_name", operationName),
log.String("graphql_operation_type", operationType),
log.Duration("graphql_operation_duration", duration),
log.Any("graphql_operation_errors", resp.Errors),
)
} else {
t.logger.InfoCtx(ctx,
fmt.Sprintf("%s %s succeed %s", operationType, operationName, duration.String()),
log.String("graphql_operation_name", operationName),
log.String("graphql_operation_type", operationType),
log.Duration("graphql_operation_duration", duration),
)
}
return resp
}
}