video_course/main.go

package main

import (
	"flag"
	"fmt"
	"os"
	"strconv"
	"strings"
	"video_course/controller"
	"video_course/docs"
	"video_course/global"
	"video_course/logger"
	"video_course/middleware"
	"video_course/repository/postgres"
	"video_course/repository/redis"

	"gitee.com/zr233/bsf"
	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"
	ginSwagger "github.com/swaggo/gin-swagger"
	"github.com/swaggo/gin-swagger/swaggerFiles"
)

var _VERSION_ = "unknown"

func flagInit() {

	flag.BoolVar(&global.Project.Debug, "d", global.Project.Debug, "是否debug")
	flag.StringVar(&global.Project.Host, "s", global.Project.Host, "swag host")
	flag.BoolVar(&help, "h", false, "this help")

	flag.Parse()
	if help {
		flag.Usage()
		os.Exit(0)
	}
}

// @title Video Course框架
// @version 1.0
// @description Video Course框架 API 文档
// @termsOfService http://swagger.io/terms/

// @contact.name API Support
// @contact.url http://www.swagger.io/support
// @contact.email support@swagger.io

// @license.name Apache 2.0
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html

// @BasePath /v1

// @securityDefinitions.basic BasicAuth

// @securityDefinitions.apikey ApiKeyAuth
// @in header
// @name Authorization

// @securitydefinitions.oauth2.application OAuth2Application
// @tokenUrl https://example.com/oauth/token
// @scope.write Grants write access
// @scope.admin Grants read and write access to administrative information

// @securitydefinitions.oauth2.implicit OAuth2Implicit
// @authorizationurl https://example.com/oauth/authorize
// @scope.write Grants write access
// @scope.admin Grants read and write access to administrative information

// @securitydefinitions.oauth2.password OAuth2Password
// @tokenUrl https://example.com/oauth/token
// @scope.read Grants read access
// @scope.write Grants write access
// @scope.admin Grants read and write access to administrative information

// @securitydefinitions.oauth2.accessCode OAuth2AccessCode
// @tokenUrl https://example.com/oauth/token
// @authorizationurl https://example.com/oauth/authorize
// @scope.admin Grants read and write access to administrative information
func main() {
	logger.Init()
	global.Init()
	logger.SetOnlineWriters()
	flagInit()
	postgres.Init()
	redis.Init()

	r := bsf.NewDefaultEngine()
	gin.SetMode(gin.ReleaseMode)
	r.Use(
		middleware.ErrorResponse(),
		middleware.Logger(),
		middleware.Session(),
		middleware.Cors(),
	)
	r.AutoRegisterController("/v1", &controller.Auth{})
	r.AutoRegisterController("/v1", &controller.HrSensors{})
	r.AutoRegisterController("/v1", &controller.User{})
	//debug模式下启用swag文档
	if global.Project.Debug {
		gin.SetMode(gin.DebugMode)
		//设置swagger文档主机
		strSilce := strings.Split(global.Project.Host, ":")
		host := strSilce[0]
		port := global.Port
		if len(strSilce) > 1 {
			port_, err := strconv.Atoi(strSilce[1])
			if err == nil {
				port = port_
			}
		}

		docs.SwaggerInfo.Host = fmt.Sprintf("%s:%d", host, port)
		// 注册swagger文档
		// 查看地址为 /v1/docs/index.html     /v1/SignIn/docs/doc.json
		r.GET("/v1/docs/*any", ginSwagger.WrapHandler(swaggerFiles.Handler))
	}

	logrus.Infof("Version:[%s]\n", _VERSION_)

	_ = r.Run(fmt.Sprintf(":%d", global.Port))

}

var (
	help bool
)