93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
_ "embed"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"${REPO_NAME_SNAKE}/internal/app"
|
|
proto "${REPO_NAME_SNAKE}/proto"
|
|
|
|
"github.com/grpc-ecosystem/grpc-gateway/v2/runtime"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/joho/godotenv"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
|
|
"github.com/swaggest/swgui/v5emb"
|
|
)
|
|
|
|
//go:embed main.swagger.json
|
|
var swaggerJSON []byte
|
|
|
|
func main() {
|
|
_ = godotenv.Load()
|
|
|
|
// PostgreSQL
|
|
dbpool, err := pgxpool.New(context.Background(), os.Getenv("DATABASE_URL"))
|
|
if err != nil {
|
|
log.Fatalf("Unable to create connection pool: %v\n", err)
|
|
}
|
|
defer dbpool.Close()
|
|
|
|
// Test request to DB
|
|
var greeting string
|
|
if err = dbpool.QueryRow(context.Background(), "select 'Hello, world!'").Scan(&greeting); err != nil {
|
|
log.Fatalf("QueryRow failed: %v\n", err)
|
|
}
|
|
|
|
// Create a listener on TCP port
|
|
lis, err := net.Listen("tcp", ":8080")
|
|
if err != nil {
|
|
log.Fatalln("Failed to listen:", err)
|
|
}
|
|
|
|
// Create a gRPC server object
|
|
s := grpc.NewServer()
|
|
// Attach the Greeter service to the server
|
|
proto.Register${REPO_NAME_PASCAL}Server(s, app.NewServer())
|
|
// Serve gRPC server
|
|
log.Println("Serving gRPC on 0.0.0.0:8080")
|
|
go func() {
|
|
log.Fatalln(s.Serve(lis))
|
|
}()
|
|
|
|
// Create a client connection to the gRPC server we just started
|
|
// This is where the gRPC-Gateway proxies the requests
|
|
conn, err := grpc.NewClient(
|
|
"0.0.0.0:8080",
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
)
|
|
if err != nil {
|
|
log.Fatalln("Failed to dial server:", err)
|
|
}
|
|
|
|
gwmux := runtime.NewServeMux()
|
|
// Register Greeter
|
|
err = proto.Register${REPO_NAME_PASCAL}Handler(context.Background(), gwmux, conn)
|
|
if err != nil {
|
|
log.Fatalln("Failed to register gateway:", err)
|
|
}
|
|
|
|
mainMux := http.NewServeMux()
|
|
|
|
docsHandler := v5emb.New("API", "/swagger/doc.json", "/docs/")
|
|
mainMux.Handle("/docs/", docsHandler)
|
|
mainMux.HandleFunc("/swagger/doc.json", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.Write(swaggerJSON)
|
|
})
|
|
mainMux.Handle("/api/", gwmux)
|
|
|
|
gwServer := &http.Server{
|
|
Addr: ":8090",
|
|
Handler: mainMux,
|
|
}
|
|
|
|
log.Println("Serving gRPC-Gateway on http://0.0.0.0:8090")
|
|
log.Println("Serving docs on http://0.0.0.0:8090/docs")
|
|
log.Fatalln(gwServer.ListenAndServe())
|
|
}
|