apollo-backend/internal/cmd/api.go
2022-05-23 14:29:15 -04:00

64 lines
1.2 KiB
Go

package cmd
import (
"context"
"os"
"strconv"
"github.com/christianselig/apollo-backend/internal/api"
"github.com/christianselig/apollo-backend/internal/cmdutil"
"github.com/spf13/cobra"
"go.uber.org/zap"
)
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 {
port = 4000
if os.Getenv("PORT") != "" {
port, _ = strconv.Atoi(os.Getenv("PORT"))
}
logger := cmdutil.NewLogger("api")
defer func() { _ = logger.Sync() }()
statsd, err := cmdutil.NewStatsdClient()
if err != nil {
return err
}
defer statsd.Close()
db, err := cmdutil.NewDatabasePool(ctx, 1)
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, redis, db)
srv := api.Server(port)
go func() { _ = srv.ListenAndServe() }()
logger.Info("started api", zap.Int("port", port))
<-ctx.Done()
_ = srv.Shutdown(ctx)
return nil
},
}
return cmd
}