2021-07-13 17:14:25 +00:00
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"os"
|
|
|
|
"strconv"
|
|
|
|
|
|
|
|
"github.com/christianselig/apollo-backend/internal/api"
|
|
|
|
"github.com/christianselig/apollo-backend/internal/cmdutil"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
)
|
|
|
|
|
|
|
|
func APICmd(ctx context.Context) *cobra.Command {
|
|
|
|
var port int
|
|
|
|
|
|
|
|
cmd := &cobra.Command{
|
|
|
|
Use: "api",
|
|
|
|
Args: cobra.ExactArgs(0),
|
|
|
|
Short: "Runs the RESTful API.",
|
|
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
2021-07-13 18:22:41 +00:00
|
|
|
port = 4000
|
2021-07-13 17:14:25 +00:00
|
|
|
if os.Getenv("PORT") != "" {
|
|
|
|
port, _ = strconv.Atoi(os.Getenv("PORT"))
|
|
|
|
}
|
|
|
|
|
|
|
|
logger := cmdutil.NewLogrusLogger(false)
|
|
|
|
|
|
|
|
statsd, err := cmdutil.NewStatsdClient()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer statsd.Close()
|
|
|
|
|
2021-07-23 00:22:46 +00:00
|
|
|
db, err := cmdutil.NewDatabasePool(ctx, 1)
|
2021-07-13 17:14:25 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer db.Close()
|
|
|
|
|
|
|
|
redis, err := cmdutil.NewRedisClient(ctx)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer redis.Close()
|
|
|
|
|
|
|
|
api := api.NewAPI(ctx, logger, statsd, db)
|
|
|
|
srv := api.Server(port)
|
|
|
|
|
2021-09-25 13:19:42 +00:00
|
|
|
go func() { _ = srv.ListenAndServe() }()
|
2021-07-13 17:14:25 +00:00
|
|
|
|
|
|
|
logger.WithFields(logrus.Fields{
|
|
|
|
"port": port,
|
|
|
|
}).Info("started api")
|
|
|
|
|
|
|
|
<-ctx.Done()
|
|
|
|
|
2021-09-25 13:19:42 +00:00
|
|
|
_ = srv.Shutdown(ctx)
|
2021-07-13 17:14:25 +00:00
|
|
|
|
|
|
|
return nil
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
return cmd
|
|
|
|
}
|