CorpusAI/cmd/http/http.go

package http

import (
	"github.com/sirupsen/logrus"
	"os"
	"os/signal"
	"strings"
	"syscall"

	"git.links123.net/Slate/CorpusAI/cmd/http/router"
	"git.links123.net/Slate/CorpusAI/service/store/cache"
	"git.links123.net/Slate/CorpusAI/service/store/mysql"
	"github.com/braintree/manners"
	"github.com/spf13/cobra"
)


// RunCommand cobra subcommand http
func RunCommand() *cobra.Command {
	var host, port string
	cmd := &cobra.Command{
		Use:   "http",
		Short: "Run the http service",
		Run: func(cmd *cobra.Command, args []string) {
			go Start(host, port)
			// 阻塞退出 捕获信号
			signalChan := make(chan os.Signal)
			signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
			logrus.Infof("caught signal %+v, begin garbage collection", <-signalChan)
			Stop()
		},
	}
	cmd.PersistentFlags().StringVarP(&host, "host", "o", "127.0.0.1", "server hostname")
	cmd.PersistentFlags().StringVarP(&port, "port", "p", "8080", "server port")

	return cmd
}

// Start use for cobra or testing
func Start(host, port string) {
	// init db
	mysql.Init()
	// init cache
	cache.Init()
	// build router
	r := router.BuildRouter()
	// start server
	err := manners.ListenAndServe(strings.Join([]string{host, port}, ":"), r)
	if err != nil {
		panic(err)
	}
}

// Stop stop the http service graceful
func Stop() {
	if manners.Close() {
		logrus.Info("http server stopped")
	}
	cache.Close()
	mysql.Close()
}