ytv/main.go at master · yueli-official/ytv · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package main

import (
	"io"
	"os"
	"strings"
	"tv/conf"
	"tv/service"

	"github.com/gin-gonic/gin"
	"github.com/rs/zerolog/log"

	"github.com/Yuelioi/gkit/logx/zero"
	"github.com/Yuelioi/gkit/web/gin/middleware/cachecontrol"
	"github.com/Yuelioi/gkit/web/gin/middleware/log/gzero"
	"github.com/Yuelioi/gkit/web/gin/middleware/ratelimit"
	"github.com/Yuelioi/gkit/web/gin/server"
)

func main() {

	// 配置日志
	logger := zero.Default()
	log.Logger = logger

	// 初始化配置
	if err := conf.InitConfig("data/config.yaml"); err != nil {
		log.Fatal().Msg("加载配置失败")
		panic(err)
	}

	// 禁用gin log
	gin.DefaultWriter = io.Discard
	gin.DefaultErrorWriter = io.Discard

	mw := []gin.HandlerFunc{
		gzero.Default(logger),
		gzero.GinRecovery(logger),
		// cachecontrol.Default(),
		ratelimit.Default(),
	}
	port := "9000"
	if conf.Cfg.App.Port != "" {
		port = conf.Cfg.App.Port
	}

	if !strings.HasPrefix(port, ":") {
		port = ":" + port
	}

	server.Start(server.ServerConfig{
		Addr:        port,
		Logger:      logger,
		Mode:        strings.ToLower(os.Getenv("APP_MODE")),
		EnableCORS:  true,
		APIPrefix:   "/api/v1",
		Middlewares: mw,
		SPAPath:     "./frontend/dist",
	}, func(api *gin.RouterGroup) {

		api.Use((cachecontrol.Default()), ratelimit.Default())
		{
			api.GET("/search", service.SearchVideoAPI)
			api.GET("/hot", service.HotMovies)
			api.GET("/vod", service.SearchVideoById)
		}
	})

}