-
Notifications
You must be signed in to change notification settings - Fork 4
/
server.go
115 lines (97 loc) · 2.78 KB
/
server.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package server
import (
"log"
"net/http"
"os"
"time"
"github.com/99designs/gqlgen/graphql/handler"
"github.com/99designs/gqlgen/graphql/handler/extension"
"github.com/99designs/gqlgen/graphql/handler/lru"
"github.com/99designs/gqlgen/graphql/handler/transport"
"github.com/99designs/gqlgen/graphql/playground"
"github.com/go-chi/chi"
"github.com/go-chi/chi/middleware"
"github.com/gorilla/websocket"
"github.com/rs/cors"
"github.com/scorpionknifes/gqlmanage/dataloader"
"github.com/scorpionknifes/gqlmanage/graphql"
customMiddleware "github.com/scorpionknifes/gqlmanage/middleware"
"github.com/scorpionknifes/gqlmanage/mongodb"
)
const defaultPort = "8080"
// Init start server
func Init() {
// get .env config
GetConfig()
// connect to db
db := ConnectDB()
// connect redis
rdb := connectRedis()
var (
userRepo = mongodb.UserRepo{DB: db.Collection("user")}
deviceRepo = mongodb.DeviceRepo{DB: db.Collection("device")}
roomRepo = mongodb.RoomRepo{DB: db.Collection("room")}
emailRepo = mongodb.EmailRepo{DB: db.Collection("email")}
)
router := chi.NewRouter()
router.Use(cors.New(cors.Options{
AllowedOrigins: []string{"*"},
AllowedMethods: []string{
http.MethodHead,
http.MethodGet,
http.MethodPost,
http.MethodPut,
http.MethodPatch,
http.MethodDelete,
},
AllowedHeaders: []string{"*"},
AllowCredentials: true,
Debug: false,
}).Handler)
router.Use(middleware.RequestID)
router.Use(middleware.Logger)
router.Use(middleware.RealIP)
router.Use(middleware.GetHead)
router.Use(middleware.Recoverer)
router.Use(customMiddleware.AuthMiddleware(userRepo))
d := &dataloader.DBLoader{
DeviceRepo: deviceRepo,
RoomRepo: roomRepo,
}
router.Use(dataloader.DataMiddleware(d))
c := graphql.Config{Resolvers: &graphql.Resolver{
UserRepo: userRepo,
DeviceRepo: deviceRepo,
RoomRepo: roomRepo,
EmailRepo: emailRepo,
Redis: *rdb,
}}
port := os.Getenv("PORT")
if port == "" {
port = defaultPort
}
srv := handler.New(graphql.NewExecutableSchema(c))
srv.AddTransport(transport.Websocket{
Upgrader: websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
CheckOrigin: func(r *http.Request) bool {
return true
},
},
KeepAlivePingInterval: 10 * time.Second,
})
srv.AddTransport(transport.Options{})
srv.AddTransport(transport.GET{})
srv.AddTransport(transport.POST{})
srv.AddTransport(transport.MultipartForm{})
srv.SetQueryCache(lru.New(1000))
srv.Use(extension.Introspection{})
srv.Use(extension.AutomaticPersistedQuery{
Cache: lru.New(100),
})
router.Handle("/", playground.Handler("GraphQL playground", "/query/"))
router.Handle("/query", srv)
log.Printf("connect to %s for GraphQL playground", os.Getenv("CORS_DEFAULT"))
log.Fatal(http.ListenAndServe(":"+port, router))
}