Merge pull request #10 from tokopedia/gqlserver_graphqlgo_config_interface

Config propogation from gqlserver to graphql-go

Author: Abhishek Srivastava

graphql.go

@@ -68,6 +68,7 @@ type Schema struct {
 	logger                log.Logger
 	useStringDescriptions bool
 	disableIntrospection  bool
+	isPanicLogEnabled     bool
 }
 
 // SchemaOpt is an option to pass to ParseSchema or MustParseSchema.
@@ -150,6 +151,19 @@ func (s *Schema) Validate(queryString string, variables map[string]interface{})
 	return false, validation.Validate(s.schema, doc, variables, s.maxDepth)
 }
 
+//EnablePanicLogging enables query info logging if panic occurs
+func (s *Schema) EnablePanicLogging() {
+	s.isPanicLogEnabled = true
+	s.logger = getLogger(s)
+}
+
+func getLogger(s *Schema) log.Logger {
+	if s != nil && s.isPanicLogEnabled {
+		return &log.CustomLogger{true}
+	}
+	return &log.DefaultLogger{}
+}
+
 // Exec executes the given query with the schema's resolver. It panics if the schema was created
 // without a resolver. If the context get cancelled, no further resolvers will be called and a
 // the context error will be returned as soon as possible (not immediately).

log/log.go

@@ -2,6 +2,7 @@ package log
 
 import (
 	"context"
+	"fmt"
 	"log"
 	"runtime"
 )
@@ -19,5 +20,24 @@ func (l *DefaultLogger) LogPanic(_ context.Context, value interface{}, info stri
 	const size = 64 << 10
 	buf := make([]byte, size)
 	buf = buf[:runtime.Stack(buf, false)]
-	log.Printf("graphql: panic occurred: %v\n%s\n\n%s", value, buf, info)
+		log.Printf("graphql: panic occurred: %v\n%s", value, buf)
 }
+
+// CustomLogger is the custom logger for use with gqlserver custome config to log panics that occur during query execution
+type CustomLogger struct{
+	VerbosePanicLog bool
+}
+
+// LogPanic is used to log recovered panic values that occur during query execution
+func (l *CustomLogger) LogPanic(_ context.Context, value interface{}, info string) {
+	const size = 64 << 10
+	buf := make([]byte, size)
+	buf = buf[:runtime.Stack(buf, false)]
+	msg := fmt.Sprintf("graphql: panic occurred: %v\n%s", value, buf)
+	if l.VerbosePanicLog {
+		msg = fmt.Sprintf("%s\n\n%s", msg, info)
+	}
+
+	log.Printf(msg)
+}
+