apollo-backend/internal/cmd/api.go

66 lines
1.1 KiB
Go
Raw Normal View History

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
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()
db, err := cmdutil.NewDatabasePool(ctx)
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)
go srv.ListenAndServe()
logger.WithFields(logrus.Fields{
"port": port,
}).Info("started api")
<-ctx.Done()
srv.Shutdown(ctx)
return nil
},
}
return cmd
}