-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
89 lines (78 loc) · 2.53 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
// Copyright 2021 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main
import (
"bytes"
"html/template"
"log"
"net/http"
"os"
"github.com/99designs/gqlgen/graphql/handler"
"github.com/99designs/gqlgen/graphql/playground"
"github.com/openconfig/catalog-server/graph"
"github.com/openconfig/catalog-server/graph/generated"
"github.com/openconfig/catalog-server/pkg/db"
)
const (
// default port of launched catalog server
defaultPort = "8080"
// updateHTMLPath is the path to the update.html file.
updateHTMLPath = "frontend/static/update.html"
)
// Main function is automatically generated by *gqlgen*, except ConnectDB() to connect to database.
func main() {
port := os.Getenv("PORT")
if port == "" {
port = defaultPort
}
// Establish connection with database, other codes are automatically generated.
err := db.ConnectDB()
if err != nil {
log.Fatal(err)
}
srv := handler.NewDefaultServer(generated.NewExecutableSchema(generated.Config{Resolvers: &graph.Resolver{}}))
// Launch built-in graphQL frontend server.
http.Handle("/", playground.Handler("GraphQL playground", "/query"))
// Set handler for all queries.
http.Handle("/query", srv)
// static file server to serve frontend webpages.
updateHTMLTemplate, err := os.ReadFile(updateHTMLPath)
if err != nil {
log.Fatal(err)
}
projectID, ok := os.LookupEnv("PROJECT_ID")
if !ok {
log.Fatal("$PROJECT_ID not set.")
}
t, err := template.New("update.html").Parse(string(updateHTMLTemplate))
var updateHTMLData bytes.Buffer
if err := t.Execute(&updateHTMLData,
struct {
ProjectID string
}{
ProjectID: projectID,
}); err != nil {
log.Fatalf("Cannot re-write update.html: %v", err)
}
os.WriteFile(updateHTMLPath, updateHTMLData.Bytes(), 0640)
fileServer := http.FileServer(http.Dir("frontend"))
http.HandleFunc(
"/static/",
func(w http.ResponseWriter, r *http.Request) {
fileServer.ServeHTTP(w, r)
},
)
log.Printf("connect to http://localhost:%s/ for GraphQL playground", port)
log.Fatal(http.ListenAndServe(":"+port, nil))
}