80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"time"
|
|
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/99designs/gqlgen/graphql/handler"
|
|
"github.com/sour-is/ev/api/gql_ev"
|
|
"github.com/sour-is/ev/internal/graph"
|
|
"github.com/sour-is/ev/internal/graph/generated"
|
|
"github.com/sour-is/ev/pkg/es"
|
|
diskstore "github.com/sour-is/ev/pkg/es/driver/disk-store"
|
|
memstore "github.com/sour-is/ev/pkg/es/driver/mem-store"
|
|
"github.com/sour-is/ev/pkg/es/service"
|
|
"github.com/sour-is/ev/pkg/playground"
|
|
)
|
|
|
|
func main() {
|
|
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
|
|
go func() {
|
|
<-ctx.Done()
|
|
defer cancel()
|
|
}()
|
|
|
|
if err := run(ctx); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
func run(ctx context.Context) error {
|
|
diskstore.Init(ctx)
|
|
memstore.Init(ctx)
|
|
|
|
es, err := es.Open(ctx, env("EV_DATA", "file:data"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
svc, err := service.New(ctx, es)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
res := graph.New(gql_ev.New(es))
|
|
gql := handler.NewDefaultServer(generated.NewExecutableSchema(generated.Config{Resolvers: res}))
|
|
|
|
s := http.Server{
|
|
Addr: env("EV_HTTP", ":8080"),
|
|
}
|
|
http.Handle("/", playground.Handler("GraphQL playground", "/gql"))
|
|
http.Handle("/gql", res.ChainMiddlewares(gql))
|
|
http.Handle("/event/", http.StripPrefix("/event/", svc))
|
|
|
|
log.Print("Listen on ", s.Addr)
|
|
g, ctx := errgroup.WithContext(ctx)
|
|
|
|
g.Go(s.ListenAndServe)
|
|
|
|
g.Go(func() error {
|
|
<-ctx.Done()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
return s.Shutdown(ctx)
|
|
})
|
|
|
|
return g.Wait()
|
|
}
|
|
func env(name, defaultValue string) string {
|
|
if v := os.Getenv(name); v != "" {
|
|
log.Println("# ", name, " = ", v)
|
|
return v
|
|
}
|
|
return defaultValue
|
|
}
|