cas_tt_cloud_backend/cmd/serve/serve.go

package serve

import (
	"SciencesServer/app"
	"SciencesServer/app/common"
	"SciencesServer/config"
	"SciencesServer/cron"
	"SciencesServer/router"
	"SciencesServer/rpc/client"
	"SciencesServer/serve/cache"
	"SciencesServer/serve/es"
	"SciencesServer/serve/logger"
	"SciencesServer/serve/orm"
	"SciencesServer/serve/web"
	"SciencesServer/task"
	"SciencesServer/tools"
	"SciencesServer/utils"
	"strings"

	"github.com/gin-gonic/gin"
)

type Serve struct {
	*Option
}

type Option struct {
	Config    string `json:"config"`
	RpcConfig string `json:"rpc_config"`
}

func (this *Serve) Run() {
	// 载入配置
	utils.LoadConfig(this.Option.Config, config.SettingInfo, func(i interface{}) {
		obj := i.(*config.Setting)
		obj.Upload.Exts = strings.Split(obj.Upload.Ext, ",")
		logger.NewLogger().Init(&logger.Option{File: obj.Log.File, LeastDay: obj.Log.LeastDay, Level: obj.Log.Level, IsStdout: false}).Load()
	})
	utils.LoadConfig(this.Option.RpcConfig, config.RPCServerSettingInfo, func(i interface{}) {
		obj := i.(*config.RPCServerSetting)
		go utils.TryCatch(func() {
			options := make(map[string]*client.ServerOption, 0)

			for k, v := range obj.Servers {
				options[k] = &client.ServerOption{
					Host: v.Host, Port: v.Port, IsTLS: v.IsTLS, TLSName: v.TLSName, Pem: v.Pem,
				}
			}
			client.NewServer(options).Run()
		})
	})
	cache.Init()
	//orm.Init()
	// TODO：待优化完善
	orm.NewInstance(
		orm.WithDebug(config.SettingInfo.Engine.Debug),
		orm.WithDBMode(config.SettingInfo.Engine.DBMode),
		orm.WithTablePrefix(config.SettingInfo.Engine.TablePrefix),
		orm.WithSingularTable(!config.SettingInfo.Engine.Complex),
		orm.WithMaxIdleConns(config.SettingInfo.Engine.MaxIdleConns),
		orm.WithMaxOpenConns(config.SettingInfo.Engine.MaxOpenConns),
		orm.WithMaxLifetime(config.SettingInfo.Engine.MaxLifetime),
		//orm.WithMysqlUser(config.SettingInfo.Engine.Mysql.User),
	).Init()
	task.Init()
	common.Init()
	cron.Init()
	app.Init()
	tools.Init()
	// 开启Elasticsearch
	es.NewEs()(&es.EsConfig{Address: []string{"http://192.168.0.188:9200"}}).Run()
	// 开启web
	web.NewWeb()(&web.WebConfig{
		Port: config.SettingInfo.Server.Port, ReadTimeout: config.SettingInfo.Server.ReadTimeout,
		WriteTimeout: config.SettingInfo.Server.WriteTimeout, IdleTimeout: config.SettingInfo.Server.IdleTimeout,
	}).Run(router.NewRouter(&router.Option{
		Mode: gin.DebugMode, IsCors: true,
		RateLimitConfig: &router.RateLimitConfig{
			IsRate: true, Limit: config.SettingInfo.Rate.Limit, Capacity: config.SettingInfo.Rate.Capacity,
		},
	}).Init())
}

func NewServe() func(option *Option) *Serve {
	return func(option *Option) *Serve {
		return &Serve{option}
	}
}