graphql-engine/cli/pkg/console/serve.go

108 lines
2.7 KiB
Go
Raw Normal View History

2020-04-07 12:23:20 +03:00
package console
import (
"fmt"
"net/http"
"os"
"sync"
"github.com/hasura/graphql-engine/cli/v2"
2020-04-07 12:23:20 +03:00
"github.com/fatih/color"
"github.com/pkg/errors"
"github.com/skratchdot/open-golang/open"
)
type ServeOpts struct {
APIServer *APIServer
ConsoleServer *ConsoleServer
EC *cli.ExecutionContext
DontOpenBrowser bool
Browser string
ConsolePort string
APIPort string
Address string
SignalChanAPIServer chan os.Signal
2020-04-07 12:23:20 +03:00
SignalChanConsoleServer chan os.Signal
WG *sync.WaitGroup
2020-04-07 12:23:20 +03:00
}
// Server console and API Server
func Serve(opts *ServeOpts) error {
// get HTTP servers
apiHTTPServer := opts.APIServer.GetHTTPServer()
consoleHTTPServer, err := opts.ConsoleServer.GetHTTPServer()
if err != nil {
return errors.Wrap(err, "cannot create console server")
}
go func() {
<-opts.SignalChanAPIServer
if err := apiHTTPServer.Close(); err != nil {
opts.EC.Logger.Debugf("unable to close server running on port %s", opts.APIPort)
}
}()
go func() {
<-opts.SignalChanConsoleServer
if err := consoleHTTPServer.Close(); err != nil {
opts.EC.Logger.Debugf("unable to close server running on port %s", opts.ConsolePort)
}
}()
// Create WaitGroup for running 2 servers
wg := opts.WG
wg.Add(1)
go func() {
if err := apiHTTPServer.ListenAndServe(); err != nil {
if err == http.ErrServerClosed {
opts.EC.Logger.Infof("server closed on port %s under signal", opts.APIPort)
} else {
opts.EC.Logger.WithError(err).Errorf("error listening on port %s", opts.APIPort)
}
}
wg.Done()
}()
wg.Add(1)
go func() {
if err := consoleHTTPServer.ListenAndServe(); err != nil {
if err == http.ErrServerClosed {
opts.EC.Logger.Infof("server closed on port %s under signal", opts.ConsolePort)
} else {
opts.EC.Logger.WithError(err).Errorf("error listening on port %s", opts.ConsolePort)
}
}
wg.Done()
}()
consoleURL := fmt.Sprintf("http://%s:%s/", opts.Address, opts.ConsolePort)
if !opts.DontOpenBrowser {
if opts.Browser != "" {
opts.EC.Spin(color.CyanString("Opening console on: %s", opts.Browser))
defer opts.EC.Spinner.Stop()
err = open.RunWith(consoleURL, opts.Browser)
if err != nil {
opts.EC.Logger.WithError(err).Warnf("failed opening console in '%s', try to open the url manually", opts.Browser)
}
} else {
opts.EC.Spin(color.CyanString("Opening console using default browser..."))
defer opts.EC.Spinner.Stop()
err = open.Run(consoleURL)
if err != nil {
opts.EC.Logger.WithError(err).Warn("Error opening browser, try to open the url manually?")
}
}
}
opts.EC.Spinner.Stop()
opts.EC.Logger.Infof("console running at: %s", consoleURL)
opts.EC.Telemetry.Beam()
wg.Wait()
return nil
}