feat(server): implement referenceAnswerResult

Author: pan93412

cmd/backend/dependencies.go

@@ -20,6 +20,7 @@ import (
 	"github.com/database-playground/backend-v2/internal/auth"
 	"github.com/database-playground/backend-v2/internal/config"
 	"github.com/database-playground/backend-v2/internal/httputils"
+	"github.com/database-playground/backend-v2/internal/sqlrunner"
 	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"github.com/redis/rueidis"
@@ -60,9 +61,13 @@ func CorsMiddleware(cfg config.Config) Middleware {
 	}
 }
 
+func SqlRunner(cfg config.Config) *sqlrunner.SqlRunner {
+	return sqlrunner.NewSqlRunner(cfg.SqlRunner)
+}
+
 // GqlgenHandler creates a gqlgen handler.
-func GqlgenHandler(entClient *ent.Client, storage auth.Storage) *handler.Server {
-	srv := handler.New(graph.NewSchema(entClient, storage))
+func GqlgenHandler(entClient *ent.Client, storage auth.Storage, sqlrunner *sqlrunner.SqlRunner) *handler.Server {
+	srv := handler.New(graph.NewSchema(entClient, storage, sqlrunner))
 
 	srv.AddTransport(transport.Options{})
 	srv.AddTransport(transport.GET{})

cmd/backend/server.go

@@ -15,6 +15,7 @@ func main() {
 		deps.FxCommonModule,
 		fx.Provide(
 			AuthStorage,
+			SqlRunner,
 			AnnotateMiddleware(AuthMiddleware),
 			AnnotateMiddleware(MachineMiddleware),
 			AnnotateMiddleware(CorsMiddleware),

graph/ent.resolvers.go

@@ -14,12 +14,14 @@ extend type Mutation {
   """
   Create a question.
   """
-  createQuestion(input: CreateQuestionInput!): Question! @scope(scope: "question:write")
+  createQuestion(input: CreateQuestionInput!): Question!
+    @scope(scope: "question:write")
 
   """
   Update a question.
   """
-  updateQuestion(id: ID!, input: UpdateQuestionInput!): Question! @scope(scope: "question:write")
+  updateQuestion(id: ID!, input: UpdateQuestionInput!): Question!
+    @scope(scope: "question:write")
 
   """
   Delete a question.
@@ -29,15 +31,26 @@ extend type Mutation {
   """
   Create a database.
   """
-  createDatabase(input: CreateDatabaseInput!): Database! @scope(scope: "database:write")
+  createDatabase(input: CreateDatabaseInput!): Database!
+    @scope(scope: "database:write")
 
   """
   Update a database.
   """
-  updateDatabase(id: ID!, input: UpdateDatabaseInput!): Database! @scope(scope: "database:write")
+  updateDatabase(id: ID!, input: UpdateDatabaseInput!): Database!
+    @scope(scope: "database:write")
 
   """
   Delete a database.
   """
   deleteDatabase(id: ID!): Boolean! @scope(scope: "database:write")
-}
+}
+
+extend type Question {
+  referenceAnswerResult: SqlResponse! @scope(scope: "question:read")
+}
+
+type SqlResponse {
+  columns: [String!]!
+  rows: [[String!]!]!
+}

graph/model/models_gen.go

@@ -9,6 +9,7 @@ import (
 	"github.com/database-playground/backend-v2/graph/defs"
 	"github.com/database-playground/backend-v2/graph/directive"
 	"github.com/database-playground/backend-v2/internal/auth"
+	"github.com/database-playground/backend-v2/internal/sqlrunner"
 	"github.com/database-playground/backend-v2/internal/useraccount"
 	"github.com/vektah/gqlparser/v2/gqlerror"
 )
@@ -19,14 +20,15 @@ import (
 
 // Resolver is the resolver root.
 type Resolver struct {
-	ent  *ent.Client
-	auth auth.Storage
+	ent       *ent.Client
+	auth      auth.Storage
+	sqlrunner *sqlrunner.SqlRunner
 }
 
 // NewSchema creates a graphql executable schema.
-func NewSchema(ent *ent.Client, auth auth.Storage) graphql.ExecutableSchema {
+func NewSchema(ent *ent.Client, auth auth.Storage, sqlrunner *sqlrunner.SqlRunner) graphql.ExecutableSchema {
 	return NewExecutableSchema(Config{
-		Resolvers: &Resolver{ent, auth},
+		Resolvers: &Resolver{ent, auth, sqlrunner},
 		Directives: DirectiveRoot{
 			Scope: directive.ScopeDirective,
 		},